impressionWheel.py

from collections import deque 

#Se puede desplazar la rueda de impresión

#inicialmente la rueda sería dos arrays
def impressionWheels(n):
    rueda1 = deque(['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z'])
    rueda2 = deque(['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z'])
    rueda2.rotate(n)    #Se aplica un desplazamiento inicial de n elementos
    return (rueda1, rueda2)

def imprime(rueda):
    return rueda[0]

def imprimeTextoR1(letra, rueda1, rueda2):
    while letra != rueda1[0]:
        #Aquí las ruedas de impresión rotan en sentido contrario pues son recíprocas (de no ser así el proceso de descifrado no funcionaría)
        rueda1.rotate(-1)
        rueda2.rotate(1)
    return imprime(rueda1)

def imprimeTextoR2(rueda1, rueda2, n, variable):
    if (not variable):  #En modo variable se desenganchan las ruedas y gira sólo la segunda
        rueda1.rotate(-n)
    rueda2.rotate(n)
    return imprime(rueda2)