1import re
  2
  3ID_REGEX = r'^[\s]*((\d)+)[\s]*$'
  4PRESENT_REGEX = r'[\'|\"]([a-zA-Z0-9 ]+)[\'|\"]'
  5
  6
  7def register_error_raising(func):
  8    def decorate(self, *args, **kwargs):
  9        try:
 10            func(self, *args, **kwargs)
 11        except Exception as e:
 12            metadata = Kid.kids_dictionary[id(self)]
 13            metadata.naughty = True
 14            raise e
 15    return decorate
 16
 17
 18def register_initialized_object(func):
 19    def decorate(self, *args, **kwargs):
 20        func(self, *args, **kwargs)
 21        Kid.kids_dictionary.__setitem__(id(self), KidMetadata(self))
 22    return decorate
 23
 24
 25class KidMetadata:
 26    def __init__(self, kid):
 27        self.ages_left = 5
 28        self.reference = kid
 29        self.naughty = False
 30
 31    def decrement(self):
 32        self.ages_left -= 1
 33
 34    def is_old(self):
 35        return self.ages_left <= 0
 36
 37    def hash(self):
 38        return hash(self.reference)
 39
 40
 41class Kid(type):
 42
 43    kids_dictionary = {}
 44
 45    def __new__(cls, *args, **kwargs):
 46        constructed_class = type.__new__(cls, *args, **kwargs)
 47        constructed_class.__init__ = register_initialized_object(getattr(constructed_class, "__init__"))
 48        if not callable(constructed_class):
 49            raise NotImplementedError("и аз съм от сдс")
 50
 51        for attr in constructed_class.__dict__:
 52            if callable(getattr(constructed_class, attr)):
 53                setattr(constructed_class, attr, register_error_raising(getattr(constructed_class, attr)))
 54
 55        return constructed_class
 56
 57
 58def match_regex(regex, string):
 59    matches = re.search(regex, string, re.MULTILINE)
 60    return matches.groups()[0]
 61
 62
 63def get_most_popular_gift(presents_list):
 64    if not len(presents_list):
 65        return None
 66    return max(presents_list, key=presents_list.count)
 67
 68
 69def iterate_through_requests(santa_reference):
 70    updated_kids = {child: metadata for (child, metadata) in Kid.kids_dictionary.items() if metadata.ages_left >= 0}
 71    Kid.kids_dictionary = updated_kids
 72
 73    most_popular_gift = get_most_popular_gift(list(santa_reference.presents_dictionary.values()))
 74    if most_popular_gift is None:
 75        return
 76
 77    kids_with_present_wishes = list(set(santa_reference.presents_dictionary.keys()).intersection(set(Kid.kids_dictionary.keys())))
 78    kids_without_present_wishes = Kid.kids_dictionary.keys() - kids_with_present_wishes
 79
 80    for kid in kids_with_present_wishes:
 81        target_kid_metadata = Kid.kids_dictionary[kid]
 82        target_kid_metadata.decrement()
 83        if target_kid_metadata.naughty:
 84            target_kid_metadata.naughty = False
 85            target_kid_metadata.reference("coal")
 86        else:
 87            target_kid_metadata.reference(santa_reference.presents_dictionary[kid])
 88            santa_reference.presents_dictionary.pop(kid)
 89
 90    for kid in kids_without_present_wishes:
 91        target_kid_metadata = Kid.kids_dictionary[kid]
 92        target_kid_metadata.decrement()
 93        if target_kid_metadata.naughty:
 94            target_kid_metadata.naughty = False
 95            target_kid_metadata.reference("coal")
 96        else:
 97            target_kid_metadata.reference(most_popular_gift)
 98
 99
100class Santa(metaclass=type):
101
102    santa_instance = None
103    presents_dictionary = {}
104
105    def __new__(cls):
106        if cls.santa_instance is None:
107            cls.santa_instance = object.__new__(cls)
108
109        return cls.santa_instance
110
111    def __matmul__(self, other):
112
113        if isinstance(other, str):
114            child_id = match_regex(ID_REGEX, other)
115            present_name = match_regex(PRESENT_REGEX, other)
116            self.presents_dictionary[child_id] = present_name
117        else:
118            raise TypeError("Unsupported type")
119
120    def __call__(self, kid, present):
121
122        present = match_regex(PRESENT_REGEX, present)
123        self.presents_dictionary[id(kid)] = present
124
125    def __iter__(self):
126        return iter(self.presents_dictionary.values())
127
128    def xmas(self):
129        iterate_through_requests(self)