emacs_config/elpa/elpy-1.35.0/elpy/yapfutil.py

"""Glue for the "yapf" library.

"""

import os
import sys

from elpy.rpc import Fault

YAPF_NOT_SUPPORTED = sys.version_info < (2, 7) or (
    sys.version_info >= (3, 0) and sys.version_info < (3, 4))

try:
    if YAPF_NOT_SUPPORTED:
        yapf_api = None
    else:
        from yapf.yapflib import yapf_api
        from yapf.yapflib import file_resources
except ImportError:  # pragma: no cover
    yapf_api = None


def fix_code(code, directory):
    """Formats Python code to conform to the PEP 8 style guide.

    """
    if not yapf_api:
        raise Fault('yapf not installed', code=400)
    style_config = file_resources.GetDefaultStyleForDir(directory or os.getcwd())
    try:
        reformatted_source, _ = yapf_api.FormatCode(code,
                                                    filename='<stdin>',
                                                    style_config=style_config,
                                                    verify=False)
        return reformatted_source
    except Exception as e:
            raise Fault("Error during formatting: {}".format(e),
                        code=400)