scapy

1 files changed, 22 insertions, 14 deletions

diff --git a/chat_client.py b/chat_client.py
index 0d1d9e8..6de5b88 100644
--- a/chat_client.py
+++ b/chat_client.py
@@ -1,15 +1,23 @@
-#!/usr/bin/env python3
-import socket
+#!/usr/bin/env python
 
-try:
-    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-    s.connect(("localhost", 1234))	# adresse ip + port du serveur (tuple))
-    while 1:
-        message= input()
-        s.sendall(message.encode())
-        data=s.recv(1024)
-        print(data.decode())
-except KeyboardInterrupt:
-    s.close()
-finally:
-    print("bye")
+import sys, socket
+
+def chat_client(host,port):
+    try:
+        s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
+        s.connect((host,port))
+        while True:
+            message = input()
+            s.sendall(message.encode())
+            data=s.recv(1024)
+            print(data.decode().strip())
+    except KeyboardInterrupt:
+        s.close()
+    finally:
+        print("bye")
+
+if __name__ == "__main__":
+    try:
+        chat_client(sys.argv[1],int(sys.argv[2]))
+    except IndexError:
+        print(f"{sys.argv[0]} demande un hôte et un port en arguments")