1import math 
  2import copy
  3
  4#написано до Преходност
  5#done except tick()
  6
  7class Potion:
  8
  9    def __init__(self, effects : dict, duration : int) -> None:
 10        self.effects = effects
 11        self.duration = duration
 12        self.used_effects = set()
 13        self.potency = dict.fromkeys(effects, 1)
 14        self.used = False
 15
 16    def __getattr__(self, name : str):
 17        if name in self.effects:
 18            if name not in self.used_effects:
 19                self.used_effects .add(name)
 20                return self.__apply_potency__(name)
 21            else:
 22                raise TypeError("Effect is depleted.")
 23    
 24    def __add__(self, other):
 25        if self.used:
 26            raise TypeError("Potion is now part of something bigger than itself.")
 27        
 28        self.used_effects .update(other.used)
 29        self.duration = max(self.duration, other.duration)
 30        for effect in other.effects:
 31            if effect in self.effects:
 32                self.potency[effect] += other.potency[effect]
 33            else:
 34                self.effects[effect] = other.effects[effect]
 35        
 36        self.used = True
 37        return self
 38    
 39    def __sub__(self, other):
 40        if self.used:
 41            raise TypeError("Potion is now part of something bigger than itself.")
 42        
 43        self.used_effects .difference(other.used)
 44        for effect in other.effects:
 45            if effect in self.effects:
 46                self.potency[effect] -= other.potency[effect]
 47            else:
 48                raise TypeError(f"Effect \"{effect}\" not present.")
 49
 50        self.__delete_removed_effects__()
 51
 52        self.used = True
 53        return self
 54    
 55    def __mul__(self, number):
 56        if self.used:
 57            raise TypeError("Potion is now part of something bigger than itself.")
 58        
 59        for effect in self.potency:
 60            self.potency[effect] *= number
 61            if (self.potency[effect] - math.floor(self.potency[effect])) > 0.5:
 62                self.potency[effect] = math.ceil(self.potency[effect])
 63            else:
 64                self.potency[effect] = math.floor(self.potency[effect])
 65
 66        self.__delete_removed_effects__()   
 67
 68        self.used = True    
 69        return self
 70    
 71    def __truediv__ (self, number : int):
 72        if self.used:
 73            raise TypeError("Potion is now part of something bigger than itself.")
 74        
 75        temp_potion = Potion(effects=self.effects, duration=self.duration)
 76        temp_potion.potency = self.potency
 77        temp_potion = temp_potion * (1 / number)
 78        temp_potion.__delete_removed_effects__()
 79
 80        self.used = True
 81        return [temp_potion for _ in range(number)]
 82
 83    def __eq__(self, other: object) -> bool:
 84        if self.__total_potency__() != other.__total_potency__():
 85            return False
 86
 87        return self.__compare__effects__(other)
 88    
 89    def __lt__(self, other: object) -> bool:
 90        return self.__total_potency__() < other.__total_potency__()
 91    
 92    def __rt__(self, other: object) -> bool:
 93        return self.__total_potency__() > other.__total_potency__()
 94    
 95    def __apply_potency__(self, name):
 96        def apply_potion(target):
 97            for _ in range(self.potency[name]):
 98                self.effects[name](target)
 99            #return target
