diff --git a/webkit_server.py b/webkit_server.py index 4591f18..b551e9e 100644 --- a/webkit_server.py +++ b/webkit_server.py @@ -513,9 +513,12 @@ def issue_command(self, cmd, *args): self._writeline(cmd) self._writeline(str(len(args))) for arg in args: - arg = str(arg) + if type(arg) != str and type(arg) != unicode: + arg = str(arg) + if type(arg) == unicode: + arg = arg.encode("utf-8") self._writeline(str(len(arg))) - self._sock.sendall(arg.encode("utf-8")) + self._sock.sendall(arg) return self._read_response() @@ -537,5 +540,7 @@ def _read_message(self): return self.buf.read(size).decode("utf-8") def _writeline(self, line): + if type(line) == unicode: + line = line.encode("utf-8") """ Writes a line to the underlying socket. """ - self._sock.sendall(line.encode("utf-8") + b"\n") + self._sock.sendall(line + b"\n")