Advertisement
mbstanchev

zoo

Jul 2nd, 2022
51
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.15 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 == 'mammals':
  12. self.mammals.append(name)
  13. elif species == 'fish':
  14. self.fishes.append(name)
  15. elif species == 'bird':
  16. self.birds.append(name)
  17.  
  18. Zoo.__animals += 1
  19.  
  20.  
  21. def get_info(self, species):
  22. result = ''
  23. if species == 'mammals':
  24. result += f"Mammals in {self.name}: {', '.join(self.mammals)}\n"
  25.  
  26. elif species == 'fish':
  27. result += f"Fishes in {self.name}: {', '.join(self.fishes)}\n"
  28.  
  29. elif species == 'birds':
  30. result += f"Birds in {self.name}: {', '.join(self.birds)}\n"
  31. result += f"Total animals: {Zoo.__animals}"
  32. return result
  33.  
  34.  
  35. zoo_name = input()
  36. zoo = Zoo(zoo_name)
  37. count = int(input())
  38.  
  39. for p in range(count):
  40. animal = input().split(' ')
  41. species = animal[0]
  42. name = animal[1]
  43. zoo.add_animal(species, name)
  44.  
  45. type = input()
  46. print(zoo.get_info(type))
  47.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement