added newline encoding logic

2025-03-04 19:15:06 +02:00
parent 2fd14fc10c
commit 7f8c63ebd9
2 changed files with 4 additions and 3 deletions
@@ -20,7 +20,7 @@ pfsize = int(math.sqrt(len(pfgrid)))

 # message/keystring preparation step: replace 0 with o then filter out all invalid chars
 def filtermsg(msg:str):
-    return ''.join(list(filter(lambda i: i in pfgrid, msg.lower().replace('0', 'o'))))
+    return ''.join(list(filter(lambda i: i in pfgrid, msg.lower().replace('0', 'o').replace('\n', '   '))))

 # grid keying algorithm
 def keygrid(key:str):
@@ -122,7 +122,7 @@ def ipfdecrypt(enc:str, key:str):
                    x2 += pfsize
            msg.append(getgridchar(kg, x2, y2))
    # finalize the decrypted message
-    return ''.join(msg).strip()
+    return ''.join(msg).replace('   ', '\n').strip()

 # entrypoint parameters: [mode] [keystring]
 # modes: e - encrypt with 6x6 IPF, d - decrypt with 6x6 IPF