Update MessageSender to implement designed message format

This commit is contained in:
Bálint Bokros 2018-01-30 17:24:17 +01:00
parent 7b76afe232
commit 6facda3520

View File

@ -1,3 +1,6 @@
import json
from datetime import datetime
from server_connector import ServerUplinkConnector from server_connector import ServerUplinkConnector
@ -8,7 +11,11 @@ class MessageSender:
custom_anchor = custom_anchor.encode('utf-8') custom_anchor = custom_anchor.encode('utf-8')
self.anchor = custom_anchor or b'message' self.anchor = custom_anchor or b'message'
def send(self, message): def send(self, originator, message):
if isinstance(message, str): response = {
message = message.encode('utf-8') 'originator': originator,
self.server_connector.send(self.anchor, message) 'timestamp': datetime.now().isoformat(),
'message': message
}
encoded_response = json.dumps(response).encode('utf-8')
self.server_connector.send(self.anchor, encoded_response)