class Railfence:
def __railfence_id(self, tam, key):
''' (Railfence, int, int) -> list of int
Retorna um lista de inteiros com a posicao da linha que o caracter do texto ira ocupar, variando de 0 ate key - 1.
'''
j = 0
inc = 0
idx = []
for i in range(tam):
if j == key - 1:
inc = -1
elif j == 0:
inc = 1
idx.append(j)
j += inc
return idx
def encrypt(self, texto, key):
''' (Railfence, str, int) -> str
Retorna o texto plano cifrado na cifra rail fence com a chave key.
'''
texto = texto.replace(' ', '')
tam = len(texto)
idx = self.__railfence_id(tam, key)
cifrado = ''
for i in range(key):
for z in range(tam):
if idx[z] == i:
cifrado += texto[z]
return cifrado.upper()
def decrypt(self, texto, key):
''' (Railfence, str, int) -> str
Retorna o texto plano para um texto cifrado com a cifra rail fence com a chave key.
'''
texto = texto.replace(' ', '')
tam = len(texto)
idx = self.__railfence_id(tam, key)
idx_sorted = sorted(idx)
texto_plano = ''
for i in range(tam):
for j in range(tam):
if idx[i] == idx_sorted[j] and idx[i] > -1:
texto_plano += texto[j]
idx[i] = -1
idx_sorted[j] = -1
return texto_plano.lower()
