1def plugboard(word, plugboard_position):
  2    transformed_word = []
  3    for letter in word:
  4        for pair_of_letters in plugboard_position:
  5            if letter in pair_of_letters:
  6                transformed_word.append((pair_of_letters - {letter}).pop())
  7                break
  8        else:
  9            transformed_word.append(letter)
 10
 11    return "".join(transformed_word)
 12
 13
 14def rotor(word, rotor_position):
 15    transformed_word = []
 16    for letter in word:
 17        value = rotor_position.get(letter, letter)
 18        transformed_word.append(value)
 19    return "".join(transformed_word)
 20
 21
 22def enigma_encrypt(plugboard_position, rotor_position):
 23    def decorator(func):
 24        def wrapper(word):
 25            plugboarded_word = plugboard(word, plugboard_position)
 26            encrypted_word = rotor(plugboarded_word, rotor_position)
 27            return func(encrypted_word)
 28
 29        return wrapper
 30
 31    return decorator
 32
 33
 34def enigma_decrypt(plugboard_position, rotor_position):
 35    def decorator(func):
 36        def wrapper(word):
 37            swapped_rotor_position = {v: k for k, v in rotor_position.items()}
 38            encrypted_word = rotor(word, swapped_rotor_position)
 39            plugboarded_word = plugboard(encrypted_word, plugboard_position)
 40            return func(plugboarded_word)
 41
 42        return wrapper
 43
 44    return decorator
 45
 46
 47def crypt_behaivior(additional_func):
 48    def decorator(logic_func):
 49        def wrapper(*args):
 50            additional_func(logic_func(*args))
 51
 52        return wrapper
 53
 54    return decorator
 55
 56
 57def encrypt(word):
 58    return word
 59
 60
 61def decrypt(word):
 62    return word
 63
 64
 65plugboard_position = [{"a", "c"}, {"t", "z"}, {"k", "l"}, {"d", "y"}, {"s", "j"}]
 66rotor_position = {
 67    "v": "d",
 68    "d": "v",
 69    "y": "u",
 70    "n": "n",
 71    "i": "w",
 72    "z": "p",
 73    "s": "e",
 74    "x": "s",
 75    "h": "f",
 76    "b": "x",
 77    "u": "c",
 78    "p": "q",
 79    "r": "g",
 80    "q": "j",
 81    "e": "t",
 82    "l": "y",
 83    "o": "z",
 84    "g": "o",
 85    "k": "b",
 86    "t": "h",
 87    "j": "m",
 88    "a": "a",
 89    "w": "i",
 90    "f": "l",
 91    "m": "r",
 92    "c": "k",
 93}
 94
 95
 96encrypt = enigma_encrypt(plugboard_position, rotor_position)(encrypt)
 97decrypt = enigma_decrypt(plugboard_position, rotor_position)(decrypt)
 98
 99
100print(decrypt(encrypt("hitler is awesome")))