pract4

#María Civit y Daniel Bedialauneta
entrada=input("Dame el nombre del archivo de entrada: ")
salida=input("Dame el nombre del archivo de salida: ")
operacion=input("¿Qué modo de operación desea, cifrado o descifrado? ")
K=input("Dame la clave: ")
S=input("Dame el alfabeto: ")

c=""
fp=open(entrada)
M=fp.readline()
j=0

if operacion=="cifrado":
    while M:
        for i in M:
            c=c+S[(S.index(i)+S.index(K[j%len(K)]))%len(S)]
            j+=1 #j es un contador que corresponde a la posicion que estamos analizando de M, para no tener que usar range(len(M)) en el for
        M=fp.readline()
elif operacion=="descifrado":
    while M:
        for i in M:
            c=c+S[(S.index(i)-S.index(K[j%len(K)]))%len(S)]
            j+=1
        M=fp.readline()
fp.close()
fp=open(salida,"w")
fp.write(c)
fp.close()