1def plugboard(string_to_encode: str, plugboard_position: list):
 2    """Encode a String based on the Pairs of Chars in the List of Sets.
 3
 4    This Function creates a Dict from the List of Pairs and extends it with the Swapped Pairs.
 5    This way it can use the rotor Function.
 6    """
 7
 8    tuple_pairs = list([tuple(curr_set) for curr_set in plugboard_position])
 9    tuple_pairs.extend([(y, x) for x, y in tuple_pairs])
10    encoding_dict = dict(tuple_pairs)
11
12    return rotor(string_to_encode, encoding_dict)
13
14
15def rotor(string_to_encode: str, rotor_position: dict):
16    """Encodes a String using the Key, Value Pairs in the Dict."""
17    return "".join([rotor_position.get(ch, ch) for ch in string_to_encode])
18
19
20def enigma_encrypt(plugboard_position: list, rotor_position: dict):
21    """Create a Decorator for a Function with the given parameters using the Functions plugboard and rotor."""
22
23    def enigma_encrypt_decorator(func):
24        """Decorate an Encrypting Function that excepts a single String, to encrypt the String before calling it."""
25
26        def decorated_func(s: str):
27            """Encrypt a String before calling a Function."""
28            return func(rotor(plugboard(s, plugboard_position), rotor_position))
29
30        return decorated_func
31
32    return enigma_encrypt_decorator
33
34
35def enigma_decrypt(plugboard_position: list, rotor_position: dict):
36    """Create a Decrypting Decorator for a Function with the given positions using the Functions plugboard and rotor."""
37    rotor_position = {y: x for x, y in rotor_position.items()}
38
39    def enigma_decrypt_decorator(func):
40        """Decorate a Function that excepts a single String, to decrypt the String before calling it."""
41
42        def decorated_func(s: str):
43            """Decrypt a String before calling a Function."""
44            return func(plugboard(rotor(s, rotor_position), plugboard_position))
45
46        return decorated_func
47
48    return enigma_decrypt_decorator