- more python 3.x fixes

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

diff --git a/TLD_Builder/wrap.py b/TLD_Builder/wrap.py
index 0c9df39281cf6d9f724f9d308ba2b573a762e2ff..3d7c4a8867013fa2f90b51bf05447a569dfa68a1 100644 (file)
--- 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")