aboutsummaryrefslogtreecommitdiffstats
path: root/queries/python
diff options
context:
space:
mode:
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