@@ -12,23 +12,12 @@ Version = namedtuple('Version', ('release', 'dev', 'labels'))

 __all__ = []

 package_root = os.path.dirname(os.path.realpath(__file__))

+package_name = os.path.basename(package_root)

 distr_root = os.path.dirname(package_root)

 STATIC_VERSION_FILE = '_static_version.py'

-def write_version(fname, version):

-    # This could be a hard link, so try to delete it first.  Is there any way

-    # to do this atomically together with opening?

-    try:

-        os.remove(fname)

-    except OSError:

-        pass

-    with open(fname, 'w') as f:

-        f.write("# This file has been created by setup.py.\n"

-                "version = '{}'\n".format(version))

-

-

 def get_version(version_file=STATIC_VERSION_FILE):

     version_info = {}

     with open(os.path.join(package_root, version_file), 'rb') as f:

@@ -140,20 +129,38 @@ def get_version_from_git_archive(version_info):

         return Version('unknown', dev=None, labels=[f'g{git_hash}'])

-def cmdclass(version, package_name):

+version = get_version()

+

+# The following section defines a module global 'cmdclass',

+# which can be used from setup.py. The 'package_name' and

+# 'version' module globals are used (but not modified).

+

+def _write_version(fname):

+    # This could be a hard link, so try to delete it first.  Is there any way

+    # to do this atomically together with opening?

+    try:

+        os.remove(fname)

+    except OSError:

+        pass

+    with open(fname, 'w') as f:

+        f.write("# This file has been created by setup.py.\n"

+                "version = '{}'\n".format(version))

+

+

+class _build(build_orig):

+    def run(self):

+        super().run()

+        _write_version(os.path.join(self.build_lib, package_name,

+                                    STATIC_VERSION_FILE),

+                      version=version)

+

-    class build(build_orig):

-        def run(self):

-            super().run()

-            write_version(os.path.join(self.build_lib, package_name,

-                                       STATIC_VERSION_FILE),

-                          version=version)

+class _sdist(sdist_orig):

+    def make_release_tree(self, base_dir, files):

+        super().make_release_tree(base_dir, files)

+        _write_version(os.path.join(base_dir, package_name,

+                                    STATIC_VERSION_FILE),

+                       version=version)

-    class sdist(sdist_orig):

-        def make_release_tree(self, base_dir, files):

-            super().make_release_tree(base_dir, files)

-            write_version(os.path.join(base_dir, package_name,

-                                       STATIC_VERSION_FILE),

-                          version=version)

-    return dict(sdist=sdist, build=build)

+cmdclass = dict(sdist=_sdist, build=_build)

@@ -1,7 +1,6 @@

 #!/usr/bin/env python3

 # -*- coding: utf-8 -*-

-import os

 from setuptools import setup, find_packages

 import sys

@@ -11,19 +10,19 @@ if sys.version_info < (3, 6):

     sys.exit(1)

-# Load version.py module without importing 'adaptive'

-def load_version_module(package_name):

+# Loads version.py module without importing the whole package.

+def get_version_and_cmdclass(package_name):

+    import os

     from importlib.util import module_from_spec, spec_from_file_location

     spec = spec_from_file_location('version',

-                                   '{}/version.py'.format(package_name))

+                                   os.path.join(package_name, 'version.py'))

     module = module_from_spec(spec)

     spec.loader.exec_module(module)

-    return module

+    return module.version, module.cmdclass

-version_module = load_version_module('adaptive')

-version = version_module.get_version()

-cmdclass = version_module.cmdclass(version, 'adaptive')

+version, cmdclass = get_version_and_cmdclass('adaptive')

+

 install_requires = [

     'scipy',