Advertisement
simeonshopov

Zoo

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