X-Git-Url: https://git.tld-linux.org/?a=blobdiff_plain;f=TLD_Builder%2Fwrap.py;h=3d7c4a8867013fa2f90b51bf05447a569dfa68a1;hb=3a6296fff210d498302ce0a2da6beed689466347;hp=0c9df39281cf6d9f724f9d308ba2b573a762e2ff;hpb=b999f53d4bf5d44586ecf028876e8bc20b5fd2ce;p=tld-builder.git

diff --git a/TLD_Builder/wrap.py b/TLD_Builder/wrap.py
index 0c9df39..3d7c4a8 100644
--- a/TLD_Builder/wrap.py
+++ b/TLD_Builder/wrap.py
@@ -3,7 +3,11 @@
 import sys
 import log
 import traceback
-import StringIO
+import sys
+if sys.version_info[0] == 2:
+    import StringIO
+else:
+    from io import StringIO
 import os
 import time
 
@@ -44,7 +48,7 @@ def wrap(main):
         exctype, value = sys.exc_info()[:2]
         if exctype == SystemExit:
             sys.exit(value)
-        s = StringIO.StringIO()
+        s = StringIO()
         traceback.print_exc(file = s, limit = 20)
 
         log.alert("fatal python exception")