From 1ffcf9f1cedd47187e70e65f63358d6abf42c9cd Mon Sep 17 00:00:00 2001 From: vinmay Date: Fri, 16 Oct 2020 00:54:39 -0400 Subject: [PATCH] Add base64 encoding when message is sent and received --- cnew.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/cnew.py b/cnew.py index ea7b583..936f3a5 100644 --- a/cnew.py +++ b/cnew.py @@ -4,7 +4,7 @@ from prompt_toolkit import print_formatted_text, HTML from prompt_toolkit.validation import Validator, ValidationError from cryptography.fernet import Fernet, InvalidToken - +import base64 sess = PromptSession() sepr = chr(969696) @@ -38,7 +38,7 @@ async def consumer_handler(cphrsuit, websocket, username, chatroom, servaddr): else: recvjson = json.loads(cphrsuit.decrjson(recvdata)) if recvjson["chatroom"] == chatroom and recvjson["username"] != username: - print("[" + obtntime() + "] " + formusnm(recvjson["username"]) + " > " + recvjson["mesgtext"]) + print("[" + obtntime() + "] " + formusnm(recvjson["username"]) + " > " + base64.b64decode(recvjson["mesgtext"])) except Exception as EXPT: pass @@ -48,7 +48,7 @@ async def producer_handler(cphrsuit, websocket, username, chatroom, servaddr): while True: with patch_stdout(): mesgtext = await sess.prompt_async(lambda:"[" + obtntime() + "] " + formusnm(str(username)) + " > ", bottom_toolbar=footelem, validator=emtyfind(), refresh_interval=0.5) - senddata = json.dumps({"username": username.strip(), "chatroom": chatroom, "mesgtext": mesgtext.strip()}) + senddata = json.dumps({"username": username.strip(), "chatroom": chatroom, "mesgtext": base64.b64encode(bytes(mesgtext.strip(), 'utf-8'))}) senddata = cphrsuit.encrjson(senddata) await websocket.send(senddata)