- more python 3.x fixes

[tld-builder.git] / TLD_Builder / pipeutil.py

diff --git a/TLD_Builder/pipeutil.py b/TLD_Builder/pipeutil.py
index 4979fea55683bdb0b3f82c59d4d43cad689b92cd..a3e78c9a1d7be8ff7b819a12d58c43c0ea5fdbe2 100644 (file)
--- a/TLD_Builder/pipeutil.py
+++ b/TLD_Builder/pipeutil.py
@@ -2,12 +2,16 @@
 
 import select
 import os
-import StringIO
+import sys
+if sys.version_info[0] == 2:
+    import StringIO
+else:
+    from io import StringIO
 
 def rw_pipe(buf_, infd, outfd):
-    buf = StringIO.StringIO()
+    buf = StringIO()
     buf.write(buf_.read())
-    ret = StringIO.StringIO()
+    ret = StringIO()
     pos = 0
     rd_fin = 0
     wr_fin = 0