1def plugboard(uncoded: str, plugboard_pos: list):
 2    return encode_with_dict(uncoded, convert_to_dictionary(plugboard_pos))
 3
 4
 5def rotor(uncoded: str, from_to_dict: dict):
 6    return encode_with_dict(uncoded, from_to_dict)
 7
 8
 9def enigma_encrypt(plugboard_position: list, rotor_position: dict):
10    def encrypting_decorator(func):
11        def encrypt_and_run_func(text):
12            text = rotor(plugboard(text, plugboard_position), rotor_position)
13            return func(text)
14        return encrypt_and_run_func
15    return encrypting_decorator
16
17
18def enigma_decrypt(plugboard_position: list, rotor_position: dict):
19    def encrypting_decorator(func):
20        def decrypt_and_run_func(text):
21            reverse_rotor_position = rotate_rotor_positions(rotor_position)
22            text = plugboard(rotor(text, reverse_rotor_position), plugboard_position)
23            return func(text)
24        return decrypt_and_run_func
25    return encrypting_decorator
26
27
28def encode_with_dict(uncoded: str, from_to_dict: dict):
29    encoded = ''
30
31    for char in uncoded:
32        if char in from_to_dict:
33            encoded += from_to_dict[char]
34        else:
35            encoded += char
36    return encoded
37
38
39def convert_to_dictionary(plugboard_pos: list):
40    from_to_dictionary = {}
41
42    for from_to_set in plugboard_pos:
43        pair = list(from_to_set)
44        from_to_dictionary[pair[0]] = pair[1]
45        from_to_dictionary[pair[1]] = pair[0]
46    return from_to_dictionary
47
48
49def rotate_rotor_positions(rotor_positions: dict):
50    new_rotation = {}
51
52    for key, value in rotor_positions.items():
53        new_rotation[value] = key
54    return new_rotation