diff options
Diffstat (limited to 'rmq')
| -rw-r--r-- | rmq/consumer.py | 11 | ||||
| -rw-r--r-- | rmq/publisher.py | 21 |
2 files changed, 24 insertions, 8 deletions
diff --git a/rmq/consumer.py b/rmq/consumer.py index 8e1ce2b..2c41e26 100644 --- a/rmq/consumer.py +++ b/rmq/consumer.py @@ -3,7 +3,14 @@ import traceback import pika from pika.exceptions import ConnectionClosed -def consume(username='guest', password='guest', host='localhost', port=5672, queue='', callback=None): + +def consume(username='guest', + password='guest', + host='localhost', + port=5672, + queue='', + callback=None): + while True: credentials = pika.PlainCredentials(username, password) con_params = pika.ConnectionParameters(host, port, '/', credentials) @@ -21,7 +28,7 @@ def consume(username='guest', password='guest', host='localhost', port=5672, que channel.start_consuming() - except: + except Exception: exc = traceback.format_exc() print(exc, flush=True) diff --git a/rmq/publisher.py b/rmq/publisher.py index 4ba9db9..1a9a5cf 100644 --- a/rmq/publisher.py +++ b/rmq/publisher.py @@ -1,6 +1,16 @@ import pika -def publish(message, username='guest', password='guest', host='localhost', port=5672, queue='', routingkey='', exchange='', exchange_type=''): + +def publish(message, + username='guest', + password='guest', + host='localhost', + port=5672, + queue='', + routingkey='', + exchange='', + exchange_type=''): + credentials = pika.PlainCredentials(username, password) connection = pika.BlockingConnection(pika.ConnectionParameters(host, port, '/', credentials)) properties = pika.BasicProperties(content_type='text/plain', delivery_mode=1) @@ -11,10 +21,10 @@ def publish(message, username='guest', password='guest', host='localhost', port= channel.queue_bind(exchange=exchange, queue=queue, routing_key=routingkey) result = channel.basic_publish( - exchange=exchange, - routing_key=routingkey, - body=message, - properties=properties + exchange=exchange, + routing_key=routingkey, + body=message, + properties=properties ) if result: @@ -23,4 +33,3 @@ def publish(message, username='guest', password='guest', host='localhost', port= print("not delivered") connection.close() - |