Python @ ФМИ

Резултати

4 точки от тестове

0 точки от учител

4 точки общо

9 успешни теста

11 неуспешни теста

Код

Скрий всички коментари

Пренеси редовете

  1from copy import deepcopy, copy
  2
  3
  4class Potion:
  5    @staticmethod
  6    def custom_round(number):
  7        if number - int(number) > 0.5:
  8            return int(number) + 1
  9        return int(number)
 10
 11    def __init__(self, effects, duration):
 12        self.effects = effects
 13        self.used_effects = []
 14        self.effects_intensity = {key: 1 for key in effects.keys()}
 15        self.duration = duration
 16        self.used = False
 17
 18    def __getattr__(self, item):
 19        def decorator(func):
 20            def wrapper(target):
 21                for _ in range(self.effects_intensity[list(self.effects.keys())[list(self.effects.values()).index(func)]]):
 22                    func(target)
 23            return wrapper
 24
 25        if self.used:
 26            raise TypeError('Potion is now part of something bigger than itself.')
 27
 28        if item in self.effects.keys():
 29            if item in self.used_effects:
 30                raise TypeError('Effect is depleted.')
 31            self.used_effects.append(item)
 32            return decorator(self.effects[item])
 33        else:
 34            raise AttributeError('Effect not found.')
 35
 36    def __copy__(self):
 37        new_potion = Potion({}, 0)
 38        new_potion.effects = self.effects.copy()
 39        new_potion.used_effects = self.used_effects.copy()
 40        new_potion.effects_intensity = self.effects_intensity.copy()
 41        new_potion.duration = self.duration
 42        new_potion.used = self.used
 43        return new_potion
 44
 45    def __add__(self, other):
 46        if self.used or other.used:
 47            raise TypeError('Potion is now part of something bigger than itself.')
 48
 49        new_potion = self.__copy__()
 50
 51        for key in other.effects.keys():
 52            if key in new_potion.effects_intensity.keys():
 53                new_potion.effects_intensity[key] += other.effects_intensity[key]
 54            else:
 55                new_potion.effects[key] = other.effects[key]
 56                new_potion.effects_intensity[key] = other.effects_intensity[key]
 57
 58        self.used = True
 59        other.used = True
 60
 61        return new_potion
 62
 63    def __mul__(self, num):
 64        if self.used:
 65            raise TypeError('Potion is now part of something bigger than itself.')
 66
 67        if num < 0:
 68            raise ValueError('Multiplier must be non negative.')
 69
 70        new_potion = self.__copy__()
 71        for key in new_potion.effects_intensity.keys():
 72            if 0 < num < 1:
 73                new_potion.effects_intensity[key] = Potion.custom_round(new_potion.effects_intensity[key] * num)
 74            else:
 75                new_potion.effects_intensity[key] *= num
 76
 77        self.used = True
 78
 79        return new_potion
 80
 81    def __sub__(self, other):
 82        if self.used or other.used:
 83            raise TypeError('Potion is now part of something bigger than itself.')
 84
 85        if not isinstance(other, Potion):
 86            raise TypeError('Cannot subtract non potion object.')
 87
 88        new_potion = self.__copy__()
 89        for key in other.effects.keys():
 90            if key not in self.effects.keys():
 91                raise TypeError('Cannot subtract potion with different effects.')
 92            if new_potion.effects_intensity[key] - other.effects_intensity[key] <= 0:
 93                del new_potion.effects[key]
 94                del new_potion.effects_intensity[key]
 95            else:
 96                new_potion.effects_intensity[key] -= other.effects_intensity[key]
 97
 98        self.used = True
 99        other.used = True
