Simplify handle_client loop / indent, check for EOF when reading from socket.

This commit is contained in:
Steve Dodd
2023-07-01 21:39:11 +01:00
parent 6bdff30283
commit 479bd272e9
+8 -5
View File
@@ -341,15 +341,18 @@ def handle_client(conn, addr):
global EXCLUDE_LIST global EXCLUDE_LIST
# print(f"[NEW CONNECTION] {addr} connected.") # print(f"[NEW CONNECTION] {addr} connected.")
connected = True while True:
while connected:
msg_length = conn.recv(HEADER).decode(FORMAT) msg_length = conn.recv(HEADER).decode(FORMAT)
if msg_length: if not msg_length:
break
msg_length = int(msg_length) msg_length = int(msg_length)
msg = conn.recv(msg_length).decode(FORMAT) msg = conn.recv(msg_length).decode(FORMAT)
if not msg:
break
if msg == DISCONNECT_MESSAGE: if msg == DISCONNECT_MESSAGE:
connected = False break
else:
# print(f"[{addr}] {msg}") # print(f"[{addr}] {msg}")
args = type('', (), {})() args = type('', (), {})()