Improve Python queries, add locals (#3284)

runtime/queries/python/highlights.scm

@@ -1,6 +1,11 @@
+; Imports
+
 (dotted_name
   (identifier)* @namespace)
   
+(aliased_import
+  alias: (identifier) @namespace)
+
 ; Builtin functions
 
 ((call
@@ -55,7 +60,6 @@
 (parameters (typed_parameter (identifier) @variable.parameter))
 (parameters (default_parameter name: (identifier) @variable.parameter))
 (parameters (typed_default_parameter name: (identifier) @variable.parameter))
-(keyword_argument name: (identifier) @variable.parameter)
 
 (parameters
   (list_splat_pattern ; *args
@@ -64,6 +68,9 @@
   (dictionary_splat_pattern ; **kwargs
     (identifier) @variable.parameter))
     
+(lambda_parameters
+  (identifier) @variable.parameter)
+
 ; Types
 
 ((identifier) @type.builtin

runtime/queries/python/locals.scm

@@ -0,0 +1,44 @@
+;; Scopes
+
+[
+  (module)
+  (function_definition)
+  (lambda)
+] @local.scope
+
+;; Definitions
+
+; Parameters
+(parameters
+  (identifier) @local.definition)
+(parameters
+  (typed_parameter
+    (identifier) @local.definition))
+(parameters
+  (default_parameter 
+    name: (identifier) @local.definition))
+(parameters 
+  (typed_default_parameter 
+    name: (identifier) @local.definition))
+(parameters
+  (list_splat_pattern ; *args
+    (identifier) @local.definition))
+(parameters
+  (dictionary_splat_pattern ; **kwargs
+    (identifier) @local.definition))
+    
+(lambda_parameters
+  (identifier) @local.definition)
+  
+; Imports
+(import_statement
+  name: (dotted_name 
+    (identifier) @local.definition))
+
+(aliased_import
+  alias: (identifier) @local.definition)
+
+;; References
+
+(identifier) @local.reference
+