diff options
Diffstat (limited to 'scripts')
| -rw-r--r-- | scripts/ingen.py | 11 |
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() |