Advertisement
DiYane

Zoo

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