Fix network communication with Python bindings

1 files changed, 6 insertions, 5 deletions

diff --git a/scripts/ingen.py b/scripts/ingen.py
index d56d7e0b..a9f2c4ed 100644
--- a/scripts/ingen.py
+++ b/scripts/ingen.py
@@ -156,11 +156,12 @@ class Remote(Interface):
         'Read from socket until a null terminator is received'
         msg = u''
         while True:
-            c = self.sock.recv(1, 0).decode('utf-8')
-            if not c or ord(c[0]) == 0:  # End of transmission
+            chunk = self.sock.recv(1, 0)
+            if not chunk or chunk[0] == 0:  # End of transmission
                 break
-            else:
-                msg += c[0]
+
+            msg += chunk.decode('utf-8')
+
         return msg
 
     def blank_closure(self, graph, node):
@@ -192,7 +193,7 @@ class Remote(Interface):
             msg = '\n'.join(msg)
 
         # Send message to server
-        self.sock.send(self.msgencode(msg))
+        self.sock.send(self.msgencode(msg) + b'\0')
 
         # Receive response and parse into a model
         response_str = self._get_prefixes_string() + self.recv()