setup.py

from setuptools import setup
from distutils.cmd import Command

import os
import sys
import subprocess


class IntegrationTests(Command):
    description = "Command to run integration tests"
    user_options = []

    def initialize_options(self):
        pass

    def finalize_options(self):
        pass

    def run(self):
        tests_basedir = os.path.join(os.path.dirname(__file__), 'tests', 'integration')
        print('Running integration tests in %s' % tests_basedir)
        tests = []
        for entry in sorted(os.listdir(tests_basedir)):
            if not os.path.isdir(os.path.join(tests_basedir, entry)):
                continue
            tests.append(entry)
        for test in tests:
            print()
            print('Starting integration test: %s' % test)
            test_basedir = os.path.join(tests_basedir, test)
            runme_path = os.path.join(test_basedir, 'runme.sh')
            if not os.path.exists(runme_path):
                print('SKIPPED...could not find runme.sh script')
                continue
            try:
                print('Running %s' % runme_path)
                # This allows the output to make sense in the case of a failure
                sys.stdout.flush()
                retcode = subprocess.call(runme_path)
            except Exception as e:
                print('FAILED with exception: %s' % str(e))
                sys.exit(1)
            if retcode != 0:
                print('FAILED with exit code: %d' % retcode)
                sys.exit(1)
            else:
                print('SUCCESS')


# Read the README.md content for the long description
with open(os.path.join(os.path.abspath(os.path.dirname(__file__)), 'README.md')) as f:
    long_description = f.read()

setup(
    name='deploy-config-generator',
    version='2.29.0',
    url='https://github.com/ApplauseOSS/deploy-config-generator',
    license='MIT',
    description='Utility to generate service deploy configurations',
    long_description=long_description,
    long_description_content_type='text/markdown',
    author='Applause',
    author_email='ops@applause.com',
    test_suite='tests.unit',
    zip_safe=False,
    # Packages are specified manually to prevent the 'tests' dir from being installed
    packages=[
        'deploy_config_generator',
        'deploy_config_generator.output',
    ],
    python_requires=">=3",
    install_requires=[
        # Jinja2 2.11 fixes a bug around dynamic templating
        'Jinja2>=2.11',
        'PyYAML',
    ],
    entry_points={
        'console_scripts': [
            'deploy-config-generator = deploy_config_generator.__main__:main',
            'app-config = deploy_config_generator.__main__:main',
        ]
    },
    cmdclass={
        'integration': IntegrationTests,
    }
)