setup.py

# -*- coding: utf-8 -*-
"""Configuration and metadata for setuptools packaging."""

from __future__ import absolute_import, division, print_function, unicode_literals

import re
from io import open
from setuptools import setup, find_packages
try:
    from urllib.parse import urlparse, urljoin
except ImportError:
    from urlparse import urlparse, urljoin


def rewrite_relative_images(base_url, text):
    """Rewrite project-relative image links to absolute URLs for PyPI."""
    return re.sub(r"(?<=\.\. image:: )(?P<relative_path>\S+)",
                  lambda match: urljoin(base_url, urlparse(match.group("relative_path")).path),
                  text)


def main():
    """Entrypoint for setuptools."""
    with open("src/asciidag/__init__.py", encoding="utf-8") as init:
        metadata = dict(re.findall(r"__([a-z]+)__\s*=\s*['\"]([^'\"]*)['\"]",
                                   init.read()))

    if not metadata:
        raise RuntimeError("Cannot find metadata information")

    with open("README.rst", encoding="utf-8") as readme_file:
        relative_path = "{}/master/".format(urlparse(metadata["url"]).path)
        image_base_url = urljoin("https://raw.githubusercontent.com/", relative_path)
        readme = rewrite_relative_images(image_base_url, readme_file.read())

    tests_require = [
        "astroid<2; python_version<'2.7' or (python_version>='3' and python_version<'3.4')",
        "docutils",
        "flake8; (python_version>='2.7' and python_version<'3') or python_version>='3.4'",
        "flake8<3.8; python_version<'2.7' or (python_version>='3' and python_version<'3.4')",
        "flake8_docstrings",
        "isort<5",
        "more-itertools<6",
        "pydocstyle<4",
        "pylint-venv<2",
        "pylint<2; python_version<'3.5'",
        "pylint>=2; python_version>='3.5'",
        "pyparsing<3",
        "pytest-cov; (python_version>='2.7' and python_version<'3') or python_version>='3.4'",
        "pytest-cov<2.6; python_version<'2.7' or (python_version>='3' and python_version<'3.4')",
        "pytest-flake8; (python_version>='2.7' and python_version<'3') or python_version>='3.4'",
        "pytest-flake8<0.9; python_version<'2.7' or (python_version>='3' and python_version<'3.4')",
        "pytest-pylint<0.10; python_version<'2.7' or (python_version>='3' and python_version<'3.4')",
        "pytest-pylint<0.15; (python_version>='2.7' and python_version<'3') or python_version>='3.4'",
        "pytest<3; python_version<'2.7' or (python_version>='3' and python_version<'3.4')",
        "pytest<5; (python_version>='2.7' and python_version<'3') or python_version>='3.4'",
        "typed_ast<1.3; python_version>='3' and python_version<'3.5'",
        "typing<3.7; python_version>='3' and python_version<'3.5'",
        "zipp<3",
    ]

    setup(
        name=metadata["title"],
        version=metadata["version"],
        description="Draw DAGs (directed acyclic graphs) as ASCII art, à la git log --graph",
        long_description=readme,
        long_description_content_type="text/x-rst",
        author=metadata["author"],
        author_email=metadata["email"],
        url=metadata["url"],
        packages=find_packages(where="src"),
        package_dir={"": "src"},
        include_package_data=True,
        zip_safe=False,
        setup_requires=[
            "pytest-runner",
        ],
        install_requires=[
            "enum34; python_version<'3.4'",
        ],
        tests_require=tests_require,
        extras_require={
            "dev": tests_require + [
                "bump2version",
                "twine",
            ],
        },
        license=metadata["license"],
        classifiers=[
            "Development Status :: 3 - Alpha",
            "Environment :: Console",
            "Intended Audience :: Developers",
            "License :: OSI Approved :: GNU General Public License v2 (GPLv2)",
            "Programming Language :: Python :: 2",
            "Programming Language :: Python :: 2.6",
            "Programming Language :: Python :: 2.7",
            "Programming Language :: Python :: 3",
            "Programming Language :: Python :: 3.0",
            "Programming Language :: Python :: 3.1",
            "Programming Language :: Python :: 3.2",
            "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",
            "Programming Language :: Python :: 3.8",
            "Programming Language :: Python :: 3.9",
            "Programming Language :: Python :: Implementation :: CPython",
            "Programming Language :: Python :: Implementation :: PyPy",
            "Topic :: Software Development",
        ],
    )


if __name__ == "__main__":
    main()