1"""Solution NOT based on the fact that there are only lowercase letters"""
 2
 3
 4def replace_text(text, dictionary):
 5    return "".join([dictionary.get(char, char) for char in text])
 6
 7
 8def invert_dictionary(dictionary):
 9    return {value: key for key, value in dictionary.items()}
10
11
12def plugboard(message, plugboard_position):
13    plugboard_dictionary = dict(plugboard_position)
14    plugboard_dictionary |= invert_dictionary(plugboard_dictionary)
15    return replace_text(message, plugboard_dictionary)
16
17
18def rotor(message, rotor_position):
19    return replace_text(message, rotor_position)
20
21
22def enigma_encrypt(plugboard_position, rotor_position):
23    def decorator(function):
24        def wrapper(text):
25            text_plugboard = plugboard(text, plugboard_position)
26            text_rotor = rotor(text_plugboard, rotor_position)
27            return function(text_rotor)
28
29        return wrapper
30
31    return decorator
32
33
34def enigma_decrypt(plugboard_position, rotor_position):
35    def decorator(function):
36        def wrapper(text):
37            inverted_rotor_position = invert_dictionary(rotor_position)
38            text_rotor = rotor(text, inverted_rotor_position)
39            text_plugboard = plugboard(text_rotor, plugboard_position)
40            return function(text_plugboard)
41
42        return wrapper
43
44    return decorator