1def translation(letter, coupled_letters):
 2    """ Finding the coupled letter of the input letter, if such exists. """
 3
 4    for (first, second) in coupled_letters:
 5        if letter == first:
 6            return second
 7        elif letter == second:
 8            return first
 9    return letter
10
11
12def plugboard(sentence, coupled_letters):
13    """ Plugboard usage. """
14
15    list_sentence = [translation(char, coupled_letters) for char in sentence]
16    return "".join(list_sentence)
17
18
19def rotor(sentence, dict_alphabet):
20    """ Rotor usage. """
21
22    new_sentence = "".join(map(lambda letter: dict_alphabet[letter] if letter.isalpha() else letter, sentence))
23    return new_sentence
24
25
26def enigma_encrypt(plugboard_position, rotor_position):
27    def decorator_encrypt(func):
28        def encrypt(sentence):
29
30            sentence = plugboard(sentence, plugboard_position)
31            sentence = rotor(sentence, rotor_position)
32
33            return func(sentence)
34        return encrypt
35    return decorator_encrypt
36
37
38def enigma_decrypt(plugboard_position, rotor_position):
39    def decorator_decrypt(func):
40        def decrypt(sentence):
41
42            new_rotor_position = dict(zip(rotor_position.values(), rotor_position.keys()))
43            sentence = rotor(sentence, new_rotor_position)
44            sentence = plugboard(sentence, plugboard_position)
45
46            return func(sentence)
47        return decrypt
48    return decorator_decrypt