Added rabbitmq host to init_receiver

slaveeks · slaveeks · commit 233adc55d859 · 2021-12-15T17:15:45.000+03:00
diff --git a/codexbot_sdk.py b/codexbot_sdk.py
@@ -49,7 +49,7 @@ def __init__(self, application_name, host, port, rabbitmq_host, db_config, token
         self.db = self.init_db(db_config)
         self.scheduler = self.init_scheduler()
         self.server = self.init_server()
-        self.broker = self.init_broker(application_name, queue_name, rabbitmq_host , self.hawk)
+        self.broker = self.init_broker(application_name, queue_name, rabbitmq_host, self.hawk)
 
         self.broker.start()
 
diff --git a/components/broker.py b/components/broker.py
@@ -20,9 +20,10 @@ def __init__(self, core, event_loop, application_name, queue_name, rabbitmq_host
         self.core = core
         self.event_loop = event_loop
         self.queue_name = queue_name
+        self.rabbitmq_host = rabbitmq_host
         self.api = API(self, application_name, hawk)
 
-        self.channel = self.event_loop.run_until_complete(connect(host = rabbitmq_host))
+        self.channel = self.event_loop.run_until_complete(connect(host=rabbitmq_host))
 
     async def callback(self, message: IncomingMessage):
         try:
@@ -38,4 +39,4 @@ async def send(self, message):
         await add_message_to_queue(message, 'core', self.channel)
 
     def start(self):
-        self.event_loop.run_until_complete(init_receiver(self.callback, self.queue_name))
+        self.event_loop.run_until_complete(init_receiver(self.callback, self.queue_name, self.rabbitmq_host))
