setup.py

import io
import os
import sys
from shutil import rmtree

from setuptools import Command, Extension, setup


__version__ = "0.0.12"
__title__ = "polf"

HERE = os.path.abspath(os.path.dirname(__file__))

LINT_EXTRAS = [
    'flake8==3.8.4',
    'flake8-print==3.1.4',
    'flake8-implicit-str-concat==0.2.0',
    'isort==5.6.4',
    'pre-commit==2.12.1',
    'yamllint==1.25.0',
]
TEST_EXTRAS = [
    'pytest==6.2.3',
]
DOC_EXTRAS = [
    'Sphinx>=3.2.1',
    'sphinx-rtd-theme==0.4.3',
]
DEV_EXTRAS = [
    'bump2version==1.0.1',
] + TEST_EXTRAS + DOC_EXTRAS

with io.open(os.path.join(HERE, "README.md"), encoding="utf-8") as f:
    LONG_DESCRIPTION = "\n" + f.read()

define_macros = []
extra_compile_args = []
extension = Extension(__title__,
                      language="c",
                      sources=[os.path.join('src', 'pypolf.c')],
                      define_macros=define_macros,
                      extra_compile_args=extra_compile_args)


class UploadCommand(Command):
    """Support setup.py upload."""

    description = "Build and publish the package."
    user_options = [
        ("test", None, "Specify if you want to test your upload to Pypi."),
    ]

    @staticmethod
    def status(s):
        """Prints things in bold."""
        sys.stdout.write("\033[1m{0}\033[0m\n".format(s))

    def initialize_options(self):
        self.test = None

    def finalize_options(self):
        pass

    def run(self):
        try:
            self.status("Removing previous builds…")
            rmtree(os.path.join(HERE, "dist"))
        except OSError:
            pass

        self.status("Building Source distribution…")
        os.system("{0} setup.py sdist".format(sys.executable))

        self.status("Uploading the package to PyPI via Twine…")
        cmd = "twine upload%s dist/*" % (
            " --repository-url https://test.pypi.org/legacy/" if self.test
            else ""
        )
        os.system(cmd)
        sys.exit()


setup(
    name=__title__,
    version=__version__,
    description='Calculate points on lines.',
    long_description=LONG_DESCRIPTION,
    long_description_content_type="text/markdown",
    license="BSD License",
    classifiers=[
        # Full list: https://pypi.python.org/pypi?%3Aaction=list_classifiers
        "License :: OSI Approved :: BSD License",
        "Operating System :: OS Independent",
        "Development Status :: 4 - Beta",
        "Programming Language :: Python",
        "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",
        "Topic :: Scientific/Engineering :: Mathematics",
        "Topic :: Scientific/Engineering :: Image Processing",
        "Topic :: Multimedia :: Graphics",
        "Intended Audience :: Developers",
        "Intended Audience :: Education",
    ],
    python_requires='>=3.6',
    zip_safe=False,
    ext_modules=[extension],
    include_package_data=True,
    cmdclass={
        "upload": UploadCommand,
    },
    extras_require={
        "dev": DEV_EXTRAS,
        "doc": DOC_EXTRAS,
        "lint": LINT_EXTRAS,
        "test": TEST_EXTRAS,
    },
)