1def plugboard(input_str, current_plugboard):
 2    new_str = []
 3    for letter in input_str:
 4        for plugboard_set in current_plugboard:
 5            if letter in plugboard_set:
 6                # gets the letter from the set which is not the current letter and turns it into char
 7                new_letter = plugboard_set.difference({letter}).pop() 
 8                new_str.append(new_letter)
 9                break
10        else:
11            new_str.append(letter)
12    return ''.join(new_str)
13
14
15def rotor(input_str, current_rotor):
16    new_str = []
17    for letter in input_str:
18        new_letter = current_rotor.get(letter, ' ')
19        new_str.append(new_letter)
20    return ''.join(new_str)
21
22
23def enigma_encrypt(plugboard_position, rotor_position):
24    def decorator(my_function):
25        def func(input_str):
26            new_str = plugboard(input_str, plugboard_position)
27            new_str = rotor(new_str, rotor_position)
28            return my_function(new_str)
29        return func
30    return decorator
31
32
33def enigma_decrypt(plugboard_position, rotor_position):
34    def decorator(my_function):
35        def func(input_str):
36            # swapping the key and the value in the current rotor
37            new_rotor_position = dict([(value, key) for key, value in rotor_position.items()]) 
38            new_str = rotor(input_str, new_rotor_position)
39            new_str = plugboard(new_str, plugboard_position)
40            return my_function(new_str)
41        return func
42    return decorator