100
101        return new_potion
102
103    def __truediv__(self, other):
104        if self.used:
105            raise TypeError('Potion is now part of something bigger than itself.')
106
107        if not isinstance(other, int):
108            raise TypeError('Cannot divide non potion object.')
109
110        new_potion = self.__copy__()
111        for key in new_potion.effects_intensity.keys():
112            self.effects_intensity[key] = Potion.custom_round(new_potion.effects_intensity[key] / other)
113
114        self.used = True
115
116        return (new_potion,) * other
117
118    def __eq__(self, other):
119        if self.used or other.used:
120            raise TypeError('Potion is now part of something bigger than itself.')
121
122        if not isinstance(other, Potion):
123            raise TypeError('Cannot compare non potion object.')
124
125        return self.effects == other.effects and self.effects_intensity == other.effects_intensity
126
127    def __lt__(self, other):
128        if self.used or other.used:
129            raise TypeError('Potion is now part of something bigger than itself.')
130
131        if not isinstance(other, Potion):
132            raise TypeError('Cannot compare non potion object.')
133
134        lhs_sum = sum(self.effects_intensity.values())
135        rhs_sum = sum(other.effects_intensity.values())
136
137        return lhs_sum < rhs_sum
138
139    def __gt__(self, other):
140        if self.used or other.used:
141            raise TypeError('Potion is now part of something bigger than itself.')
142
143        if not isinstance(other, Potion):
144            raise TypeError('Cannot compare non potion object.')
145
146        lhs_sum = sum(self.effects_intensity.values())
147        rhs_sum = sum(other.effects_intensity.values())
148
149        return lhs_sum > rhs_sum
150
151
152class ГоспожатаПоХимия:
153    def __init__(self):
154        self.ticks = 0
155        self.applied_effects = []
156        self.used_potions = []
157
158    def apply(self, target, potion):
159        def get_moll_mass(name):
160            result = 0
161            for symbol in name:
162                result += ord(symbol)
163            return result
164
165        if potion in self.used_potions:
166            raise TypeError('Potion is depleted.')
167
168        self.used_potions.append(potion)
169
170        if potion.duration == 0:
171            return
172
173        effect_names = list(potion.effects.keys())
174        effect_names.sort(key=lambda x: get_moll_mass(x), reverse=True)
175
176        for key in effect_names:
177            try:
178                state = {attr: val for attr, val in target.__dict__.items() if not attr.startswith('__')}
179                potion.__getattr__(key)(target)
180                for attr in dir(target):
181                    if attr.startswith('__'):
182                        continue
183                    state[attr] = getattr(target, attr) - state[attr]
184                self.applied_effects.append(
185                    (potion.duration, self.ticks, target, state))
186            except TypeError:
187                raise TypeError('Potion is depleted.')
188
189    def tick(self):
190        self.ticks += 1
191
192        self.applied_effects.sort(key=lambda x: x[0])
193        for (duration, ticks_atm, target, obj_before) in self.applied_effects:
194            if self.ticks == ticks_atm + duration:
195                for attr, val in obj_before.items():
196                    cur_value = getattr(target, attr)
197                    setattr(target, attr, cur_value - val)

........E.EFEEEEEEEE
======================================================================
ERROR: test_dilution (test.TestPotionOperations)
Test dilution of a potion.
----------------------------------------------------------------------
Traceback (most recent call last):
File "/tmp/test.py", line 122, in test_dilution
self._target = copy.deepcopy(self._target)
AttributeError: 'function' object has no attribute 'deepcopy'

======================================================================
ERROR: test_purification (test.TestPotionOperations)
Test purification of a potion.
----------------------------------------------------------------------
Traceback (most recent call last):
File "/tmp/test.py", line 171, in test_purification
self._target = copy.deepcopy(self._target)
AttributeError: 'function' object has no attribute 'deepcopy'

======================================================================
ERROR: test_applying_depleted_potion (test.TestГоспожатаПоХимия)
Test applying a depleted potion or a potion that was used in a reaction.
----------------------------------------------------------------------
Traceback (most recent call last):
File "/tmp/solution.py", line 183, in apply
state[attr] = getattr(target, attr) - state[attr]
TypeError: unsupported operand type(s) for -: 'dict' and 'dict'