Merge pull request #275 from karl-johan-grahn/python-tutorial-1-keyboardinterrupt

Python Tutorial 1 - Handle KeyboardInterupt

Author: michaelklishin

python/receive.py

@@ -1,19 +1,26 @@
 #!/usr/bin/env python
-import pika
+import pika, sys, os
 
-connection = pika.BlockingConnection(
-    pika.ConnectionParameters(host='localhost'))
-channel = connection.channel()
+def main():
+    connection = pika.BlockingConnection(pika.ConnectionParameters(host='localhost'))
+    channel = connection.channel()
 
-channel.queue_declare(queue='hello')
+    channel.queue_declare(queue='hello')
 
+    def callback(ch, method, properties, body):
+        print(" [x] Received %r" % body)
 
-def callback(ch, method, properties, body):
-    print(" [x] Received %r" % body)
+    channel.basic_consume(queue='hello', on_message_callback=callback, auto_ack=True)
 
+    print(' [*] Waiting for messages. To exit press CTRL+C')
+    channel.start_consuming()
 
-channel.basic_consume(
-    queue='hello', on_message_callback=callback, auto_ack=True)
-
-print(' [*] Waiting for messages. To exit press CTRL+C')
-channel.start_consuming()
+if __name__ == '__main__':
+    try:
+        main()
+    except KeyboardInterrupt:
+        print('Interrupted')
+        try:
+            sys.exit(0)
+        except SystemExit:
+            os._exit(0)