1def plugboard(text, elements):
 2    """Returns the result from puting a text(sentence, word, letter) through the plugboard"""
 3    converted_text = ''
 4    for letter in text:
 5        for enter_code in elements:
 6            if letter in enter_code:
 7                res_letter = (enter_code - set(letter)).pop()
 8                converted_text += res_letter
 9                break
10        else:
11            converted_text += letter
12    return converted_text
13
14
15def rotor(text, codes):
16    """Returns the result from puting a text(sentence, word, letter) through the rotor"""
17    converted_text = ''
18    for letter in text:
19        if letter in codes:
20            converted_text += codes[letter]
21        else:
22            converted_text += letter
23    return converted_text
24
25
26def enigma_encrypt(plugboard_position, rotor_position):
27    """Returns decorator using plugboard and rotor functions to encrypt a text(sentence, word, letter)"""
28    def encrypt_wrapper(func):
29        def encrypt_text(text):
30            first_step_plugboard = plugboard(text, plugboard_position)
31            second_step_rotor = rotor(first_step_plugboard, rotor_position)
32            return func(second_step_rotor)
33        return encrypt_text
34    return encrypt_wrapper
35
36
37def enigma_decrypt(plugboard_position, rotor_position):
38    """Returns decorator using plugboard and rotor functions to decrypt a text(sentence, word, letter)"""
39    def decrypt_wrapper(func):
40        def decrypt_text(text):
41            rotor_position_reverse = {value: key for key, value in rotor_position.items()}
42            first_step_rotor = rotor(text, rotor_position_reverse)
43            second_step_plugboard = plugboard(first_step_rotor, plugboard_position)
44            return func(second_step_plugboard)
45        return decrypt_text
46    return decrypt_wrapper