aboutsummaryrefslogtreecommitdiffstats
path: root/queries/python
diff options
context:
space:
mode:
authorStephan Seitz <stephan.seitz@fau.de>2020-07-08 17:35:12 +0200
committerThomas Vigouroux <39092278+vigoux@users.noreply.github.com>2020-07-08 17:47:44 +0200
commitdc324e44ad6ffc0f2089549afa6ba0fa2ba2f7f6 (patch)
treea4f64120653b54f0ba0162214b26097e617b5030 /queries/python
parentAdd Rust queries (#57) (diff)
downloadnvim-treesitter-dc324e44ad6ffc0f2089549afa6ba0fa2ba2f7f6.tar
nvim-treesitter-dc324e44ad6ffc0f2089549afa6ba0fa2ba2f7f6.tar.gz
nvim-treesitter-dc324e44ad6ffc0f2089549afa6ba0fa2ba2f7f6.tar.bz2
nvim-treesitter-dc324e44ad6ffc0f2089549afa6ba0fa2ba2f7f6.tar.lz
nvim-treesitter-dc324e44ad6ffc0f2089549afa6ba0fa2ba2f7f6.tar.xz
nvim-treesitter-dc324e44ad6ffc0f2089549afa6ba0fa2ba2f7f6.tar.zst
nvim-treesitter-dc324e44ad6ffc0f2089549afa6ba0fa2ba2f7f6.zip
Python locals/highlights: Parameters can be optionally typed
Diffstat (limited to 'queries/python')
-rw-r--r--queries/python/highlights.scm4
-rw-r--r--queries/python/locals.scm31
2 files changed, 21 insertions, 14 deletions
diff --git a/queries/python/highlights.scm b/queries/python/highlights.scm
index fb316bea5..b41d36891 100644
--- a/queries/python/highlights.scm
+++ b/queries/python/highlights.scm
@@ -66,6 +66,10 @@
; Naming parameters on call-site
(default_parameter
name: (identifier) @parameter)
+(typed_parameter
+ (identifier) @parameter)
+(typed_default_parameter
+ (identifier) @parameter)
; Variadic parameters *args, **kwargs
(parameters
(list_splat ; *args
diff --git a/queries/python/locals.scm b/queries/python/locals.scm
index cdf4d4117..bcf1b1572 100644
--- a/queries/python/locals.scm
+++ b/queries/python/locals.scm
@@ -9,24 +9,27 @@
(identifier) @definition.associated))))) @scope
; Function with parameters, defines parameters
-(function_definition
- name: (identifier)
- parameters: (parameters
- (identifier) @definition.var))
+(parameters
+ (identifier) @definition.var)
+
+(default_parameter
+ (identifier) @definition.var)
+
+(typed_parameter
+ (identifier) @definition.var)
+
+(typed_default_parameter
+ (identifier) @definition.var)
; *args parameter
-(function_definition
- name: (identifier)
- parameters: (parameters
- (list_splat
- (identifier) @definition.var)))
+(parameters
+ (list_splat
+ (identifier) @definition.var))
; **kwargs parameter
-(function_definition
- name: (identifier)
- parameters: (parameters
- (dictionary_splat
- (identifier) @definition.var)))
+(parameters
+ (dictionary_splat
+ (identifier) @definition.var))
; Function defines function and scope
(function_definition