zenon/dse-assignment
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

small refactorings

Browse Source
This commit is contained in:
Marco Zeisler 2021-05-24 17:56:10 +02:00
parent aaa4390c1d
commit 8cbb99ad5e
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
components/shared/message_broker_wrapper.py
View File

@ -27,6 +27,10 @@ class MBWrapper:
self.callback = callback self.callback = callback
self.verbose = verbose self.verbose = verbose
def print(self, *msg):
if self.verbose:
print(*msg)
def setup_sender(self): def setup_sender(self):
assert self._type != 'receiver', 'MBWrapper is already a receiver. Use another MBWrapper.' assert self._type != 'receiver', 'MBWrapper is already a receiver. Use another MBWrapper.'
self._type = 'sender' self._type = 'sender'
@ -61,10 +65,6 @@ class MBWrapper:
def close(self): def close(self):
self._connection.close() self._connection.close()
def print(self, *msg):
if self.verbose:
print(*msg)
def _setup_channel(self): def _setup_channel(self):
self._connection = pika.BlockingConnection( self._connection = pika.BlockingConnection(
pika.ConnectionParameters(host=self.host)) pika.ConnectionParameters(host=self.host))
@ -73,4 +73,7 @@ class MBWrapper:
self._channel.exchange_declare(exchange=self.exchange_name, exchange_type=self.exchange_type) self._channel.exchange_declare(exchange=self.exchange_name, exchange_type=self.exchange_type)
def _receive(self, ch, method, properties, body): def _receive(self, ch, method, properties, body):
"""
Reduce complexity, only forward the message body to the callback method
"""
self.callback(body) self.callback(body)