setup.py

# Copyright 2020-present NAVER Corp. Under BSD 3-clause license

import setuptools
import warnings
import os.path as path
from glob import glob
from subprocess import check_call, SubprocessError
import os
import tempfile
import sys

HERE = path.normpath(path.dirname(__file__))


# Documentation ########################################################################################################
# setuptools only support md documentation: convert from asciidoc
def read_file(filepath):
    with open(filepath, 'r', encoding='utf-8') as f:
        return f.read()


def read_doc(filepath):
    try:  # from asciidoc to markdown
        # https://tinyapps.org/blog/201701240700_convert_asciidoc_to_markdown.html
        with tempfile.TemporaryDirectory() as tmpdirname:
            xml_filepath = path.join(tmpdirname, 'README.xml')
            md_filepath = path.join(tmpdirname, 'README.md')
            use_shell = sys.platform.startswith("win")
            check_call(['asciidoctor', '-b', 'docbook', filepath, '-o', xml_filepath], shell=use_shell)
            check_call(['pandoc', '-f', 'docbook', '-t', 'markdown_strict', xml_filepath, '-o', md_filepath],
                       shell=use_shell)
            content = read_file(md_filepath)

    except (FileNotFoundError, SubprocessError):
        warnings.warn('cannot convert asciidoc to markdown.')
        content = read_file(filepath)

    return content


readme_filepath = path.join(HERE, 'README.adoc')
long_description = read_doc(readme_filepath)

########################################################################################################################
setuptools.setup(
    # description
    name='kapture-localization',
    version="1.1.10",
    author="naverlabs",
    author_email="kapture@naverlabs.com",
    description="kapture-localization: localization pipelines based on kapture.",
    long_description=long_description,
    long_description_content_type="text/markdown",
    url="https://github.com/naver/kapture-localization/",
    classifiers=[
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: BSD License",
        "Operating System :: OS Independent",
    ],

    # dependencies
    python_requires='>=3.8',
    install_requires=[
        'kapture>=1.1.10,<1.2.0',
        'numpy>=1.16',
        'numpy-quaternion',
        'numba',
        'cvxpy>=1.1.6',
        'torch>=1.4.0',
        'tabulate>=0.8.7',
        'opencv-python',
        'build>=1.0.3',
    ],
    extras_require={
        'dev': ['pytest'],
    },
    # sources
    packages=setuptools.find_packages(),
    data_files=[(".", ["README.adoc"])],
    scripts=[file
             for dir in ["tools", 'pipeline']
             for file in glob(os.path.join(HERE, dir, '*.py'))],
)