Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class Drink:
- def __init__(self, name, energy):
- self.name = name
- self.energy = energy
- class Food:
- def __init__(self, name, energy):
- self.name = name
- self.energy = energy
- class SomeClass:
- def __init__(self):
- self.items = []
- def add_item(self, *args):
- for item in args:
- self.items.append(item)
- def consume_supply(self, supply_type):
- for i in range(len(self.items) - 1, - 1, -1):
- current_supply = self.items[i]
- if supply_type == current_supply.__class__.__name__:
- self.items.remove(current_supply)
- break
- def __str__(self):
- result = ""
- result += ", ".join([x.name for x in self.items])
- return result
- water = Drink("water", 25)
- tea = Drink("tea", 35)
- milk_tea = Drink("milk tea", 30)
- coffee = Drink("coffee", 25)
- apple = Food("apple", 15)
- banana = Drink("banana", 28)
- grilled_chicken = Food("grilled chicken", 25)
- trader = SomeClass()
- item_list = [water, apple, grilled_chicken, milk_tea, tea, coffee, banana, banana, water, apple]
- trader.add_item(*item_list)
- print(trader)
- trader.consume_supply("Drink")
- print(trader)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement