setup.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import codecs

from setuptools import setup, find_packages

import os
import re

###############################################################################

NAME = "simages"
PACKAGES = find_packages(where="src", exclude=["*duplicate_images*"])
HERE = os.path.abspath(os.path.dirname(__file__))

ENTRY_POINTS = {
    "console_scripts": [
        "simages = simages.main:cli",
        "simages-show = simages.main:main",
    ]
}


def read(*parts):
    """
    Build an absolute path from *parts* and and return the contents of the
    resulting file.  Assume UTF-8 encoding.
    """
    with codecs.open(os.path.join(HERE, *parts), "rb", "utf-8") as f:
        return f.read()


META_PATH = os.path.join("src", NAME, "__init__.py")
META_FILE = read(META_PATH)
KEYWORDS = ["images", "photos", "duplicates", "preprocessing", "similar data"]


def find_meta(meta):
    """
    Extract __*meta*__ from META_FILE.
    """
    meta_match = re.search(
        r"^__{meta}__ = ['\"]([^'\"]*)['\"]".format(meta=meta), META_FILE, re.M
    )
    if meta_match:
        return meta_match.group(1)
    raise RuntimeError("Unable to find __{meta}__ string.".format(meta=meta))


URL = find_meta("url")
PROJECT_URLS = {
    "Documentation": URL,
    "Bug Tracker": "https://github.com/hynek/argon2_cffi/issues",
    "Source Code": "https://github.com/hynek/argon2_cffi",
}
CLASSIFIERS = [
    "Intended Audience :: Developers",
    "License :: OSI Approved :: MIT License",
    "Intended Audience :: Education",
    "Intended Audience :: Science/Research",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.6",
    "Topic :: Scientific/Engineering :: Mathematics",
    "Topic :: Software Development :: Libraries :: Python Modules",
    "Topic :: Software Development :: Libraries",
]

PYTHON_REQUIRES = ">= 3.6"
INSTALL_REQUIRES = ["numpy", "scipy", "torch>=1.0", "torchvision>=0.3", "Pillow", "closely",
        "flask",
        "jinja2>=2.10",
        "more-itertools",
        "pymongo==3.7.0",
        "bokeh",
        "Flask-Cors",
        "dnspython",
        "Werkzeug",
        "python-magic",
        "termcolor",
        "scikit-learn",
]
EXTRAS_REQUIRE = {
    "docs": ["sphinx"],
    "tests": ["coverage", "pytest"],
}
EXTRAS_REQUIRE["dev"] = (
    EXTRAS_REQUIRE["tests"]
    + EXTRAS_REQUIRE["docs"]
    + ["wheel", "pre-commit"]
)

VERSION = find_meta("version")


# README.rst
if os.path.exists(os.path.join(HERE, "README.rst")):
    LONG = (
        read("README.rst")
        + "\n\n"
        + "Release Information\n"
        + "===================\n\n"
        + re.search(
            r"(\d+.\d.\d \(.*?\)\n.*?)\n\n\n----\n\n\n", read("CHANGELOG.rst"), re.S
        ).group(1)
        + "\n\n`Full changelog "
        + "<{url}en/stable/changelog.html>`_.\n\n".format(url=URL)
        + read("AUTHORS.rst")
    )
else:
    # README.md
    with open(os.path.join(HERE, "README.md"), encoding="utf-8") as f:
        LONG = f.read()


setup(
    name=NAME,
    version=find_meta("version"),
    author=find_meta("author"),
    author_email=find_meta("email"),
    maintainer=find_meta("author"),
    maintainer_email=find_meta("email"),
    description=find_meta("description"),
    license=find_meta("license"),
    keywords=KEYWORDS,
    url=URL,
    packages=PACKAGES,
    long_description=LONG,
    long_description_content_type="text/markdown",
    classifiers=CLASSIFIERS,
    install_requires=INSTALL_REQUIRES,
    extras_require=EXTRAS_REQUIRE,
    python_requires=PYTHON_REQUIRES,
    include_package_data=True,
    package_dir={"": "src"},
    entry_points=ENTRY_POINTS,
    zip_safe=False,
)