Move serialization in one place

lib/tfw/networking/event_handler_connector.py

@@ -2,6 +2,7 @@ import zmq
 from zmq.eventloop import ioloop
 from zmq.eventloop.zmqstream import ZMQStream
 
+from .serialization import serialize_all
 from ..config import PUBLISHER_PORT, RECEIVER_PORT
 from ..config.logs import logging
 log = logging.getLogger(__name__)
@@ -40,5 +41,4 @@ class EventHandlerConnector(EventHandlerDownlinkConnector, EventHandlerUplinkCon
     def send_message(self, message: str, anchor: str = None):
         if not anchor:
             anchor = parse_anchor_from_message(message)
-        encoded_message = [part.encode('utf-8') for part in (anchor, message)]
-        self._zmq_pub_socket.send_multipart(encoded_message)
+        self._zmq_pub_socket.send_multipart(serialize_all(anchor, message))

lib/tfw/networking/serialization.py

@@ -0,0 +1,18 @@
+def encode_if_needed(value):
+    if isinstance(value, str):
+        value = value.encode('utf-8')
+    return value
+
+
+def decode_if_needed(value):
+    if isinstance(value, (bytes, bytearray)):
+        value = value.decode('utf-8')
+    return value
+
+
+def serialize_all(*args):
+    return [encode_if_needed(a) for a in args]
+
+
+def deserialize_all(*args):
+    return [decode_if_needed(a) for a in args]

lib/tfw/networking/server_connector.py

@@ -3,6 +3,7 @@ from functools import partial
 from zmq.eventloop import ioloop
 from zmq.eventloop.zmqstream import ZMQStream
 
+from .serialization import serialize_all
 from ..config import PUBLISHER_PORT, RECEIVER_PORT
 from ..util import ZMQConnectorBase
 
@@ -29,7 +30,7 @@ class ServerUplinkConnector(ZMQConnectorBase):
         self._zmq_push_socket.connect('tcp://localhost:{}'.format(RECEIVER_PORT))
 
     def send(self, anchor, response):
-        self._zmq_push_socket.send_multipart([anchor, response])
+        self._zmq_push_socket.send_multipart(serialize_all(anchor, response))
 
 
 class ServerConnector(ServerUplinkConnector, ServerDownlinkConnector):

lib/tfw/networking/zmq_websocket_handler.py

@@ -1,6 +1,7 @@
 import json
 from tornado.websocket import WebSocketHandler
 
+from .serialization import deserialize_all
 from ..util import parse_anchor_from_message
 from .event_handler_connector import EventHandlerConnector
 from ..config.logs import logging
@@ -20,9 +21,9 @@ class ZMQWebSocketHandler(WebSocketHandler):
         self._event_handler_connector.register_callback(self.zmq_callback)
 
     def zmq_callback(self, msg_parts):
-        anchor, data = msg_parts
-        log.debug('Received on pull socket: {}'.format(data.decode()))
-        self.write_message(data.decode())
+        anchor, data = deserialize_all(*msg_parts)
+        log.debug('Received on pull socket: {}'.format(data))
+        self.write_message(data)
 
     def on_message(self, message):
         log.debug('Received on WebSocket: {}'.format(message))