CI(travis): remove deprecated `sudo` keyword

[ppu.git] / setup.py

diff --git a/setup.py b/setup.py
index dd7d8635d430cb5ec74518725a31b06b55143537..3dce134013dd6d53e49e7c8fa6dcda993f9cad8f 100755 (executable)
--- a/setup.py
+++ b/setup.py
@@ -1,8 +1,24 @@
 #!/usr/bin/env python
 
-from imp import load_source
 from os.path import abspath, dirname, join
 from setuptools import setup
+import sys
+
+if sys.version_info[:2] == (2, 7):
+    from imp import load_source
+
+elif sys.version_info >= (3, 4):
+    from importlib.machinery import SourceFileLoader
+    import types
+
+    def load_source(fullname, path):
+        loader = SourceFileLoader(fullname, path)
+        loaded = types.ModuleType(loader.name)
+        loader.exec_module(loaded)
+        return loaded
+
+else:
+    raise ImportError("ppu requires Python 2.7 or 3.4+")
 
 versionpath = join(abspath(dirname(__file__)), "ppu", "__version__.py")
 ppu_version = load_source("ppu_version", versionpath)
@@ -41,6 +57,7 @@ setup(
         'Programming Language :: Python :: 3.4',
         'Programming Language :: Python :: 3.5',
         'Programming Language :: Python :: 3.6',
+        'Programming Language :: Python :: 3.7',
     ],
     packages=['ppu'],
     scripts=[