enigmaencrypt.py

from enigma.machine import EnigmaMachine

# setup machine according to specs from a daily key sheet:

machine = EnigmaMachine.from_key_sheet(
       rotors='II IV V',
       reflector='B',
       ring_settings='B U L',
       plugboard_settings='AV BS CG DL FU HZ IN KM OW RX')

rotorpositions = input("Anna roottorin alkuasetukset, 3 isoa kirjainta, esim WXC ")
print("initial rotor positions =",rotorpositions)
# set initial rotor position
machine.set_display(rotorpositions)
msg_key = input("Anna message key, 3 isoa kirjainta, esim KCH ")
# encrypt message key
enc_key = machine.process_text(msg_key)  # encrypt message key
print("encrypted message key = ",enc_key)
machine.set_display(msg_key)  # use message key
plaintext = input("Anna selväkielinen teksti, isot kirjaimet ")
ciphertext = machine.process_text(plaintext)
print("ciphertext = ",ciphertext)
print("ciphertext in five letter groups:")
for i in range(0,len(ciphertext)-4):
    print(ciphertext[5*i:5*i+5],end=" ")