From 8cbb99ad5eb1b5cc496f2d575ac15d0e72b49f23 Mon Sep 17 00:00:00 2001 From: Marco Zeisler Date: Mon, 24 May 2021 17:56:10 +0200 Subject: [PATCH] small refactorings --- components/shared/message_broker_wrapper.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/components/shared/message_broker_wrapper.py b/components/shared/message_broker_wrapper.py index 8a6f300..fbe5ae0 100644 --- a/components/shared/message_broker_wrapper.py +++ b/components/shared/message_broker_wrapper.py @@ -27,6 +27,10 @@ class MBWrapper: self.callback = callback self.verbose = verbose + def print(self, *msg): + if self.verbose: + print(*msg) + def setup_sender(self): assert self._type != 'receiver', 'MBWrapper is already a receiver. Use another MBWrapper.' self._type = 'sender' @@ -61,10 +65,6 @@ class MBWrapper: def close(self): self._connection.close() - def print(self, *msg): - if self.verbose: - print(*msg) - def _setup_channel(self): self._connection = pika.BlockingConnection( pika.ConnectionParameters(host=self.host)) @@ -73,4 +73,7 @@ class MBWrapper: self._channel.exchange_declare(exchange=self.exchange_name, exchange_type=self.exchange_type) def _receive(self, ch, method, properties, body): + """ + Reduce complexity, only forward the message body to the callback method + """ self.callback(body)