OK

client/client.py

@@ -89,6 +89,7 @@ if not os.path.isfile(CONFIG_LOCATION) or os.stat(CONFIG_LOCATION).st_size == 0:
 SERVER_PUBLIC_KEY = loadPublicKey()
 CLIENT_PRIVATE_KEY = loadPrivateKey(PASSPHRASE)
 CLIENT_ADDRESS = loadAddress()
+LOGGED_IN = False
 
 while True:
     command = input("Type a command:")
@@ -99,21 +100,25 @@ while True:
             print("Invalid command format!")
             continue
 
-        if separatedCommand[0] == 'LIN' and len(separatedCommand) == 3:
+        if not LOGGED_IN:
-            network = NetWrapper(CLIENT_PRIVATE_KEY, CLIENT_ADDRESS, separatedCommand[1], SERVER_PUBLIC_KEY)
+            if separatedCommand[0] == 'LIN' and len(separatedCommand) == 3:
-            try:
+                network = NetWrapper(CLIENT_PRIVATE_KEY, CLIENT_ADDRESS, separatedCommand[1], SERVER_PUBLIC_KEY)
-                network.connectToServer(separatedCommand[2])
+                try:
-                response = network.recieveMessage().decode('UTF-8')
+                    network.connectToServer(separatedCommand[2])
-                print(response)
+                    response = network.recieveMessage().decode('UTF-8')
-                if response == 'OK':
+                    print(response)
-                    LOGGED_IN = True
+                    if response == 'OK':
-                else:
+                        LOGGED_IN = True
+                    else:
+                        LOGGED_IN = False
+                except Exception as e:
+                    print("Error: "+str(e))
                     LOGGED_IN = False
-            except Exception as e:
+                    continue
-                print("Error: "+str(e))
-                LOGGED_IN = False
                 continue
-            continue
+        else:
+            print('You are already logged in!')
+
 
         if separatedCommand[0] == 'EXIT':
             sys.exit(1)