Реализовать алгоритмы шифрования и дешифрования табличной маршрутной перестановки

from math import ceil
def encrypt(s, key):
    
    s = (w for w in s + ' '*(key*key - len(s)))
    arr = [['']*key for _ in range(key)]
    
    for j in range(key)[::-1]:
        a, b, c = key-1, -1, -1
        if (key-1-j)%2:
            a, b, c = 0, key, 1
        for i in range(a, b, c):
            arr[i][j] = next(s)
    
    return ''.join(''.join(elem) for elem in arr)
 
 
def decrypt(s, key):
    arr = []
    for i in range(0, key*key, key):
        arr.append(list(s[i:i+key])[::-1])
    
    s = ''
    k = 0
    for row in zip(*arr[::-1]):
        if k%2:
            s += ''.join(row[::-1])
        else:
            s += ''.join(row)
        k += 1
    
    return s
 
 
a = 'Программно реализовать\
 алгоритмы шифрования и\
  дешифрования табличной\
   маршрутной перестановки для таблицы'
 
key = ceil(len(a)**.5)
a_encr = encrypt(a, key)
print(a_encr)
print()
 
key = ceil(len(a_encr)**.5)
a_decr = decrypt(a_encr,key)
print(a_decr)

s = (w for w in s + ' '*(key*key - len(s))) 
    arr = [['']*key for _ in range(key)]

for i in range(0, key*key, key):
        arr.append(list(s[i:i+key])[::-1])

s = (w for w in s + ' '*(key*key - len(s))) 
arr = [['']*key for _ in range(key)]

s = iter(s + ' '*(key*key - len(s)))

for i in range(0, key*key, key):
        arr.append(list(s[i:i+key])[::-1])