setup.py

import os
import subprocess
from setuptools import setup

##########################
VERSION = "1.0.6"
ISRELEASED = False
__version__ = VERSION
##########################


def read_readme(fname):
    return open(os.path.join(os.path.dirname(__file__), fname)).read()


##########################
# Function for determining current git commit
##########################


def git_version():
    # Return the git revision as a string
    # copied from numpy setup.py
    def _minimal_ext_cmd(cmd):
        # construct minimal environment
        env = {}
        for k in ['SYSTEMROOT', 'PATH']:
            v = os.environ.get(k)
            if v is not None:
                env[k] = v
        # LANGUAGE is used on win32
        env['LANGUAGE'] = 'C'
        env['LANG'] = 'C'
        env['LC_ALL'] = 'C'
        out = subprocess.Popen(
            cmd, stdout=subprocess.PIPE, env=env).communicate()[0]
        return out

    try:
        out = _minimal_ext_cmd(['git', 'rev-parse', 'HEAD'])
        GIT_REVISION = out.strip().decode('ascii')
    except OSError:
        GIT_REVISION = 'Unknown'

    return GIT_REVISION


##########################
# Function for writing version.py (this will be copied to the install directory)
##########################

ensembler_version_filepath = 'ensembler/version.py'


def write_version_py(filename=ensembler_version_filepath):
    cnt = """# THIS FILE IS GENERATED FROM ENSEMBLER SETUP.PY
short_version = '%(version)s'
version = '%(version)s'
full_version = '%(full_version)s'
git_revision = '%(git_revision)s'
release = %(isrelease)s

if not release:
    version = full_version
"""
    # Adding the git rev number needs to be done inside write_version_py(),
    # otherwise the import of numpy.version messes up the build under Python 3.
    FULLVERSION = VERSION
    if os.path.exists('.git'):
        GIT_REVISION = git_version()
    else:
        GIT_REVISION = 'Unknown'

    if not ISRELEASED:
        FULLVERSION += '.dev-' + GIT_REVISION[:7]

    a = open(filename, 'w')
    try:
        a.write(cnt % {'version': VERSION,
                       'full_version': FULLVERSION,
                       'git_revision': GIT_REVISION,
                       'isrelease': str(ISRELEASED)})
    finally:
        a.close()


##########################
# Find package data
##########################


def find_package_data(dir_to_search=None):
    package_data = []
    for dir, subdirs, files in os.walk(dir_to_search):
        for file in files:
            if file[0] != '.':
                filepath = os.path.join(dir, file).replace(dir_to_search + os.path.sep, '')
                package_data.append(filepath)
    return package_data


##########################
# Setup
##########################

write_version_py()
setup(
    name = 'ensembler',
    version = __version__,
    author = 'Daniel L Parton',
    author_email = 'daniel.parton@choderalab.org',
    description = 'Software pipeline for automating omics-scale protein modeling and simulation setup',
    license='GPLv2',
    long_description = read_readme('README.md'),
    packages = [
        'ensembler',
        'ensembler.cli_commands',
        'ensembler.tools',
        'ensembler.tests',
    ],
    package_data = {
        'ensembler': find_package_data(dir_to_search='ensembler'),
        'ensembler.tests': find_package_data(dir_to_search=os.path.join('ensembler', 'tests')),
    },

    entry_points = {'console_scripts':
        [
            'ensembler = ensembler.cli:main'
        ]
    },
)

# Delete version.py file from this directory
os.remove(ensembler_version_filepath)