1def convert_to_dict(plugboard_position):
 2    dictionary = {}
 3    for i in plugboard_position:
 4        for j in i:
 5            dictionary[j] = (i - {j}).pop()
 6    return dictionary
 7
 8
 9def convert_to_dict_decorator(func):
10    def wrapper(text, plugboard_position):
11        return func(text, convert_to_dict(plugboard_position))
12    return wrapper
13
14
15def encrypt_text_with_dict(text, dictionary):
16    return "".join(map(lambda key: dictionary.get(key, key), text))
17
18
19def reverse_dict(dictionary):
20    return {value: key for key, value in dictionary.items()}
21
22
23def reverse_dict_decorator(func):
24    def wrapper(text, dictionary):
25        return func(text, reverse_dict(dictionary))
26    return wrapper
27
28
29@convert_to_dict_decorator
30def plugboard(text, plugboard_position):
31    return encrypt_text_with_dict(text, plugboard_position)
32
33
34def rotor(text, rotor_position):
35    return encrypt_text_with_dict(text, rotor_position)
36
37
38@reverse_dict_decorator
39def rotor_decrypt(text, rotor_position):
40    return encrypt_text_with_dict(text, rotor_position)
41
42
43def enigma_encrypt(plugboard_position, rotor_position):
44    def decorator(func):
45        def wrapper(text):
46            encrypted_text = rotor(
47                plugboard(text, plugboard_position), rotor_position)
48            return func(encrypted_text)
49        return wrapper
50    return decorator
51
52
53def enigma_decrypt(plugboard_position, rotor_position):
54    def decorator(func):
55        def wrapper(text):
56            decrypted_text = plugboard(rotor_decrypt(
57                text, rotor_position), plugboard_position)
58            return func(decrypted_text)
59        return wrapper
60    return decorator