Python 2x compatible

erlastic/__init__.py

@@ -1,28 +1,39 @@
 """Erlang External Term Format serializer/deserializer"""
+import struct
+import sys
+import six
 
 from erlastic.codec import ErlangTermDecoder, ErlangTermEncoder
 from erlastic.types import *
 
 encode = ErlangTermEncoder().encode
 decode = ErlangTermDecoder().decode
 
-import struct
-import sys
+if six.PY3:
+    stdread = sys.stdin.buffer.read
+    stdwrite = sys.stdout.buffer.write
+else:
+    stdread = sys.stdin.read
+    stdwrite = sys.stdout.write
+
 
 def mailbox_gen():
-  while True:
-    len_bin = sys.stdin.buffer.read(4)
-    if len(len_bin) != 4: return
-    (length,) = struct.unpack('!I',len_bin)
-    yield decode(sys.stdin.buffer.read(length))
+    while True:
+        len_bin = stdread(4)
+        if len(len_bin) != 4:
+            return
+        (length,) = struct.unpack('!I', len_bin)
+        yield decode(stdread(length))
+
 
 def port_gen():
-  while True:
-    term = encode((yield))
-    sys.stdout.buffer.write(struct.pack('!I',len(term)))
-    sys.stdout.buffer.write(term)
+    while True:
+        term = encode((yield))
+        stdwrite(struct.pack('!I', len(term)))
+        stdwrite(term)
+
 
 def port_connection():
-  port = port_gen()
-  next(port)
-  return mailbox_gen(),port
+    port = port_gen()
+    next(port)
+    return mailbox_gen(), port