diff --git a/classification_service/utils/rabbit.py b/classification_service/utils/rabbit.py new file mode 100644 index 0000000..3370599 --- /dev/null +++ b/classification_service/utils/rabbit.py @@ -0,0 +1,26 @@ +#!/usr/bin/env python3 + +import pika +from config import * + +""" +Rabbitmq setup +""" + +__author__ = '@tormakris' +__copyright__ = "Copyright 2020, Birbnetes Team" +__module_name__ = "rabbit" +__version__text__ = "1" + +credentials = pika.PlainCredentials(RABBITMQ_USERNAME, RABBITMQ_PASSWORD) +rabbitmq = pika.BlockingConnection(pika.ConnectionParameters(host=RABBITMQ_HOST, credentials=credentials)) +rabbitmq_channel = rabbitmq.channel() +rabbitmq_channel.exchange_declare(exchange=RABBITMQ_EXCHANGE, exchange_type='fanout') + +"""Usage: + from rabbit import rabbitmq_channel + rabbitmq_channel.basic_publish( + exchange=RABBITMQ_EXCHANGE, + routing_key='feature', + body=generated_tag) +""" \ No newline at end of file