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/master/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={"blib2to3": ["*.txt"], "black": ["py.typed"]},
  68     python_requires=">=3.6.2",
  69     zip_safe=False,
  70     install_requires=[
  71         "click>=7.1.2",
  72         "appdirs",
  73         "toml>=0.10.1",
  74         "typed-ast>=1.4.2; python_version < '3.8'",
  75         "regex>=2020.1.8",
  76         "pathspec>=0.6, <1",
  77         "dataclasses>=0.6; python_version < '3.7'",
  78         "typing_extensions>=3.7.4; python_version < '3.8'",
  79         "mypy_extensions>=0.4.3",
  80     ],
  81     extras_require={
  82         "d": ["aiohttp>=3.3.2", "aiohttp-cors"],
  83         "colorama": ["colorama>=0.4.3"],
  84         "python2": ["typed-ast>=1.4.2"],
  85     },
  86     test_suite="tests.test_black",
  87     classifiers=[
  88         "Development Status :: 4 - Beta",
  89         "Environment :: Console",
  90         "Intended Audience :: Developers",
  91         "License :: OSI Approved :: MIT License",
  92         "Operating System :: OS Independent",
  93         "Programming Language :: Python",
  94         "Programming Language :: Python :: 3.6",
  95         "Programming Language :: Python :: 3.7",
  96         "Programming Language :: Python :: 3.8",
  97         "Programming Language :: Python :: 3.9",
  98         "Programming Language :: Python :: 3 :: Only",
  99         "Topic :: Software Development :: Libraries :: Python Modules",
 100         "Topic :: Software Development :: Quality Assurance",
 101     ],
 102     entry_points={
 103         "console_scripts": [
 104             "black=black:patched_main",
 105             "blackd=blackd:patched_main [d]",
 106             "black-primer=black_primer.cli:main",
 107         ]
 108     },
 109 )