From ff1ef600522c26481a06b426c14b041f91e35fba Mon Sep 17 00:00:00 2001
From: penguin0001 <robynpgreer@gmail.com>
Date: Mon, 9 May 2022 12:27:07 +0100
Subject: [PATCH] client end messages

---
 client.py | 14 +++++---------
 1 file changed, 5 insertions(+), 9 deletions(-)

diff --git a/client.py b/client.py
index 473e245..d607654 100644
--- a/client.py
+++ b/client.py
@@ -28,17 +28,13 @@ user_input = input("Enter username: ")
 while connect:
     server_msg = send(user_input)
     print(server_msg)
-    user_input = input()
-    if user_input.upper() == "BYE" or server_msg == "ERR" or server_msg == "DONE" or server_msg == "END":
+    if server_msg == "ERR" or server_msg == "DONE" or server_msg == "END":
         connect = False
+    else:
+        user_input = input()
+        if user_input.upper() == "BYE":
+            connect = False
 
 
 # disconnect
 print(send("BYE"))
-
-
-
-
-
-
-
-- 
GitLab