TheEnigmaMachine/main.py at master · tisu19021997/TheEnigmaMachine · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
import argparse

from components.Rotor import Rotor
from components.PlugBoard import PlugBoard
from components.Machine import Machine
from components.Scrambler import Scrambler
from constants.rotor_models import models
from simulation.Game import Game

if __name__ == '__main__':
    # Construct the Enigma Machine with 3 rotors, 1 type B reflector and a plugboard.
    rotor_1 = Rotor('rotor_i', 'Rotor I')
    rotor_2 = Rotor('rotor_ii', 'Rotor II')
    rotor_3 = Rotor('rotor_iii', 'Rotor III')
    rotor_1.set_offset(1)

    # rotor_1.set_ring(25)
    # rotor_2.set_ring(1)
    # rotor_3.set_ring(1)

    reflector_b = Scrambler(models['reflector_b']['flow'], name='Reflector B')
    plugboard = PlugBoard([])

    machine = Machine()
    machine.add_rotor(rotor_1).add_rotor(rotor_2).add_rotor(rotor_3)
    machine.add_reflector(reflector_b).add_plugboard(plugboard)

    # parser = argparse.ArgumentParser()
    # parser.add_argument('enci', help='encipher the message using The Enigma Machine')
    # parser.add_argument('-v', '--verbose', help='whether to print the enciphering flow of the machine',
    #                     action='store_true')
    # args = parser.parse_args()
    # message = args.enci.replace(' ', '').upper()
    # verbose = args.verbose
    #
    # enciphered = machine.encipher(message, verbose)
    #
    # # assert enciphered == 'wts'.replace(' ', '').upper()
    # print(f'Enciphered "{message}" into: "{enciphered}"')

    game = Game(machine=machine, background=(255, 255, 0))
    game.start()