=)

Author: zavatskiy

zavatskiy/client01.py

@@ -1,3 +1,4 @@
+import time
 import socket
 
 from work.helpers import make_message, parse_message
@@ -30,7 +31,10 @@ def close(self):
     while True:
         data = raw_input('pingd> ')
         client.send('pingd', data)
+
         ans = client.recive()
+        if not ans[0]:
+            client.close()
         if ans[0] in ['ackquit', 'ackfinish']:
             break
 

zavatskiy/server01.py

@@ -53,14 +53,19 @@ def handle_request(self, conn, addr):
     def serve(self):
         """ Run server """
         while not self.__shutdown:
-            conn, addr = self.socket.accept()
-            self.new_stream(conn, addr)
+            try:
+                conn, addr = self.socket.accept()
+                self.new_stream(conn, addr)
+            except KeyboardInterrupt:
+                self.__shutdown = True
+
         self.shutdown()
 
     def shutdown(self):
         """ Shutdown server """
         for thread in self.threads:
             thread.join()
+
         self.socket.close()