setup.py

# -*- coding: utf-8 -*-
# Copyright (c) 2013-2014 Simon Jagoe
# All rights reserved.
#
# This software may be modified and distributed under the terms
# of the 3-clause BSD license.  See the LICENSE.txt file for details.
import os
import sys
import subprocess

from setuptools import setup


MAJOR = 0
MINOR = 9
MICRO = 0

IS_RELEASED = False

VERSION = '%d.%d.%d' % (MAJOR, MINOR, MICRO)


# Return the git revision as a string
def git_version():
    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"

    try:
        out = _minimal_ext_cmd(['git', 'rev-list', '--count', 'HEAD'])
        git_count = out.strip().decode('ascii')
    except OSError:
        git_count = '0'

    return git_revision, git_count


def write_version_py(filename='haas/_version.py'):
    template = """\
# -*- coding: utf-8 -*-
# Copyright (c) 2013-2014 Simon Jagoe
# All rights reserved.
#
# This software may be modified and distributed under the terms
# of the 3-clause BSD license.  See the LICENSE.txt file for details.
from __future__ import absolute_import, unicode_literals

# THIS FILE IS GENERATED FROM SETUP.PY
version = '{version}'
full_version = '{full_version}'
git_revision = '{git_revision}'
is_released = {is_released}

if not is_released:
    version = full_version
"""
    fullversion = VERSION
    if os.path.exists('.git'):
        git_rev, dev_num = git_version()
    elif os.path.exists('haas/_version.py'):
        # must be a source distribution, use existing version file
        try:
            from haas._version import git_revision as git_rev
            from haas._version import full_version as full_v
        except ImportError:
            raise ImportError("Unable to import git_revision. Try removing "
                              "haas/_version.py and the build "
                              "directory before building.")
        import re
        match = re.match(r'.*?\.dev(?P<dev_num>\d+)$', full_v)
        if match is None:
            dev_num = '0'
        else:
            dev_num = match.group('dev_num')
    else:
        git_rev = "Unknown"
        dev_num = '0'

    if not IS_RELEASED:
        fullversion += '.dev{0}'.format(dev_num)

    with open(filename, "wt") as fp:
        fp.write(template.format(version=VERSION,
                                 full_version=fullversion,
                                 git_revision=git_rev,
                                 is_released=IS_RELEASED))


if __name__ == "__main__":
    install_requires = ['six']
    common_requires = ['enum34', 'statistics']
    py26_requires = [
        'stevedore >= 1.0.0, < 1.10.0',
        'unittest2',
        'argparse',
        'ordereddict'] + common_requires
    py34_requires = ['stevedore >= 1.0.0, < 1.12.0']
    py33_requires = py34_requires + common_requires
    if sys.version_info < (2, 7):
        install_requires += py26_requires
    elif sys.version_info < (3, 4):
        install_requires += py33_requires
    else:
        install_requires = py34_requires

    write_version_py()
    from haas import __version__

    with open('README.rst') as fh:
        long_description = fh.read()

    setup(
        name='haas',
        version=__version__,
        url='https://github.com/scalative/haas',
        author='Simon Jagoe',
        author_email='simon@scalative.com',
        classifiers=[
            'Development Status :: 3 - Alpha',
            'Intended Audience :: Developers',
            'License :: OSI Approved :: BSD License',
            'Operating System :: MacOS',
            'Operating System :: Microsoft :: Windows',
            'Operating System :: OS Independent',
            'Operating System :: POSIX',
            'Operating System :: Unix',
            'Programming Language :: Python',
            'Programming Language :: Python :: 2',
            'Programming Language :: Python :: 2.6',
            'Programming Language :: Python :: 2.7',
            'Programming Language :: Python :: 3',
            'Programming Language :: Python :: 3.3',
            'Programming Language :: Python :: 3.4',
            'Programming Language :: Python :: 3.5',
            'Programming Language :: Python :: 3.6',
            'Programming Language :: Python :: 3.7',
            'Topic :: Software Development',
            'Topic :: Software Development :: Testing',
        ],
        description='Extensible Python Test Runner',
        long_description=long_description,
        license='BSD',
        packages=['haas', 'haas.plugins', 'haas.tests', 'haas.plugins.tests'],
        install_requires=install_requires,
        entry_points={
            'console_scripts': [
                'haas = haas.main:main',
            ],
            'haas.hooks.environment': [
                'coverage = haas.plugins.coverage:Coverage',
            ],
            'haas.discovery': [
                'default = haas.plugins.discoverer:Discoverer',
            ],
            'haas.runner': [
                'default = haas.plugins.runner:BaseTestRunner',
                'parallel = haas.plugins.parallel_runner:ParallelTestRunner',  # noqa
            ],
            'haas.result.handler': [
                'default = haas.plugins.result_handler:StandardTestResultHandler',  # noqa
                'quiet = haas.plugins.result_handler:QuietTestResultHandler',
                'verbose = haas.plugins.result_handler:VerboseTestResultHandler',  # noqa
                'timing = haas.plugins.result_handler:TimingResultHandler',
            ]
        },
        extras_require={
            ':python_version=="2.6"': py26_requires,
            ':python_version=="2.7"': py33_requires,
            ':python_version=="3.3"': py33_requires,
            ':python_version=="3.4"': py34_requires,
            ':python_version=="3.5"': py34_requires,
            ':python_version=="3.6"': py34_requires,
            ':python_version=="3.7"': py34_requires,
        },
    )