100        return apply_potion
101    
102    def __delete_removed_effects__(self) -> None:
103        #self.effects = {effect:self.effects[effects] for effect in self.effects if self.potency[effect] > 0}
104        #self.potency = {effect:self.potency[effects] for effect in self.potency if effect in self.effect}
105    
106        for effect in self.potency:
107            if self.potency[effect] <= 0:
108                self.effects.pop(effect)
109                self.potency[effect] = 0
110
111        self.potency = {effect:self.potency[effect] for effect in self.effects}
112
113    def __total_potency__(self) -> int:
114        return sum([self.potency[effect] for effect in self.potency])
115    
116    def __compare__effects__(self, __other):
117        for effect in self.effects:
118            if effect not in __other.effects:
119                return False
120            
121        return True
122
123def calculate_molecular_mass(name : str) -> int:
124    return sum(ord(symbol) for symbol in name)
125
126class ГоспожатаПоХимия:
127
128    def __init__(self) -> None:
129        self.potion_effects = dict()
130        self.original_targets = dict()
131        self.affected_targets = set()
132        
133    def apply(self, target, potion : Potion) -> None:
134        if potion.used:
135            raise TypeError("Potion is depleted.")
136        
137        self.affected_targets.add(target)
138        original_target = copy.deepcopy(target)
139        self.original_targets[id(target)] = original_target
140        if id(target) not in self.potion_effects:
141            self.potion_effects[id(target)] = [potion]
142        else:
143            self.potion_effects[id(target)].append(potion)
144
145        for effect in potion.effects:
146            potion.effects[effect](target)
147            
148        potion.used = True
149
150    def tick(self) -> None:
151        for target in self.affected_targets:
152            flag = False
153            for potion in self.potion_effects[id(target)]:
154                if potion.duration > 0:
155                    potion.duration -= 1
156                else:
157                    flag = True
158            if flag:
159                self.potion_effects[id(target)] = [potion for potion in self.potion_effects[id(target)] if potion.duration > 0]
160                temp_target = copy.deepcopy(self.original_targets[id(target)])
161                for potion in self.potion_effects[id(target)]:
162                    potion.used = False
163                    self.apply(temp_target, potion)
164                target = temp_target
165
166
167                    
168  
169"""
170
171effects = {'grow': lambda target: setattr(target, 'size', target.size*2)}
172
173class Target:
174
175    def __init__(self) -> None:
176        self.size = 5
177
178class TestTarget:
179    def __init__(self, x , y, size) -> None:
180        self.x = x
181        self.y = y
182        self.size = size
183
184target = Target()
185
186grow = Potion(effects=effects, duration=3)
187print(target.size)
188narco = ГоспожатаПоХимия()
189narco.apply(target, grow)
190
191print(target.size)
192narco.tick()
193print(target.size)
194narco.tick()
195print(target.size)
196narco.tick()
197print(target.size)
198narco.tick()
199print(target.size)
200narco.tick()
201print(target.size)
202
203
204teleport_potion
205combined_potion = teleport_potion + grow_potion
206 
207target = TestTarget(1, 1, 1)
208 
209combined_potion.teleport(target)
210combined_potion.grow(target)
211 
212"""
213
214#target.size = 5
215#grow = Potion(effects=effects, duration=2) * 3
216#grow.grow(target)
217#print(target.size)
218
219#target.size = 5
220#grow = Potion(effects=effects, duration=2) * 3
221#grow = grow * 0.33
222#grow.grow(target)
223#print(target.size)
224
225#target.size = 5
226#grow = Potion(effects=effects, duration=2) * 3
227#grow = grow - Potion(effects=effects, duration=2)
228#grow = grow - Potion(effects=effects, duration=2)
229#grow = grow - Potion(effects=effects, duration=2)
230#grow.grow(target)
231#print(target.size)
232
233
234#grow = Potion(effects=effects, duration=2)
235#grow.potency["grow"] = 4
236
237#small_grow ,_= grow / 2
238#small_grow.grow(target)
239#print(target.size)
240
241#grow = Potion(effects=effects, duration=2)
242#grow2 = Potion(effects=effects, duration=2)
243#grow2.potency["grow"] = 1
244#print(grow == grow2)
245
246
247#d = { "d":3454, "22":3456, "agds":5345, "A":456, "X":4567, "CCCCCCCCCCCCCCCCCCC":56}
248#d={k:d[k] for k in sorted(d,key=calculate_molecular_mass)}
249#print(d)
250
251#target = Target()
252#target2 = copy.copy(target)
253#print(target.size, target2.size)
254#target2.size = 10
255#print(target.size, target2.size)