diff --git a/components/shared/message_broker_wrapper.py b/components/shared/message_broker_wrapper.py index 3f310e1..8dc3ca9 100644 --- a/components/shared/message_broker_wrapper.py +++ b/components/shared/message_broker_wrapper.py @@ -43,7 +43,7 @@ class MBWrapper: self._channel.queue_bind(exchange=self.exchange_name, queue=queue_name) - self.print(' [*] Waiting for messages. To exit press CTRL+C') + self.print('[{}] Waiting for messages.'.format(self.exchange_name)) self._channel.basic_consume( queue=queue_name, on_message_callback=self._receive, auto_ack=True) @@ -52,7 +52,7 @@ class MBWrapper: Thread(target=consumer).start() - def send(self, message: bytes = b"info: Hello World!"): + def send(self, message: bytes): if type(message) is not bytes: message = str(message).encode() self._channel.basic_publish(exchange=self.exchange_name, routing_key='', body=message)