1class Potion:
  2    def __init__(effects, duration):
  3        self.__effects__ = {}
  4        self.__duration__ = duration
  5        self.__is_used__ = False
  6
  7        for key, effect in effects.items():
  8            self.__effects__[key] = (1, effect)
  9
 10            setattr(self, key, lambda effect_name=key: self.apply_effect(effect_name))
 11
 12    def apply_effect(self, effect_key):
 13        if self.__is_used__:
 14            Potion.__raise_exception__()
 15
 16        intensity, lambda_effect = self.__effects__.get(effect_key, (0, None))
 17
 18        for _ in range(intensity):
 19            lambda_effect()
 20
 21    @staticmethod
 22    def __raise_exception__():
 23        raise TypeError("Potion is now part of something bigger than itself.")
 24
 25    def set_used(self):
 26        self.__is_used__ = True
 27
 28        for key in self.__effects__.keys():
 29            self.__effects__[key] = Potion.__raise_exception__
 30
 31    @staticmethod
 32    def check_and_set_used(*args):
 33        for obj in args:
 34            if obj.__is_used__:
 35                Potion.__raise_exception()
 36
 37            obj.set_used()
 38
 39    @staticmethod
 40    def __combine_effects__(effects1, effects2):
 41        combined_effects = {}
 42
 43        for key, (first_intensity, first_effect) in effects1.items():
 44            second_intensity = effects2[key][0] if key in effects2 else 0
 45
 46            combined_effects[key] = (first_intensity + second_intensity, first_effect)
 47
 48        for key, (second_intensity, second_effect) in effects2.items():
 49            if key not in effects1:
 50                combined_effects[key] = (second_intensity, second_effect)
 51
 52        return combined_effects
 53
 54    def __add__(self, other):
 55        Potion.check_and_set_used(self, other)
 56
 57        return Potion(Potion.__combine_effects__(self.__effects__, other.__effects__), max(self.__duration__, other.__duration__))
 58
 59    # тъпо, защото 0.5 трябва да се закръгля до 1, а не до 0.
 60    @staticmethod
 61    def __custom_round__(x):
 62        if x - int(x) >= 0.5:
 63            return int(x) + 1
 64        else:
 65            return int(x)
 66
 67    def __mul__(self, multiplier):
 68        Potion.check_and_set_used(self)
 69
 70        for key, (intensity, effect) in self.__effects__.items():
 71            self.__effects__[key] = (Potion.__custom_round__(intensity * multiplier), effect)
 72
 73    @staticmethod
 74    def __remove_effects__(effects1, effects2):
 75        removed_effects = {}
 76
 77        if set(dict2.keys()) - set(dict1.keys()):
 78            raise TypeError("Мамка му, аз нямам такъв ефект.")
 79
 80        for key, (first_intensity, effect) in effects1.items():
 81            second_intensity = effects2.get(key, (None, None))[0]
 82
 83            if second_intensity is None:
 84                removed_effects[key] = (first_intensity, effect)
 85            else if first_intensity > second_intensity:
 86                removed_effects[key] = (first_intensity - second_intensity, effect)
 87
 88        return removed_effects
 89
 90    def __sub__(self, other):
 91        Potion.check_and_set_used(self, other)
 92
 93        return Potion(Potion.__remove_effects__(self.__effects__, other.__effects__), self.__duration__)
 94
 95    def __truediv__(self, divider):
 96        Potion.check_and_set_used(self)
 97
 98        result_potion = self * (1 / divider)
 99
100        return (result_potion, ) * divider
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116# effects = {'grow': lambda target: setattr(target, 'size', target.size*2)}