Advertisement
pacho_the_python

removing_last_obf_from_list_by_ class_type

Jul 23rd, 2022
148
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.24 KB | None | 0 0
  1. class Drink:
  2.     def __init__(self, name, energy):
  3.         self.name = name
  4.         self.energy = energy
  5.  
  6.  
  7. class Food:
  8.     def __init__(self, name, energy):
  9.         self.name = name
  10.         self.energy = energy
  11.  
  12.  
  13. class SomeClass:
  14.     def __init__(self):
  15.         self.items = []
  16.  
  17.     def add_item(self, *args):
  18.         for item in args:
  19.             self.items.append(item)
  20.  
  21.     def consume_supply(self, supply_type):
  22.         for i in range(len(self.items) - 1, - 1, -1):
  23.             current_supply = self.items[i]
  24.             if supply_type == current_supply.__class__.__name__:
  25.                 self.items.pop(i)
  26.                 break
  27.  
  28.     def __str__(self):
  29.         result = ""
  30.         result += ", ".join([x.name for x in self.items])
  31.         return result
  32.  
  33.  
  34. water = Drink("water", 25)
  35. tea = Drink("tea", 35)
  36. milk_tea = Drink("milk tea", 30)
  37. coffee = Drink("coffee", 25)
  38. apple = Food("apple", 15)
  39. banana = Food("banana", 28)
  40. grilled_chicken = Food("grilled chicken", 25)
  41.  
  42. trader = SomeClass()
  43. item_list = [water, apple, grilled_chicken, milk_tea, tea, coffee, banana, banana, water, apple]
  44. trader.add_item(*item_list)
  45. print(trader)
  46. trader.consume_supply("Drink")
  47. print(trader)
  48. trader.consume_supply("Drink")
  49. print(trader)
  50.  
  51.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement