dgp/setup.py at master · TRI-ML/dgp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
# Copyright 2025 Toyota Research Institute. All rights reserved.
import importlib.util
import logging
import os
from pathlib import Path

from setuptools import find_packages, setup
from setuptools.command.build_py import build_py
from setuptools.command.develop import develop
from setuptools.command.install import install

logger = logging.getLogger(__file__)

_PACKAGE = "dgp"
_ROOT_DIRPATH = Path(__file__).parent.absolute()

_init_py = _ROOT_DIRPATH / _PACKAGE / "__init__.py"
_spec = importlib.util.spec_from_file_location(_PACKAGE, _init_py)
_module = importlib.util.module_from_spec(_spec)
_spec.loader.exec_module(_module)

__version__ = _module.__version__

# Specify the dev version for development.
_DEV_VERSION = str(os.environ.get("DGP_DEV_VERSION", ""))

_VERSION = f"{__version__}.{_DEV_VERSION}" if _DEV_VERSION else __version__

with open("requirements.txt", encoding="utf-8") as f:
    requirements = f.read().splitlines()

with open("requirements-dev.txt", encoding="utf-8") as f:
    requirements_dev = f.read().splitlines()

install_requires = requirements + [
    "protobuf>=6.30.1",
]
setup_requires = [
    "protobuf>=6.30.1",
    "grpcio>=1.70.0",
    "grpcio-tools>=1.70.0",
]


def _build_py():
    from grpc_tools import command

    command.build_package_protos(_ROOT_DIRPATH)


class _CustomBuildPyCommand(build_py):
    def run(self):
        _build_py()
        build_py.run(self)


class _CustomInstallCommand(install):
    def run(self):
        _build_py()
        install.run(self)


class _CustomDevelopCommand(develop):
    def run(self):
        _build_py()
        develop.run(self)


packages = find_packages(exclude=["tests"])

setup(
    name=_PACKAGE,
    version=_VERSION,
    description="Dataset Governance Policy (DGP) for Autonomous Vehicle ML datasets.",
    long_description=open("README.md", encoding="utf-8").read(),
    long_description_content_type="text/markdown",
    author="Toyota Research Institute",
    author_email="ml@tri.global",
    url="https://github.com/TRI-ML/dgp",
    packages=packages,
    entry_points={
        "console_scripts": [
            "dgp_cli=dgp.cli:cli",
        ],
    },
    include_package_data=True,
    install_requires=install_requires,
    setup_requires=setup_requires,
    extras_require={
        "dev": requirements_dev,
    },
    zip_safe=False,
    python_requires=">=3.10",
    cmdclass={
        "build_py": _CustomBuildPyCommand,
        "install": _CustomInstallCommand,
        "develop": _CustomDevelopCommand,
    },
)