ensure that the python version used is supported

1 files changed, 19 insertions, 0 deletions

diff --git a/setup.py b/setup.py
index 02ea588ce..1da0ba793 100644
--- a/setup.py
+++ b/setup.py
@@ -123,6 +123,25 @@ def get_extension_folder(ext):
     folder = difflib.get_close_matches(ext, maybe)[0]
     return folder
 
+# make sure that setup.py is run with an allowed python version
+def check_allowed_python_version():
+    import re
+    pattern = "'Programming Language :: Python :: (\d+)\.(\d+)'"
+    supported = []
+    with open(__file__) as setup:
+        for line in setup.readlines():
+            found = re.search(pattern, line)
+            if found:
+                major = int(found.group(1))
+                minor = int(found.group(2))
+                supported.append( (major, minor) )
+    this_py = sys.version_info[:2]
+    if this_py not in supported:
+        print("only these python versions are supported:", supported)
+        sys.exit(1)
+
+check_allowed_python_version()
+
 # Declare options
 OPTION_DEBUG = has_option("debug")
 OPTION_RELWITHDEBINFO = has_option('relwithdebinfo')