1import re
  2from collections import OrderedDict, Counter
  3
  4
  5class Singleton(type):
  6    def __new__(cls, name, bases, attrs):
  7        def singleton_new(cls):
  8            if not hasattr(cls, 'instance'):
  9                cls.instance = object.__new__(cls)
 10            return cls.instance
 11        attrs['__new__'] = singleton_new
 12        return super().__new__(cls, name, bases, attrs)
 13
 14
 15class Santa(metaclass=Singleton):
 16    """HOHOHO."""
 17
 18    kid_present = OrderedDict()
 19
 20    def __init__(self):
 21        self.presents_iter = None
 22
 23    def __call__(self, kid, wish: str):
 24        extracted_info = Wish(wish)
 25        self.kid_present[kid] = extracted_info.present
 26
 27    def __matmul__(self, letter: str):
 28        extracted_info = Wish(letter)
 29        self.kid_present[Kid.find_kid(extracted_info.kid_id)] = extracted_info.present
 30
 31    def __iter__(self):
 32        return iter(self.kid_present.values())
 33
 34    def xmas(self):
 35        Kid.increment_age_of_all_kids()
 36
 37        # if there are no wishes return
 38        if not self.kid_present:
 39            return
 40
 41        counted = Counter(self.__iter__())
 42        most_common_present = counted.most_common(1)[0][0]
 43
 44        for kid in Kid.kids_created:
 45            if kid.age > 5:
 46                continue
 47
 48            if kid.naughty:
 49                kid('coal')
 50                kid.naughty = False
 51                continue
 52
 53            present = self.kid_present.get(kid, most_common_present)
 54            kid(present)
 55
 56            # remove kid from wishlist
 57            if kid in self.kid_present:
 58                self.kid_present.pop(kid)
 59
 60
 61class Wish:
 62    """Decipher wish and extract only relevant info."""
 63
 64    def __init__(self, wish: str):
 65        self.present = None
 66        self.kid_id = None
 67
 68        # find present
 69        present_pattern = r'([\'\"])(?P<present>(\s*\w+\s*)+)(\1)'
 70        match = re.search(present_pattern, wish)
 71        if match:
 72            self.present = match.group('present')
 73
 74        # find id if there is one
 75        id_pattern = r'\n\s*(?P<kid_id>\d+)\s*(\n|$)'
 76        match = re.search(id_pattern, wish)
 77        if match:
 78            self.kid_id = int(match.group('kid_id'))
 79
 80
 81class Kid(type):
 82
 83    kids_created = [] # who knows how exactly
 84
 85    def __new__(cls, name, bases, attrs):
 86        # add naughty flag and age to all kids
 87        attrs['naughty'] = False
 88        attrs['age'] = 0
 89
 90        # check if kid is callable
 91        if not attrs.get('__call__', None):
 92            raise NotImplementedError('kur cici kurviiiiiiiiiiii - съобщение от човек не в курса')
 93
 94        # decorate __new__ if it exists in child class and create it if it doesn't
 95        try:
 96            attrs['__new__'] = cls.new_kid_decorator(attrs['__new__'])
 97        except Exception:
 98            attrs['__new__'] = cls.new_kid
 99
100        # wrap all public methods in decorator to check for naughtiness
101        for name, attr in attrs.items():
102            if callable(attr) and not name.startswith('__'):
103                attrs[name] = cls.naughty_decorator(attr)
104
105        return super().__new__(cls, name, bases, attrs)
106
107    @classmethod
108    def find_kid(cls, kid_id: int):
109        for kid in cls.kids_created:
110            if id(kid) == kid_id:
111                return kid
112
113    @staticmethod
114    def new_kid_decorator(func):
115        def decorated(self, *args, **kwargs):
116            Kid.kids_created.append(self)
117            return func(self, *args, **kwargs)
118        return decorated
119
120    @staticmethod
121    def new_kid(cls_kid):
122        instance = object.__new__(cls_kid)
123        Kid.kids_created.append(instance)
124        return instance
125
126    @staticmethod
127    def naughty_decorator(func):
128        def decorated(self, *args, **kwargs):
129            try:
130                res = func(self, *args, **kwargs)
131            except Exception:
132                self.naughty = True
133                raise
134            return res
135        return decorated
136
137    @classmethod
138    def increment_age_of_all_kids(cls):
139        for kid in cls.kids_created:
140            kid.age += 1