1def change_letter(letter, converting_list):
 2    for single_set in converting_list:
 3        if letter in single_set:
 4            return "".join(single_set.difference({letter}))
 5    return letter
 6
 7
 8def plugboard(text_to_convert, converting_set):
 9    converted_text = [change_letter(letter, converting_set) for letter in text_to_convert]
10    return "".join(converted_text)
11
12
13def rotor(text_to_convert, converting_dict):
14    converted_text = ""
15    for symbol in text_to_convert:
16        converted_text += converting_dict.get(symbol, ' ')
17    return converted_text
18
19
20def enigma_encrypt(plugboard_position, rotor_position):
21    def decorator(func):
22        def wrapped_func(text_to_encrypt):
23            return func(rotor(plugboard(text_to_encrypt, plugboard_position), rotor_position))
24        return wrapped_func
25    return decorator
26
27
28def enigma_decrypt(plugboard_position, rotor_position):
29    def decorator(func):
30        def wrapped_func(text_to_decrypt):
31            reverse_rotor_position = {v: k for k, v in rotor_position.items()}
32            return func(plugboard(rotor(text_to_decrypt, reverse_rotor_position), plugboard_position))
33        return wrapped_func
34    return decorator