40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
#!/usr/bin/env python3
|
|
import logging
|
|
import os
|
|
import sys
|
|
import pika
|
|
import json
|
|
|
|
from extraction import run_everything
|
|
|
|
|
|
def message_callback(ch, method, properties, body):
|
|
run_everything(json.loads(body.decode('utf-8')))
|
|
|
|
|
|
def main():
|
|
logging.basicConfig(filename="", format="%(asctime)s - %(name)s [%(levelname)s]: %(message)s",
|
|
level=logging.DEBUG if '--debug' in sys.argv else logging.INFO)
|
|
|
|
logging.info("Connecting to MQ service...")
|
|
connection = pika.BlockingConnection(pika.connection.URLParameters(os.environ['PIKA_URL']))
|
|
channel = connection.channel()
|
|
channel.exchange_declare(exchange='wave-extract', exchange_type='fanout')
|
|
|
|
queue_declare_result = channel.queue_declare(queue='', exclusive=True)
|
|
queue_name = queue_declare_result.method.queue
|
|
|
|
channel.queue_bind(exchange='wave-extract', queue=queue_name)
|
|
channel.basic_consume(queue=queue_name, on_message_callback=message_callback, auto_ack=True)
|
|
|
|
logging.info("Connection complete! Listening to messages...")
|
|
try:
|
|
channel.start_consuming()
|
|
except KeyboardInterrupt:
|
|
logging.info("SIGINT Received! Stopping stuff...")
|
|
channel.stop_consuming()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|