1def plugboard(plugboard_str, plugboard_list):
 2    conv = ''
 3    for symbol in plugboard_str:
 4        for curr_set in plugboard_list:
 5            if symbol in curr_set:
 6                conv += (curr_set - {symbol}).pop()
 7                break
 8        else:
 9            conv += symbol
10
11    return conv
12
13
14def rotor(rotor_str, rotor_dict):
15    conv = ''
16    for symbol in rotor_str:
17        conv += rotor_dict.get(symbol, symbol)
18
19    return conv
20
21
22def enigma_encrypt(plugboard_position, rotor_position):
23
24    def encrypt_decorator(encrypt_func):
25
26        def encrypt(encrypt_str):
27            new_str = plugboard(encrypt_str, plugboard_position)
28            return encrypt_func(rotor(new_str, rotor_position))
29
30        return encrypt
31        
32    return encrypt_decorator
33
34
35def enigma_decrypt(plugboard_position, rotor_position):
36
37    def decrypt_decorator(decrypt_func):
38
39        def decrypt(decrypt_str):
40            new_dict = {rotor_position[element]: element for element in rotor_position}
41            
42            new_str = rotor(decrypt_str, new_dict)
43            return decrypt_func(plugboard(new_str, plugboard_position))
44        
45        return decrypt
46    
47    return decrypt_decorator