setup.py

import ast
from ast import NodeVisitor
import os
from setuptools import setup, find_packages


class VersionExtractor(NodeVisitor):
    def __init__(self):
        super().__init__()
        self.version = None

    def visit_Assign(self, node):
        if hasattr(node.targets[0], "id") and node.targets[0].id == "__version__":
            self.version = node.value.s


def parse_version():

    with open(os.path.join("gomatch", "__init__.py"), "r") as f:
        content = f.read()

    tree = ast.parse(content)
    visitor = VersionExtractor()
    visitor.visit(tree)
    return visitor.version


def long_description():
    with open("README.md", "r") as f:
        return f.read()


setup(
    name="gomatch",
    version=parse_version(),
    license="MIT",
    description="GoMatch: a geometric-only matcher for visual localization",
    long_description=long_description(),
    long_description_content_type="text/markdown",
    install_requires=[
        "bpnpnet @ git+https://github.com/SergioRAgostinho/bpnpnet.git@package#egg=bpnpnet",
        "numpy",
        "opencv-python",
        "Pillow",
        "pytorch-lightning",
        "scipy",
        "torch==1.7.1",
        "torch-scatter==2.0.6",
        "tqdm",
    ],
    extras_require=dict(
        full=[
            "wget",
            "colmap @ git+https://github.com/SergioRAgostinho/colmap.git@python-packaging#egg=colmap",
        ]
    ),
    author="Qunjie Zhou & Sérgio Agostinho",
    author_email="gomatch@googlegroups.com",
    packages=find_packages(),
    python_requires=">=3.7",
    classifiers=[
        "License :: OSI Approved :: MIT License",
        "Operating System :: OS Independent",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.7",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
        "Programming Language :: Python :: 3.10",
        "Programming Language :: Python :: 3 :: Only",
        "Topic :: Scientific/Engineering",
    ],
)