adapt Py2-only 'raise' statements to Py2/Py3

1 files changed, 7 insertions, 3 deletions

diff --git a/pyximport/pyximport.py b/pyximport/pyximport.py
index 3769050b8..c1ad92772 100644
--- a/pyximport/pyximport.py
+++ b/pyximport/pyximport.py
@@ -217,10 +217,14 @@ def load_module(name, pyxfilename, pyxbuild_dir=None, is_package=False,
             mod = imp.load_source(name, pyxfilename)
             assert mod.__file__ in (pyxfilename, pyxfilename+'c', pyxfilename+'o'), (mod.__file__, pyxfilename)
         else:
+            tb = sys.exc_info()[2]
             import traceback
-            raise ImportError("Building module %s failed: %s" %
-                              (name,
-                               traceback.format_exception_only(*sys.exc_info()[:2]))), None, sys.exc_info()[2]
+            exc = ImportError("Building module %s failed: %s" % (
+                name, traceback.format_exception_only(*sys.exc_info()[:2])))
+            if sys.version_info[0] >= 3:
+                raise exc.with_traceback(tb)
+            else:
+                exec("raise exc, None, tb", {'exc': exc, 'tb': tb})
     return mod