1import re
  2from collections import defaultdict
  3
  4
  5def extract_wish(text):
  6    match = re.search(r'["\']([a-zA-Z0-9 ]+)["\']', text)
  7    return match.group(1)
  8
  9
 10def extract_kid_id(text):
 11    match = re.search(r'\b(\d+)\b', text)
 12    return int(match.group(1))
 13
 14
 15def track_exceptions(method):  # as in the lection
 16    def wrapper(self, *args, **kwargs):
 17        try:
 18            return method(self, *args, **kwargs)
 19        except Exception:
 20            self._threw_exception = True
 21            raise
 22    return wrapper
 23
 24
 25def is_bad_kid(kid):
 26    return kid._threw_exception
 27
 28
 29class Santa(object):
 30    def __new__(cls):
 31        if not hasattr(cls, 'instance'):
 32            cls.instance = super(Santa, cls).__new__(cls)
 33        return cls.instance
 34
 35    def __init__(self):
 36        self.wishes = {}
 37        self.kids_ages = {}
 38
 39    def __call__(self, kid, talk):
 40        wish = extract_wish(talk)
 41        self.wishes[kid] = wish
 42
 43    def __matmul__(self, letter):
 44        kid_id = extract_kid_id(letter)
 45        kid = next((child for child in Kid.get_all_kids() if id(child) == kid_id), None)
 46        wish = extract_wish(letter)
 47        self.wishes[kid] = wish
 48
 49    def __iter__(self):
 50        return iter(self.wishes.values())     # is this right
 51
 52    def xmas(self):
 53        for every_kid in Kid.get_all_kids():
 54            if every_kid not in self.kids_ages:
 55                self.kids_ages[every_kid] = 1
 56            else:
 57                self.kids_ages[every_kid] += 1
 58
 59        if not self.wishes:
 60            return
 61
 62        wish_counts = defaultdict(int)
 63        for wish in self.wishes.values():
 64            wish_counts[wish] += 1
 65        most_wished = max(wish_counts.items(), key=lambda x: x[1])[0]  # is this actually right
 66
 67        for every_kid in Kid.get_all_kids():
 68            if self.kids_ages[every_kid] > 5:
 69                continue
 70
 71            if is_bad_kid(every_kid):
 72                every_kid("coal")
 73
 74            else:
 75                wish = self.wishes.get(every_kid)
 76                if wish:
 77                    every_kid(wish)
 78                else:
 79                    every_kid(most_wished)
 80
 81            every_kid._threw_exception = False
 82
 83        self.wishes.clear()
 84
 85
 86class Kid(type):
 87    _all_kids = set()
 88
 89    def __new__(cls, name, bases, dct):
 90        for attr_name, attr_value in dct.items():
 91            if callable(attr_value) and not attr_name.startswith("_"):
 92                dct[attr_name] = track_exceptions(attr_value)
 93        return super().__new__(cls, name, bases, dct)
 94
 95    def __call__(cls, *args, **kwargs):
 96        instance = super().__call__(*args, **kwargs)
 97        Kid._all_kids.add(instance)
 98        instance._threw_exception = False
 99        if not callable(instance):
100            raise NotImplementedError("Стига де! Това все пак са деца! Трябва да можем да им викаме!")
101        return instance
102
103    @staticmethod
104    def get_all_kids():
105        return Kid._all_kids