chat

1 files changed, 11 insertions, 8 deletions

diff --git a/chat_client.py b/chat_client.py
index 6de5b88..3b3c082 100644
--- a/chat_client.py
+++ b/chat_client.py
@@ -1,23 +1,26 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 
-import sys, socket
+import sys,socket,os
 
 def chat_client(host,port):
+    whoami = os.getenv("USER")
     try:
-        s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
+        s = socket.socket(socket.AF_INET,socket.SOCK_STREAM) 
         s.connect((host,port))
         while True:
-            message = input()
+            message = input("moi > ")
+            message = whoami+" > "+message
             s.sendall(message.encode())
-            data=s.recv(1024)
-            print(data.decode().strip())
+            data = s.recv(1024)
+            print(data.decode())
     except KeyboardInterrupt:
         s.close()
     finally:
         print("bye")
 
-if __name__ == "__main__":
+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")
+        print(f"{sys.argv[0]} demande un hôte où se connecter et un numéro de port")
+