1def plugboard(text, plugboard_position):
 2    letter_mapping = {}
 3    for a, b in plugboard_position:
 4        letter_mapping[a], letter_mapping[b] = b, a
 5    return ''.join(letter_mapping.get(letter, letter) for letter in text)
 6
 7
 8def rotor(text, rotor_position):
 9    return ''.join(rotor_position.get(letter, letter) for letter in text)
10
11
12def reversed_rotor(rotor_position):
13    return {value: key for key, value in rotor_position.items()}
14
15
16def enigma_encrypt(plugboard_position=[], rotor_position={}):
17    def decorator(func):
18        def wrapper(text):
19            encrypted_text = plugboard(text, plugboard_position)
20            encrypted_text = rotor(encrypted_text, rotor_position)
21            return func(encrypted_text)
22        return wrapper
23    return decorator
24
25
26def enigma_decrypt(plugboard_position=[], rotor_position={}):
27    reversed_rotor_mapping = reversed_rotor(rotor_position)
28
29    def decorator(func):
30        def wrapper(text):
31            decrypted_text = rotor(text, reversed_rotor_mapping)
32            decrypted_text = plugboard(decrypted_text, plugboard_position)
33            return func(decrypted_text)
34        return wrapper
35    return decorator