sockets

1 files changed, 5 insertions, 3 deletions

diff --git a/chat_server.py b/chat_server.py
index 13ad880..aecf8c9 100644
--- a/chat_server.py
+++ b/chat_server.py
@@ -1,8 +1,9 @@
 #!/usr/bin/env python3
 
-import sys,socket
+import sys,socket,os
 
 def chat_server(port):
+    whoami = os.getenv("USER")
     try:
         s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
         s.bind(('',port))
@@ -12,8 +13,9 @@ def chat_server(port):
         while True:
             data = conn.recv(1024)
             print(data.decode())
-            reponse = input()
-            conn.sendall(reponse.encode())
+            reponse = input("moi > ")
+            reponse = whoami+" > "+reponse
+            conn.sendall(reponse.encode().strip())
     except KeyboardInterrupt:
         s.close()
     finally: