-
Notifications
You must be signed in to change notification settings - Fork 0
/
ChatClient.py
54 lines (48 loc) · 1.47 KB
/
ChatClient.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
import sys
import socket
import threading
clients = []
clients_lock = threading.Lock()
def client(host, port):
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as client_socket:
try:
client_socket.connect((host, port))
receive_thread = threading.Thread(target=receive_messages, args=(client_socket,))
receive_thread.start()
while True:
try:
message = input()
if not message:
break
client_socket.send(message.encode('ASCII'))
except EOFError:
break
except KeyboardInterrupt:
print("\n")
except Exception:
break
except Exception as e:
print(f"Error in client: {e}")
finally:
client_socket.shutdown(socket.SHUT_RDWR)
client_socket.close()
def receive_messages(client_socket):
while True:
try:
data = client_socket.recv(1024)
if not data:
break
print(data.decode())
except ConnectionResetError:
break
except Exception as e:
print("ERROR")
break
if __name__ == "__main__":
if len(sys.argv) < 2:
print("Invalid Args")
sys.exit(1)
else:
port = int(sys.argv[1])
host = socket.gethostbyname('localhost')
client(host, port)