setup.py

#! /usr/bin/env python
#
# Copyright (C) 2013 Dan McCloy <drmccloy@uw.edu>
#    Re-used based on mne-python code.

import os
import os.path as op
import subprocess

# we are using a setuptools namespace
import setuptools  # noqa, analysis:ignore
from numpy.distutils.core import setup

descr = """Experiment controller functions."""

DISTNAME = 'expyfun'
DESCRIPTION = descr
MAINTAINER = 'Dan McCloy'
MAINTAINER_EMAIL = 'drmccloy@uw.edu'
URL = 'http://github.com/LABSN/expyfun'
LICENSE = 'BSD (3-clause)'
DOWNLOAD_URL = 'http://github.com/LABSN/expyfun'

# version
version_file = os.path.join('expyfun', '_version.py')
with open(version_file, 'r') as fid:
    line = fid.readline()
    VERSION = line.strip().split(' = ')[1][1:-1]


def git_version():
    """Helper adapted from Numpy"""
    def _minimal_ext_cmd(cmd):
        # minimal env; LANGUAGE is used on win32
        return subprocess.Popen(cmd, stdout=subprocess.PIPE,
                                ).communicate()[0]
    GIT_REVISION = 'Unknown'
    if os.path.exists('.git'):
        try:
            out = _minimal_ext_cmd(['git', 'rev-parse', 'HEAD'])
            GIT_REVISION = out.decode('utf-8').strip()
        except OSError:
            pass
    return GIT_REVISION[:7]


FULL_VERSION = VERSION + '+' + git_version()


def write_version(version):
    with open(version_file, 'w') as fid:
        fid.write('__version__ = \'{0}\'\n'.format(version))


def package_tree(pkgroot):
    """Get the submodule list."""
    # Adapted from VisPy
    path = op.dirname(__file__)
    subdirs = [op.relpath(i[0], path).replace(op.sep, '.')
               for i in os.walk(op.join(path, pkgroot))
               if '__init__.py' in i[2]]
    return sorted(subdirs)


def setup_package(script_args=None):
    """Actually invoke the setup call"""
    if os.path.exists('MANIFEST'):
        os.remove('MANIFEST')
    with open('README.rst') as fid:
        long_description = fid.read()
    kwargs = dict(
        name=DISTNAME,
        maintainer=MAINTAINER,
        include_package_data=True,
        maintainer_email=MAINTAINER_EMAIL,
        description=DESCRIPTION,
        license=LICENSE,
        url=URL,
        version=FULL_VERSION,
        download_url=DOWNLOAD_URL,
        long_description=long_description,
        python_requires=">=3.7",
        zip_safe=False,  # the package can run out of an .egg file
        classifiers=['Intended Audience :: Science/Research',
                     'Intended Audience :: Developers',
                     'License :: OSI Approved',
                     'Programming Language :: Python',
                     'Topic :: Software Development',
                     'Topic :: Scientific/Engineering',
                     'Operating System :: Microsoft :: Windows',
                     'Operating System :: POSIX',
                     'Operating System :: Unix',
                     'Operating System :: MacOS'],
        platforms='any',
        packages=package_tree('expyfun'),
        package_data={'expyfun': [os.path.join('data', '*')]},
        scripts=[])
    if script_args is not None:
        kwargs['script_args'] = script_args
    try:
        write_version(FULL_VERSION)
        setup(**kwargs)
    finally:
        write_version(VERSION)


if __name__ == '__main__':
    setup_package()