catch KeyboardInterrupt

kafka/consumer.py

@@ -16,6 +16,6 @@ try:
         for message in consumer:
             DATETIMES.append(datetime.now())
     consumer.close()
-except Exception:
+except KeyboardInterrupt:
     consumer.close()
     print(DATETIMES)

kafka/publisher.py

@@ -11,5 +11,5 @@ try:
     binstring = string.encode('UTF-8')
     while True:
         producer.send('test', binstring)
-except Exception:
+except KeyboardInterrupt:
     producer.close()