Fix support for Python 3

Closes #144.
This commit is contained in:
Tim Pope 2014-04-13 16:44:37 -04:00
parent 7b19c1b2f6
commit c0a4901181
1 changed files with 12 additions and 6 deletions

View File

@ -4,7 +4,10 @@ import select
import socket
import sys
from StringIO import StringIO
try:
from StringIO import StringIO
except ImportError:
from io import StringIO
def noop():
pass
@ -84,13 +87,16 @@ class Connection:
return self.socket.close()
def send(self, payload):
if sys.version_info[0] >= 3:
self.socket.sendall(bytes(payload, 'UTF-8'))
else:
self.socket.sendall(payload)
return ''
def receive(self, char=None):
while len(select.select([self.socket], [], [], 0.1)[0]) == 0:
self.poll()
f = self.socket.makefile()
while len(select.select([f], [], [], 0.1)[0]) == 0:
self.poll()
try:
return bdecode(f)
finally:
@ -118,8 +124,8 @@ def dispatch(host, port, poll, keepalive, command, *args):
def main(host, port, keepalive, command, *args):
try:
sys.stdout.write(vim_encode(dispatch(host, port, noop, keepalive, command, *[bdecode(StringIO(arg)) for arg in args])))
except Exception, e:
print(e)
except Exception:
print((sys.exc_info()[1]))
exit(1)
if __name__ == "__main__":