mrtoth/baseimage-tutorial-framework
1
0
Fork 0
mirror of https://github.com/avatao-content/baseimage-tutorial-framework synced 2024-11-22 18:51:31 +00:00
Code Issues Releases Wiki Activity

Rework MessageSender to conform defined API

Browse Source
This commit is contained in:
Bálint Bokros 2018-02-02 16:49:15 +01:00
parent c99630172b
commit 91d5e560eb
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
lib/tfw/message_sender.py
View File

@ -7,15 +7,18 @@ from .networking.server_connector import ServerUplinkConnector
class MessageSender:
def __init__(self, custom_anchor=None):
self.server_connector = ServerUplinkConnector()
if isinstance(custom_anchor, str):
custom_anchor = custom_anchor.encode('utf-8')
self.anchor = custom_anchor or b'message'
if isinstance(custom_anchor, bytes):
custom_anchor = custom_anchor.decode('utf-8')
self.anchor = custom_anchor or 'message'
def send(self, originator, message):
response = {
data = {
'originator': originator,
'timestamp': datetime.now().isoformat(),
'message': message
}
encoded_response = json.dumps(response).encode('utf-8')
self.server_connector.send(self.anchor, encoded_response)
response = {
'anchor': self.anchor,
'data': data
}
self.server_connector.send(*[frame.encode('utf-8') for frame in (self.anchor, json.dumps(response))])