FilesExpand file tree

develop

/

highlight.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

27 lines (20 loc) · 995 Bytes

develop

/

highlight.py

Copy path

Top

File metadata and controls

• Code
• Blame

27 lines (20 loc) · 995 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

# Copyright 2017-2020 Palantir Technologies, Inc.

# Copyright 2021- Python Language Server Contributors.

import logging

from pylsp import hookimpl, lsp, _utils

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)]