1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
# Copyright 2017-2020 Palantir Technologies, Inc.
# Copyright 2021- Python Language Server Contributors.
import logging
from pylsp import _utils, hookimpl, lsp
log = logging.getLogger(__name__)
@hookimpl
def pylsp_document_highlight(document, position):
code_position = _utils.position_to_jedi_linecolumn(document, position)
usages = document.jedi_script().get_references(**code_position)
def is_valid(definition):
return definition.line is not None and definition.column is not None
def local_to_document(definition):
return (
not definition.module_path or str(definition.module_path) == document.path
)
return [
{
"range": {
"start": {"line": d.line - 1, "character": d.column},
"end": {"line": d.line - 1, "character": d.column + len(d.name)},
},
"kind": lsp.DocumentHighlightKind.Write
if d.is_definition()
else lsp.DocumentHighlightKind.Read,
}
for d in usages
if is_valid(d) and local_to_document(d)
]
|
