| | 1 | """Glue for the "black" library. |
| | 2 | |
| | 3 | """ |
| | 4 | |
| | 5 | import sys |
| | 6 | |
| | 7 | from elpy.rpc import Fault |
| | 8 | # in case pkg_resources is not properly installed |
| | 9 | # (see https://github.com/jorgenschaefer/elpy/issues/1674). |
| | 10 | try: |
| | 11 | from pkg_resources import parse_version |
| | 12 | except ImportError: # pragma: no cover |
| | 13 | def parse_version(*arg, **kwargs): |
| | 14 | raise Fault("`pkg_resources` could not be imported, " |
| | 15 | "please reinstall Elpy RPC virtualenv with" |
| | 16 | " `M-x elpy-rpc-reinstall-virtualenv`", code=400) |
| | 17 | |
| | 18 | import os |
| | 19 | |
| | 20 | try: |
| | 21 | import toml |
| | 22 | except ImportError: |
| | 23 | toml = None |
| | 24 | |
| | 25 | |
| | 26 | BLACK_NOT_SUPPORTED = sys.version_info < (3, 6) |
| | 27 | |
| | 28 | try: |
| | 29 | if BLACK_NOT_SUPPORTED: # pragma: no cover |
| | 30 | black = None |
| | 31 | else: |
| | 32 | import black |
| | 33 | current_version = parse_version(black.__version__) |
| | 34 | if current_version >= parse_version("21.5b1"): |
| | 35 | from black.files import find_pyproject_toml |
| | 36 | elif current_version >= parse_version("20.8b0"): |
| | 37 | from black import find_pyproject_toml |
| | 38 | else: |
| | 39 | find_pyproject_toml = None |
| | 40 | |
| | 41 | except ImportError: # pragma: no cover |
| | 42 | black = None |
| | 43 | |
| | 44 | |
| | 45 | def fix_code(code, directory): |
| | 46 | """Formats Python code to conform to the PEP 8 style guide. |
| | 47 | |
| | 48 | """ |
| | 49 | if not black: |
| | 50 | raise Fault("black not installed", code=400) |
| | 51 | # Get black config from pyproject.toml |
| | 52 | line_length = black.DEFAULT_LINE_LENGTH |
| | 53 | string_normalization = True |
| | 54 | if find_pyproject_toml: |
| | 55 | pyproject_path = find_pyproject_toml((directory,)) |
| | 56 | else: |
| | 57 | pyproject_path = os.path.join(directory, "pyproject.toml") |
| | 58 | if toml and pyproject_path and os.path.exists(pyproject_path): |
| | 59 | pyproject_config = toml.load(pyproject_path) |
| | 60 | black_config = pyproject_config.get("tool", {}).get("black", {}) |
| | 61 | if "line-length" in black_config: |
| | 62 | line_length = black_config["line-length"] |
| | 63 | if "skip-string-normalization" in black_config: |
| | 64 | string_normalization = not black_config["skip-string-normalization"] |
| | 65 | try: |
| | 66 | if parse_version(black.__version__) < parse_version("19.0"): |
| | 67 | reformatted_source = black.format_file_contents( |
| | 68 | src_contents=code, line_length=line_length, fast=False) |
| | 69 | else: |
| | 70 | fm = black.FileMode( |
| | 71 | line_length=line_length, |
| | 72 | string_normalization=string_normalization) |
| | 73 | reformatted_source = black.format_file_contents( |
| | 74 | src_contents=code, fast=False, mode=fm) |
| | 75 | return reformatted_source |
| | 76 | except black.NothingChanged: |
| | 77 | return code |
| | 78 | except Exception as e: |
| | 79 | raise Fault("Error during formatting: {}".format(e), code=400) |
projects / config.git / blame_incremental