python: Fix version detection

python: Use list comprehension so setup.py runs on python3 - fixes #429 Signed-off-by: Benn Snyder <benn.snyder@gmail.com>
OpenKinect · Jan 7, 2015 · d8729aa · d8729aa
1 parent e551c9a
commit d8729aa
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/wrappers/python/setup.py b/wrappers/python/setup.py
@@ -14,10 +14,16 @@ def get_cython_version():
         ImportError: Can't load cython or find version
     """
     import Cython.Compiler.Main
-    match = re.search('^([0-9]+)\.([0-9]+)',
-                      Cython.Compiler.Main.Version.version)
+
+    try:
+        # old way, fails for me
+        version = Cython.Compiler.Main.Version.version
+    except AttributeError:
+        version = Cython.Compiler.Main.version
+
+    match = re.search('^([0-9]+)\.([0-9]+)', version)
     try:
-        return map(int, match.groups())
+        return [int(g) for g in match.groups()]
     except AttributeError:
         raise ImportError