python-language-server/pyls/plugins/yapf_format.py at develop · deepnote/python-language-server · GitHub

History

56 lines (45 loc) · 1.64 KB

Raw

import logging

import os

from yapf.yapflib import file_resources

from yapf.yapflib.yapf_api import FormatCode

from pyls import hookimpl

log = logging.getLogger(__name__)

@hookimpl

def pyls_format_document(document):

return _format(document)

@hookimpl

def pyls_format_range(document, range): # pylint: disable=redefined-builtin

# First we 'round' the range up/down to full lines only

range['start']['character'] = 0

range['end']['line'] += 1

range['end']['character'] = 0

# From Yapf docs:

# lines: (list of tuples of integers) A list of tuples of lines, [start, end],

# that we want to format. The lines are 1-based indexed. It can be used by

# third-party code (e.g., IDEs) when reformatting a snippet of code rather

# than a whole file.

# Add 1 for 1-indexing vs LSP's 0-indexing

lines = [(range['start']['line'] + 1, range['end']['line'] + 1)]

return _format(document, lines=lines)

def _format(document, lines=None):

new_source, changed = FormatCode(

document.source,

lines=lines,

filename=document.filename,

style_config=file_resources.GetDefaultStyleForDir(

os.path.dirname(document.path)

)

if not changed:

return []

# I'm too lazy at the moment to parse diffs into TextEdit items

# So let's just return the entire file...

return [{

'range': {

'start': {'line': 0, 'character': 0},

# End char 0 of the line after our document

'end': {'line': len(document.lines), 'character': 0}

'newText': new_source

}]

python-language-server/pyls/plugins/yapf_format.py at develop · deepnote/python-language-server · GitHub

Uh oh!

Expand file tree

Breadcrumbs

yapf_format.py

Latest commit

History

Breadcrumbs

yapf_format.py

File metadata and controls