1
0
mirror of https://github.com/SpaceVim/SpaceVim.git synced 2025-01-27 12:30:04 +08:00
SpaceVim/bundle/nvim-treesitter/queries/python/locals.scm
2022-04-14 12:01:23 +08:00

116 lines
2.5 KiB
Scheme
Vendored

;;; Program structure
(module) @scope
(class_definition
body: (block
(expression_statement
(assignment
left: (identifier) @definition.field)))) @scope
(class_definition
body: (block
(expression_statement
(assignment
left: (_
(identifier) @definition.field))))) @scope
; Imports
(aliased_import
alias: (identifier) @definition.import)
(import_statement
name: (dotted_name ((identifier) @definition.import)))
(import_from_statement
name: (dotted_name ((identifier) @definition.import)))
; Function with parameters, defines parameters
(parameters
(identifier) @definition.parameter)
(default_parameter
(identifier) @definition.parameter)
(typed_parameter
(identifier) @definition.parameter)
(typed_default_parameter
(identifier) @definition.parameter)
; *args parameter
(parameters
(list_splat_pattern
(identifier) @definition.parameter))
; **kwargs parameter
(parameters
(dictionary_splat_pattern
(identifier) @definition.parameter))
; Function defines function and scope
((function_definition
name: (identifier) @definition.function) @scope
(#set! definition.function.scope "parent"))
((class_definition
name: (identifier) @definition.type) @scope
(#set! definition.type.scope "parent"))
(class_definition
body: (block
(function_definition
name: (identifier) @definition.method)))
;;; Loops
; not a scope!
(for_statement
left: (pattern_list
(identifier) @definition.var))
(for_statement
left: (tuple_pattern
(identifier) @definition.var))
(for_statement
left: (identifier) @definition.var)
; not a scope!
;(while_statement) @scope
; for in list comprehension
(for_in_clause
left: (identifier) @definition.var)
(for_in_clause
left: (tuple_pattern
(identifier) @definition.var))
(for_in_clause
left: (pattern_list
(identifier) @definition.var))
(dictionary_comprehension) @scope
(list_comprehension) @scope
(set_comprehension) @scope
;;; Assignments
(assignment
left: (identifier) @definition.var)
(assignment
left: (pattern_list
(identifier) @definition.var))
(assignment
left: (tuple_pattern
(identifier) @definition.var))
(assignment
left: (attribute
(identifier)
(identifier) @definition.field))
; Walrus operator x := 1
(named_expression
(identifier) @definition.var)
(as_pattern
alias: (as_pattern_target) @definition.var)
;;; REFERENCES
(identifier) @reference