1import copy
 2
 3def plugboard(the_text, matches):
 4    from_set_to_str = ""
 5    ready = ""
 6
 7    for character in the_text:
 8        new_matches = ""
 9        for one_pair in  matches:
10            if character in one_pair:
11               new_matches = ''.join(one_pair)
12               from_set_to_str = copy.deepcopy(one_pair)
13               from_set_to_str.remove(character) 
14               res=''.join(from_set_to_str)
15               ready += res
16               break
17        if character not in new_matches:
18            ready += character
19    return ready
20
21def rotor(the_text, matches):
22    ready=""
23
24    for character in the_text:
25        if character == " ":
26             ready += " "
27        if character in matches: 
28            res = matches.get(character)
29            ready += res
30    return ready
31
32def enigma_encrypt(plugboard_position, rotor_position):
33    def encryptor(func):
34        def encrypt_print(txt):
35            first = plugboard(txt, plugboard_position)
36            result = rotor(first, rotor_position)
37            return func(result)
38        return encrypt_print
39    return encryptor
40
41def enigma_decrypt(plugboard_position, rotor_position):
42    def decryptor(func):
43        def decrypt_print(txt):
44            res = {v: k for k, v in rotor_position.items()}
45            first = rotor(txt, res)
46            result = plugboard(first, plugboard_position)
47            return func(result)
48        return decrypt_print
49    return decryptor
50
51
52
53
54
55plugboard_position = [{'a', 'c'}, {'t', 'z'}]
56rotor_position = {'v': 'd', 'd': 'v', 'y': 'u', 'n': 'n', 'i': 'w', 'z': 'p',
57                  's': 'e', 'x': 's', 'h': 'f', 'b': 'x', 'u': 'c', 'p': 'q',
58                  'r': 'g', 'q': 'j', 'e': 't', 'l': 'y', 'o': 'z', 'g': 'o',
59                  'k': 'b', 't': 'h', 'j': 'm', 'a': 'a', 'w': 'i', 'f': 'l',
60                  'm': 'r', 'c': 'k'}
61
62rotor('enigma', rotor_position) # tnwora
63plugboard('enigma', plugboard_position) # enigmc
64
65encryptor = enigma_encrypt(plugboard_position=plugboard_position,
66                           rotor_position=rotor_position)
67decryptor = enigma_decrypt(plugboard_position=plugboard_position,
68                           rotor_position=rotor_position)
69
70encrypt_print = encryptor(print)
71decrypt_print = decryptor(print)
72
73encrypt_print('enigma') # tnwork
74decrypt_print('tnwork') # enigma