simeonshopov

Zoo

Jan 17th, 2020
227
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.01 KB | None | 0 0
  1. class Zoo:
  2.   __animals = 0
  3.  
  4.   def __init__(self, name):
  5.     self.name = name
  6.     self.mammals = []
  7.     self.fishes = []
  8.     self.birds = []
  9.  
  10.   def add_animal(self, species, name):
  11.     if species == 'mammal':
  12.       self.mammals.append(name)
  13.     elif species == 'fish':
  14.       self.fishes.append(name)
  15.     elif species == 'birds':
  16.       self.birds.append(name)
  17.     zoo.__animals += 1
  18.  
  19.   def get_info(self, species):
  20.     result = ''
  21.     if species == 'mammal':
  22.       result += f'Mammals in {self.name}: {', '.join(self.mammals}\n'
  23.     elif species == 'fish':
  24.       result += f'Fishes: {', '.join(self.fishes)}\n'
  25.     elif species == 'birds':
  26.       result += f'Birds: {', '.join(self.birds)}\n'
  27.     result += f'Total animals: {zoo.__animals}'
  28.     return result
  29.  
  30. zoo_name = input()
  31. zoo = Zoo(zoo_name)
  32. count = int(input())
  33.  
  34. for i in range(count):
  35.   animal = input().split(' ')
  36.   species = animal[0]
  37.   name = animal[1]
  38.   zoo.add_animal(species, name)
  39.  
  40. info = input()
  41.  
  42. print(zoo.get_info(info))
Advertisement
Add Comment
Please, Sign In to add comment