1def plugboard(text: str, plugboard_positions: list):
 2    """Puts the input through the plugboard."""
 3    new_text = ''
 4    for letter in text:
 5        for plugboard_position in plugboard_positions:
 6            plugboard_position_copy = plugboard_position.copy()
 7            if letter in plugboard_position_copy:
 8                plugboard_position_copy.remove(letter)
 9                new_text += plugboard_position_copy.pop()
10                break
11        else:
12            new_text += letter
13    return new_text
14
15
16def rotor(text: str, rotor_positions: dict):
17    """Puts the input through the rotor."""
18    new_text = ''
19    for letter in text:
20        if letter not in rotor_positions.values():
21            new_text += letter
22            continue
23        new_text += rotor_positions[letter]
24    return new_text
25
26
27def enigma_encrypt(plugboard_position: list, rotor_position: dict):
28    """Returns decorator that executes the steps to encrypt the input."""
29    def encrypt(func):
30        def encrypt_string(text: str):
31            text = plugboard(text, plugboard_position)
32            text = rotor(text, rotor_position)
33            return func(text)
34        return encrypt_string
35    return encrypt
36
37
38def enigma_decrypt(plugboard_position: list, rotor_position: dict):
39    """Returns decorator that executes the steps to decrypt the input."""
40    def decrypt(func):
41        def decrypt_string(text: str):
42            new_rotor_position = {v: k for k, v in rotor_position.items()}
43            text = rotor(text, new_rotor_position)
44            text = plugboard(text, plugboard_position)
45            return func(text)
46        return decrypt_string
47    return decrypt