diff --git a/consumer.py b/consumer.py index 72e3635..9cf2638 100644 --- a/consumer.py +++ b/consumer.py @@ -45,6 +45,9 @@ class MessageConsumer: if not hasattr(msg, "server") or msg.server is None: msg.server = self.srv.id + if hasattr(msg, "frm_owner"): + msg.frm_owner = (self.srv.owner == msg.frm) + return msg diff --git a/message/__init__.py b/message/__init__.py index 3595f3a..f3a107a 100644 --- a/message/__init__.py +++ b/message/__init__.py @@ -40,6 +40,8 @@ class AbstractMessage: self._to_response = to_response if to_response is None or isinstance(to_response, list) else [ to_response ] self.frm = frm # None allowed when it designate this bot + self.frm_owner = False # Filled later, in consumer + @property def to_response(self):