Merge branch 'master' of https://git.kmlabz.com/birbnetes/classification-service

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)
+"""

requirements.txt

@@ -4,6 +4,7 @@ Flask
 marshmallow
 Flask-Classful
 sentry-sdk
+pika
 
 pyAudioanalysis
 numpy