You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.8 KiB
59 lines
1.8 KiB
"""Glue for the "black" library.
|
|
|
|
"""
|
|
|
|
import sys
|
|
from pkg_resources import parse_version
|
|
|
|
import os
|
|
|
|
try:
|
|
import toml
|
|
except ImportError:
|
|
toml = None
|
|
|
|
from elpy.rpc import Fault
|
|
|
|
BLACK_NOT_SUPPORTED = sys.version_info < (3, 6)
|
|
|
|
try:
|
|
if BLACK_NOT_SUPPORTED:
|
|
black = None
|
|
else:
|
|
import black
|
|
except ImportError: # pragma: no cover
|
|
black = None
|
|
|
|
|
|
def fix_code(code, directory):
|
|
"""Formats Python code to conform to the PEP 8 style guide.
|
|
|
|
"""
|
|
if not black:
|
|
raise Fault("black not installed", code=400)
|
|
# Get black config from pyproject.toml
|
|
line_length = black.DEFAULT_LINE_LENGTH
|
|
string_normalization = True
|
|
pyproject_path = os.path.join(directory, "pyproject.toml")
|
|
if toml is not None and os.path.exists(pyproject_path):
|
|
pyproject_config = toml.load(pyproject_path)
|
|
black_config = pyproject_config.get("tool", {}).get("black", {})
|
|
if "line-length" in black_config:
|
|
line_length = black_config["line-length"]
|
|
if "skip-string-normalization" in black_config:
|
|
string_normalization = not black_config["skip-string-normalization"]
|
|
try:
|
|
if parse_version(black.__version__) < parse_version("19.0"):
|
|
reformatted_source = black.format_file_contents(
|
|
src_contents=code, line_length=line_length, fast=False)
|
|
else:
|
|
fm = black.FileMode(
|
|
line_length=line_length,
|
|
string_normalization=string_normalization)
|
|
reformatted_source = black.format_file_contents(
|
|
src_contents=code, fast=False, mode=fm)
|
|
return reformatted_source
|
|
except black.NothingChanged:
|
|
return code
|
|
except Exception as e:
|
|
raise Fault("Error during formatting: {}".format(e), code=400)
|