1import re
  2
  3class Kid(type):
  4    _all_kids = {}
  5    _kid_indexes = {}
  6    _kid_list = []
  7    _next_index = 0
  8
  9    def __new__(mcs, name, bases, attrs):
 10        cls = super().__new__(mcs, name, bases, attrs)
 11
 12        if "__call__" not in attrs:
 13            has_call = any(hasattr(base, "__call__") for base in bases)
 14            if not has_call:
 15                raise NotImplementedError("Bruhhh")
 16
 17        def decorate_with_exception_handling(method):
 18            def wrapper(self, *args, **kwargs):
 19                kid_id = id(self)
 20                try:
 21                    return method(self, *args, **kwargs)
 22                except Exception:
 23                    santa = Santa()
 24                    kid_index = Kid._kid_indexes[kid_id]
 25                    santa._mark_naughty(kid_index)  
 26                    raise
 27            return wrapper
 28
 29        for attr_name, attr_value in attrs.items():
 30            if callable(attr_value) and not attr_name.startswith('_'):
 31                setattr(cls, attr_name, decorate_with_exception_handling(attr_value))
 32
 33        return cls
 34
 35    def __call__(cls, *args, **kwargs):
 36        obj = super().__call__(*args, **kwargs)
 37        kid_id = id(obj)
 38        Kid._all_kids[kid_id] = obj
 39        kid_index = Kid._next_index
 40        Kid._kid_list.append(obj)
 41        Kid._kid_indexes[kid_id] = kid_index
 42        Kid._next_index += 1
 43
 44        Santa()._register_kid(obj)  # Register the kid with Santa
 45
 46        return obj
 47
 48
 49class Santa:
 50    _instance = None
 51    AGE_LIMIT = 5  
 52
 53    def __new__(cls):
 54        if cls._instance is None:
 55            obj = super().__new__(cls)
 56            cls._instance = obj
 57        return cls._instance
 58
 59    def __init__(self):
 60        if not hasattr(self, 'initialized'):
 61            self.xmas_count = 0
 62            self.kid_birth_xmas = {}  # Tracks the Christmas count when the kid was registered
 63            self.last_requests = {}  # Stores the last gift requested by each kid
 64            self.requests_since_last_xmas = []  # List of all requests since last Christmas
 65            self.kid_flags_mask = 0  # Bitmask to track kid statuses
 66            self.initialized = True
 67
 68    def _mark_naughty(self, kid_index):
 69        self.kid_flags_mask |= (1 << (kid_index * 3))  
 70
 71    def _is_naughty(self, kid_index):
 72        return (self.kid_flags_mask & (1 << (kid_index * 3))) != 0
 73
 74    def _mark_requested(self, kid_index):
 75        self.kid_flags_mask |= (1 << (kid_index * 3 + 1))  
 76
 77    def _is_requested(self, kid_index):
 78        return (self.kid_flags_mask & (1 << (kid_index * 3 + 1))) != 0
 79
 80    def _set_age_flag(self, kid_index, is_over_age):
 81        if is_over_age:
 82            self.kid_flags_mask &= ~(1 << (kid_index * 3 + 2))  
 83        else:
 84            self.kid_flags_mask |= (1 << (kid_index * 3 + 2))  
 85
 86    def _is_over_age(self, kid_index):
 87        return not ((self.kid_flags_mask & (1 << (kid_index * 3 + 2))) != 0)
 88
 89    def _register_kid(self, kid):
 90        kid_id = id(kid)
 91        if kid_id not in self.kid_birth_xmas:
 92            # Register the kid's birth Christmas
 93            self.kid_birth_xmas[kid_id] = self.xmas_count
 94            kid_index = Kid._kid_indexes[kid_id]
 95            self._set_age_flag(kid_index, is_over_age=False)
 96
 97    def __call__(self, kid, input_string):
 98        gift = self._get_gift(input_string)
 99        kid_id = id(kid)
100        if kid_id not in self.kid_birth_xmas:
101            self.kid_birth_xmas[kid_id] = self.xmas_count  # Register the kid if not already registered
102
103        self.requests_since_last_xmas.append((kid_id, gift))  # Add the request to the lists
104        self.last_requests[kid_id] = gift
105        kid_index = Kid._kid_indexes[kid_id]
106        self._mark_requested(kid_index) 
107
108    def __matmul__(self, input_string):
109        gift = self._get_gift(input_string)
110        kid_id = self._get_kid_id_from_letter(input_string)
111        kid = Kid._all_kids[kid_id]
112        if kid_id not in self.kid_birth_xmas:
113            self.kid_birth_xmas[kid_id] = self.xmas_count  # Register the kid if not already registered
114
115        self.requests_since_last_xmas.append((kid_id, gift))  # Add the request to the lists
116        self.last_requests[kid_id] = gift
117        kid_index = Kid._kid_indexes[kid_id]
118        self._mark_requested(kid_index)  
119
120        return self
121
122    def _get_gift(self, input_string):
123        pattern = r'(["\'])([A-Za-z0-9 ]+)\1'
124        match = re.search(pattern, input_string)
125        if not match:
126            raise ValueError("Error")
127        return match.group(2)
128
129    def _get_kid_id_from_letter(self, input_string):
130        match = re.search(r'^\s*(\d+)\s*$', input_string, re.MULTILINE)
131        if match:
132            kid_id = int(match.group(1))
133            if kid_id in Kid._all_kids:
134                return kid_id
135        raise ValueError("Error")
136
137    def __iter__(self):
138        return iter(self.last_requests.values())
139
140    def xmas(self):
141        if not self.requests_since_last_xmas:
142            self._reset()
143            return
144
145        chosen_most_wanted = self._most_wanted_gift()
146        self._deliver(chosen_most_wanted)
147        self._reset()
148
149    def _most_wanted_gift(self):
150        if not self.last_requests:
151            return None
152
153        gift_counts = {}
154        for gift in self.last_requests.values():
155            gift_counts[gift] = gift_counts.get(gift, 0) + 1
156
157        if not gift_counts:
158            return None
159
160        max_count = max(gift_counts.values())
161        most_wanted = [g for g, c in gift_counts.items() if c == max_count]
162        return most_wanted[0] if most_wanted else None 
163
164    def _deliver(self, chosen_most_wanted):
165        for kid_id, kid in Kid._all_kids.items():
166            birth = self.kid_birth_xmas.get(kid_id, self.xmas_count)
167            age = self.xmas_count - birth
168            kid_index = Kid._kid_indexes[kid_id]
169
170            if age >= self.AGE_LIMIT:
171                self._set_age_flag(kid_index, is_over_age=True)  # Mark the kid as over age and skip delivery
172                continue
173            else:
174                self._set_age_flag(kid_index, is_over_age=False)  # Reset the age flag if kid is not over age
175
176            naughty = self._is_naughty(kid_index)  
177
178            if kid_id in self.last_requests:
179                gift = self.last_requests[kid_id]
180            else:
181                gift = chosen_most_wanted
182                if gift is None:
183                    continue
184
185            if naughty:
186                gift = "coal"
187
188            kid(gift)  
189
190    def _reset(self):
191        # Reset the state for the next Christmas
192        self.xmas_count += 1
193        self.requests_since_last_xmas.clear()
194        self.last_requests.clear()
195        self.kid_flags_mask = 0