make script version agnostic

9 years ago · 60acda0bce
parent 09e241fb12
commit 60acda0bce
1 changed files with 5 additions and 2 deletions
--- 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