Encrypt longer messages

from enigma.machine import EnigmaMachine

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

# Set the initial position of the Enigma rotors
machine.set_display('QJF')

def pad_string(input):
    padding_needed = (5 - len(input) % 5) % 5
    # Append the "X"s to the end of the string
    padded_string = input + "X" * padding_needed
    return padded_string

def format_for_transmission(input_string):
    # Use list comprehension to insert a space after every 5th character
    # [input_string[i:i+5] for i in range(0, len(input_string), 5)] creates substrings of every 5 characters
    # ' '.join(...) then joins these substrings with a space

    # Calculate the number of "X"s needed to make the length divisible by 5
    
    return ' '.join(input_string[i:i+5] for i in range(0, len(input_string), 5))

plaintext = 'HEY ROBOT MAKERS THIS IS A SECRET MESSAGE'
plaintext = plaintext.replace(" ","X")
plaintext = pad_string(plaintext)
ciphertext = machine.process_text(plaintext)

print(format_for_transmission(ciphertext))

# The output should be:
# SUPSH GPYCV JTPYF TDQWV HIBEW FPDBN TAUEL IQXMS ZBDCT