setup.py

import os
from distutils.util import convert_path

from setuptools import find_packages, setup

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

namespace = {}
version_file = convert_path("src/taskgraph/__init__.py")
with open(version_file) as fh:
    exec(fh.read(), namespace)

with open(os.path.join(project_dir, "requirements/base.in")) as fp:
    requirements = fp.read().splitlines()

with open(os.path.join(project_dir, "README.rst")) as fh:
    long_description = fh.read()

setup(
    name="taskcluster-taskgraph",
    version=namespace["__version__"],
    description="Build taskcluster taskgraphs",
    long_description=long_description,
    url="https://github.com/taskcluster/taskgraph",
    packages=find_packages("src"),
    package_dir={"": "src"},
    install_requires=requirements,
    extras_require={
        "load-image": ["zstandard"],
    },
    classifiers=[
        "Development Status :: 5 - Production/Stable",
        "Environment :: Console",
        "License :: OSI Approved :: Mozilla Public License 2.0 (MPL 2.0)",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
        "Programming Language :: Python :: 3.10",
        "Programming Language :: Python :: 3.11",
        "Programming Language :: Python :: 3.12",
        "Topic :: Software Development",
    ],
    entry_points={"console_scripts": ["taskgraph = taskgraph.main:main"]},
    package_data={
        "taskgraph": [
            "run-task/run-task",
            "run-task/fetch-content",
            "run-task/hgrc",
            "run-task/robustcheckout.py",
        ],
        "taskgraph.test": ["automationrelevance.json"],
    },
)