setup.py

import sys

from setuptools import find_packages, setup


def main():
    setup_requires = ["setuptools_scm"]

    install_requires = ["numpy", "scipy", "click", "tabulate", "svgwrite"]
    if sys.platform == "win32":
        install_requires.append("colorama")

    extras_require = {}
    # extras_require['draw'] = ['svgwrite']
    extras_require["tests"] = ["pytest", "pytest-sugar", "pytest-cov"]

    setup(
        name="Hammlet",
        description="Hybridization Models Maximum Likelihood Estimator",
        url="https://github.com/ctlab/hammlet",
        author="Konstantin Chukharev",
        author_email="lipen00@gmail.com",
        license="GNU GPLv3",
        python_requires=">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*, <4",
        package_dir={"": "src"},
        packages=find_packages("src"),
        use_scm_version={
            "write_to": "src/hammlet/version.py",
            "version_scheme": "post-release",
            # 'local_scheme': lambda _: '',
            "local_scheme": "dirty-tag",
        },
        install_requires=install_requires,
        setup_requires=setup_requires,
        extras_require=extras_require,
        tests_require=extras_require["tests"],
        entry_points={"console_scripts": ["hammlet = hammlet.cli:cli"]},
    )


if __name__ == "__main__":
    main()