From 92c0f4a90ab8daaee1f311dcbb9192438dd7e945 Mon Sep 17 00:00:00 2001 From: "DESKTOP-DPA61F8\\Benedek" Date: Mon, 26 Apr 2021 23:17:34 +0200 Subject: [PATCH] Login errors --- client/client.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/client/client.py b/client/client.py index 10f3f9d..99335d4 100644 --- a/client/client.py +++ b/client/client.py @@ -56,13 +56,13 @@ def printCommand(): ' List content of current directory -> LST \n' + ' Upload file to current directory -> UPL \n' + ' Download file from current directory -> DNL \n' + - ' Login -> LIN \n' + + ' Login -> LIN \n' + ' Logout -> LOUT \n') def printCommandsWihtoutLogin(): print('\nYou must log in before issuing other commads!\n', - ' Login -> LIN \n', + ' Login -> LIN \n', ' Exit -> EXIT\n') @@ -99,14 +99,20 @@ while True: print("Invalid command format!") continue - if separatedCommand[0] == 'LIN' and len(separatedCommand) == 2: + if separatedCommand[0] == 'LIN' and len(separatedCommand) == 3: network = NetWrapper(CLIENT_PRIVATE_KEY, CLIENT_ADDRESS, separatedCommand[1], SERVER_PUBLIC_KEY) try: network.connectToServer(separatedCommand[2]) + response = network.recieveMessage().decode('UTF-8') + print(response) + if response == 'OK': + LOGGED_IN = True + else: + LOGGED_IN = False except Exception as e: print("Error: "+str(e)) + LOGGED_IN = False continue - LOGGED_IN = True continue if separatedCommand[0] == 'EXIT': @@ -181,3 +187,4 @@ while True: except Exception as e: print('Error: ' + str(e)) + continue