1import string
  2from inspect import getmembers, isfunction, ismethod
  3
  4import secret
  5
  6
  7class FunctionTester:
  8    def _is_specific_exception_function(
  9        self, function, exception_type, exception_message=None
 10    ):
 11        try:
 12            function()
 13        except exception_type as exception:
 14            is_correct_type = type(exception) is exception_type
 15            if exception_message:
 16                return is_correct_type and str(exception) == exception_message
 17
 18            return is_correct_type
 19        except BaseException:
 20            return False
 21
 22        return False
 23
 24    @staticmethod
 25    def _exception_guard(function):
 26        def wrapper(*args, **kwargs):
 27            try:
 28                return function(*args, **kwargs)
 29            except BaseException:
 30                return False
 31
 32        return wrapper
 33
 34    @_exception_guard
 35    def _is_same_as_function(self, function, stub_function, test_cases):
 36        for args in test_cases:
 37            if function(*args) != stub_function(*args):
 38                return False
 39        return True
 40
 41
 42class ChallengeFunctionTester(FunctionTester):
 43    def is_type_error_function(self, function):
 44        type_error_message = "Опаааааа, тука има нещо нередно."
 45        return self._is_specific_exception_function(
 46            function, TypeError, type_error_message
 47        )
 48
 49    def is_base_error_function(self, function):
 50        return self._is_specific_exception_function(function, BaseException)
 51
 52    def is_int_function(self, function):
 53        def int_function_stub(x):
 54            return x**2 if x % 2 == 0 else 0
 55
 56        int_test_range = 100
 57
 58        test_cases = [
 59            (el,) for el in range(-int_test_range, int_test_range + 1)
 60        ]
 61
 62        return self._is_same_as_function(
 63            function, int_function_stub, test_cases
 64        )
 65
 66    def is_string_function(self, function):
 67        def string_function_stub(left, right):
 68            return left + right
 69
 70        string_test_cases = [
 71            ("a", "b"),
 72            ("abc", "123"),
 73            ("XxX", ""),
 74            ("", "YyY"),
 75            ("", ""),
 76        ]
 77        return self._is_same_as_function(
 78            function, string_function_stub, string_test_cases
 79        )
 80
 81    def is_static_function(self, function):
 82        """Not ideal"""
 83        return not ismethod(function) and function not in [
 84            function for _, function in getmembers(secret)
 85        ]
 86
 87    def is_function_name_ok(self, function):
 88        allowed_characters = list(string.ascii_uppercase + string.digits)
 89        return function.__name__ in allowed_characters
 90
 91    def is_interesting(self, function):
 92        if not self.is_function_name_ok(function):
 93            return False
 94
 95        checkers = [
 96            self.is_type_error_function,
 97            self.is_base_error_function,
 98            self.is_int_function,
 99            self.is_string_function,
100            self.is_static_function,
101        ]
102
103        return any(checker(function) for checker in checkers)
104
105
106def list_all_functions(module_name, member_keyword):
107    stack = [module_name]
108    functions = []
109
110    while stack:
111        current = stack.pop()
112
113        members = getmembers(current)
114        functions.extend([value for _, value in members if isfunction(value)])
115
116        stack.extend(
117            [value for name, value in members if member_keyword in name]
118        )
119
120    return functions
121
122
123def get_interesting_functions():
124    member_keyword = "clue"
125
126    functions = list_all_functions(secret, member_keyword)
127    function_tester = ChallengeFunctionTester()
128
129    return [
130        function
131        for function in functions
132        if function_tester.is_interesting(function)
133    ]
134
135
136def methodify():
137    faculty_number = "FN4MI0600097"
138    interesting_functions = set(get_interesting_functions())
139
140    functions_by_name = {
141        function.__name__: function for function in interesting_functions
142    }
143    fn_functions = [functions_by_name[letter] for letter in faculty_number]
144
145    return tuple(fn_functions)