X-Git-Url: https://git.phdru.name/?p=cookiecutter.git;a=blobdiff_plain;f=project_template%2Fsetup.py;h=8297efa6656c7c47a32c9c8d2f7051cf30a50c83;hp=2ca49ae7358659578027b365a3afabbd58715527;hb=30a18d3f2facb32032ab8036b212111f6f715131;hpb=ae876aff04182070cab37ec32feda22af8603ee7

diff --git a/project_template/setup.py b/project_template/setup.py
index 2ca49ae..8297efa 100755
--- a/project_template/setup.py
+++ b/project_template/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("{project} requires Python 2.7 or 3.4+")
 
 versionpath = join(abspath(dirname(__file__)), '{project}', '__version__.py')
 {project}_version = load_source('{project}_version', versionpath)