X-Git-Url: https://git.phdru.name/?p=ppu.git;a=blobdiff_plain;f=setup.py;h=3dce134013dd6d53e49e7c8fa6dcda993f9cad8f;hp=d2a518d1045bc69b223e426552f0cf8a12213258;hb=8d50f5116b9fdf53cede21f7e5b15dcd15f7e5dd;hpb=5d25acd019dc55806da2b2d8c5609877696e7d78

diff --git a/setup.py b/setup.py
index d2a518d..3dce134 100755
--- 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)