setup.py

#!/usr/bin/env python
# coding: utf8
import os
import sys

from setuptools import setup, find_packages

PREFIX = 'toloka'

setup_py_dir = os.path.dirname(__file__)
version_module_path = os.path.join(setup_py_dir, 'src', '__version__.py')

about = {}

with open(version_module_path) as f:
    exec(f.read(), about)

with open('README.md') as f:
    readme = f.read()


def get_ipython_with_version():
    if sys.version_info < (3, 8):
        return 'ipython < 8'
    elif sys.version_info < (3, 9):
        return 'ipython < 8.13'
    return 'ipython'


EXTRAS_REQUIRE = {
    'dev': [
        'aiohttp',
        'data-science-types',
        'flake8',
        'mypy',
        'pytest',
        'pytest-asyncio',
        'pytest-lazy-fixture',
        'pytest-mock',
        'pytest-timeout',
        'respx',
        'tox',
        'types-urllib3',
    ],
    'pandas': ['pandas'],
    'autoquality': ['crowd-kit >= 1.0.0'],
    's3': ['boto3 >= 1.4.7'],
    'zookeeper': ['kazoo >= 2.6.1'],
    'jupyter-metrics': ['plotly', 'ipyplot', 'jupyter-dash', get_ipython_with_version()],
}
EXTRAS_REQUIRE['all'] = sum((deps for env, deps in EXTRAS_REQUIRE.items() if env != 'dev'), [])

setup(
    name=about['__title__'],
    package_dir={PREFIX: 'src'},
    packages=['toloka', *(f'{PREFIX}.{package}' for package in find_packages('src'))],
    version=about['__version__'],
    description='Toloka API client',
    long_description=readme,
    long_description_content_type='text/markdown',
    license=about['__license__'],
    author='Vladimir Losev',
    author_email='losev@yandex-team.ru',
    python_requires='>=3.8.0',
    install_requires=[
        'attrs >= 20.3.0',
        'cattrs >= 1.9',
        'cached-property; python_version < "3.8.0"',
        'filelock >= 3.2.0',
        'requests',
        'httpx',
        'tenacity >= 7.0, < 8.3',  # https://github.com/jd/tenacity/issues/139
        'typing-extensions',
        'urllib3 >= 1.26.0',
        'simplejson',
        'docstring-parser',
        'tqdm',
    ],
    extras_require=EXTRAS_REQUIRE,
    include_package_data=True,
    project_urls={
        'Documentation': 'https://toloka.ai/docs/toloka-kit',
        'Source': 'https://github.com/Toloka/toloka-kit',
    },
    classifiers=[
        'Development Status :: 5 - Production/Stable',
        'Intended Audience :: Developers',
        'Intended Audience :: Science/Research',
        'License :: OSI Approved :: Apache Software License',
        'Operating System :: OS Independent',
        'Programming Language :: Python',
        'Programming Language :: Python :: 3',
        'Programming Language :: Python :: 3.8',
        'Programming Language :: Python :: 3.9',
        'Programming Language :: Python :: 3.10',
        'Programming Language :: Python :: 3.11',
        'Programming Language :: Python :: Implementation :: CPython',
        'Topic :: Scientific/Engineering',
        'Topic :: Software Development',
        'Typing :: Typed',
    ],
)