setup.py

import os
import re
import sys
from setuptools import setup, find_packages
from setuptools.command.test import test as TestCommand

DIR = os.path.dirname(__file__)

with open(os.path.join(DIR, "README.rst")) as f:
    readme = f.read().splitlines()

with open(os.path.join(DIR, "lib/replisome/version.py")) as f:
    m = re.search(r"""^VERSION\s*=\s*["']([^'"]+)""", f.read(), re.MULTILINE)
    assert m, "version not found"
    version = m.group(1)

classifiers = """
Development Status :: 3 - Alpha
License :: OSI Approved :: BSD License
Programming Language :: Python :: 2
Programming Language :: Python :: 2.7
Programming Language :: Python :: 3
Programming Language :: Python :: 3.5
Topic :: Database
"""


class PyTest(TestCommand):
    """
    Class to integrate py.test with setuptools.
    """
    user_options = [('pytest-args=', 'a', "Arguments to pass to pytest")]

    def initialize_options(self):
        TestCommand.initialize_options(self)
        self.pytest_args = ''

    def run_tests(self):
        import shlex
        # import here, cause outside the eggs aren't loaded
        import pytest
        errno = pytest.main(shlex.split(self.pytest_args))
        sys.exit(errno)


def parse_requirements(fn):
    """
    Parse a requirements file into a list of package specs.
    """
    with open(fn) as f:
        rv = []
        for line in f:
            line = line.strip()
            if not line or line.startswith('#'):
                continue
            rv.append(line)
        return rv


setup(
    name='replisome',
    package_dir={'': 'lib'},
    packages=find_packages('lib'),
    version=version,
    description=readme[0],
    long_description='\n'.join(readme[2:]).lstrip(),
    author='Daniele Varrazzo',
    author_email='daniele.varrazzo@gmail.com',
    url='https://github.com/GambitResearch/replisome',
    keywords=['database', 'replication', 'PostgreSQL'],
    classifiers=[x for x in classifiers.strip().splitlines()],
    install_requires=['six', 'PyYAML', 'psycopg2>=2.7'],
    tests_require=parse_requirements(
        os.path.join(DIR, 'tests/pytests/requirements.txt')),
    zip_safe=False,
    entry_points={
        'console_scripts': [
            'replisome = replisome.cli:entry_point',
        ],
    },
    cmdclass={'test': PyTest},
)