Added a Common KafkaConsumer for all of the plugins

[osm/MON.git] / core / message_bus / consumer.py

diff --git a/core/message_bus/consumer.py b/core/message_bus/consumer.py
index 3995084..c9021d2 100644 (file)
--- a/core/message_bus/consumer.py
+++ b/core/message_bus/consumer.py
@@ -27,7 +27,7 @@ alarms and metrics responses.
 '''
 
 __author__ = "Prithiv Mohan"
-__date__   = "06/Sep/2017"
+__date__ = "06/Sep/2017"
 
 
 from kafka import KafkaConsumer
@@ -37,6 +37,7 @@ import logging
 import logging.config
 import os
 
+
 def logging_handler(filename, mode='a+', encoding=None):
     if not os.path.exists(filename):
         open(filename, 'a').close()
@@ -50,9 +51,9 @@ log_config = {
         },
     },
     'handlers': {
-        'file':{
+        'file': {
             '()': logging_handler,
-            'level':'DEBUG',
+            'level': 'DEBUG',
             'formatter': 'default',
             'filename': '/var/log/osm_mon.log',
             'mode': 'a+',
@@ -78,8 +79,10 @@ if "BROKER_URI" in os.environ:
 else:
     broker = "localhost:9092"
 
-alarm_consumer = KafkaConsumer('alarm_response', 'osm_mon', bootstrap_servers = broker)
-metric_consumer = KafkaConsumer('metric_response', 'osm_mon', bootstrap_servers = broker)
+alarm_consumer = KafkaConsumer(
+    'alarm_response', 'osm_mon', bootstrap_servers=broker)
+metric_consumer = KafkaConsumer(
+    'metric_response', 'osm_mon', bootstrap_servers=broker)
 try:
     for message in alarm_consumer:
         logger.debug(message)