setup.py

   1 # Copyright (C) 2020 Łukasz Langa
   2 from setuptools import setup
   3 import sys
   4 import os
   5
   6 assert sys.version_info >= (3, 6, 2), "black requires Python 3.6.2+"
   7 from pathlib import Path  # noqa E402
   8
   9 CURRENT_DIR = Path(__file__).parent
  10 sys.path.insert(0, str(CURRENT_DIR))  # for setuptools.build_meta
  11
  12
  13 def get_long_description() -> str:
  14     return (
  15         (CURRENT_DIR / "README.md").read_text(encoding="utf8")
  16         + "\n\n"
  17         + (CURRENT_DIR / "CHANGES.md").read_text(encoding="utf8")
  18     )
  19
  20
  21 USE_MYPYC = False
  22 # To compile with mypyc, a mypyc checkout must be present on the PYTHONPATH
  23 if len(sys.argv) > 1 and sys.argv[1] == "--use-mypyc":
  24     sys.argv.pop(1)
  25     USE_MYPYC = True
  26 if os.getenv("BLACK_USE_MYPYC", None) == "1":
  27     USE_MYPYC = True
  28
  29 if USE_MYPYC:
  30     mypyc_targets = [
  31         "src/black/__init__.py",
  32         "src/blib2to3/pytree.py",
  33         "src/blib2to3/pygram.py",
  34         "src/blib2to3/pgen2/parse.py",
  35         "src/blib2to3/pgen2/grammar.py",
  36         "src/blib2to3/pgen2/token.py",
  37         "src/blib2to3/pgen2/driver.py",
  38         "src/blib2to3/pgen2/pgen.py",
  39     ]
  40
  41     from mypyc.build import mypycify
  42
  43     opt_level = os.getenv("MYPYC_OPT_LEVEL", "3")
  44     ext_modules = mypycify(mypyc_targets, opt_level=opt_level)
  45 else:
  46     ext_modules = []
  47
  48 setup(
  49     name="black",
  50     use_scm_version={
  51         "write_to": "src/_black_version.py",
  52         "write_to_template": 'version = "{version}"\n',
  53     },
  54     description="The uncompromising code formatter.",
  55     long_description=get_long_description(),
  56     long_description_content_type="text/markdown",
  57     keywords="automation formatter yapf autopep8 pyfmt gofmt rustfmt",
  58     author="Łukasz Langa",
  59     author_email="lukasz@langa.pl",
  60     url="https://github.com/psf/black",
  61     project_urls={"Changelog": "https://github.com/psf/black/blob/main/CHANGES.md"},
  62     license="MIT",
  63     py_modules=["_black_version"],
  64     ext_modules=ext_modules,
  65     packages=["blackd", "black", "blib2to3", "blib2to3.pgen2", "black_primer"],
  66     package_dir={"": "src"},
  67     package_data={
  68         "blib2to3": ["*.txt"],
  69         "black": ["py.typed"],
  70         "black_primer": ["primer.json"],
  71     },
  72     python_requires=">=3.6.2",
  73     zip_safe=False,
  74     install_requires=[
  75         "click>=7.1.2",
  76         "appdirs",
  77         "toml>=0.10.1",
  78         "typed-ast>=1.4.2; python_version < '3.8'",
  79         "regex>=2020.1.8",
  80         "pathspec>=0.8.1, <1",
  81         "dataclasses>=0.6; python_version < '3.7'",
  82         "typing_extensions>=3.7.4; python_version < '3.8'",
  83         "mypy_extensions>=0.4.3",
  84     ],
  85     extras_require={
  86         "d": ["aiohttp>=3.6.0", "aiohttp-cors>=0.4.0"],
  87         "colorama": ["colorama>=0.4.3"],
  88         "python2": ["typed-ast>=1.4.2"],
  89     },
  90     test_suite="tests.test_black",
  91     classifiers=[
  92         "Development Status :: 4 - Beta",
  93         "Environment :: Console",
  94         "Intended Audience :: Developers",
  95         "License :: OSI Approved :: MIT License",
  96         "Operating System :: OS Independent",
  97         "Programming Language :: Python",
  98         "Programming Language :: Python :: 3.6",
  99         "Programming Language :: Python :: 3.7",
 100         "Programming Language :: Python :: 3.8",
 101         "Programming Language :: Python :: 3.9",
 102         "Programming Language :: Python :: 3 :: Only",
 103         "Topic :: Software Development :: Libraries :: Python Modules",
 104         "Topic :: Software Development :: Quality Assurance",
 105     ],
 106     entry_points={
 107         "console_scripts": [
 108             "black=black:patched_main",
 109             "blackd=blackd:patched_main [d]",
 110             "black-primer=black_primer.cli:main",
 111         ]
 112     },
 113 )