make script version agnostic

60acda0b · guoci · GitHub · 09e241fb · 60acda0b
Commit 60acda0b authored 7 years ago by guoci Committed by GitHub 7 years ago
Hide whitespace changes
Inline Side-by-side

Showing

with 5 additions and 2 deletions
+5 -2
--- a/scripts/MakeHeader.py
+++ b/scripts/MakeHeader.py
@@ -2,8 +2,11 @@
 import os, sys, string
 try:
   from cStringIO import StringIO
-except:
-   from StringIO import StringIO
+except ImportError:
+   try:
+      from StringIO import StringIO
+   except ImportError:
+      from io import StringIO

 ANY=1
 COPY=2