From 807d6e11fa7150481de8c80a7cf973be2e456d50 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Wed, 27 May 2020 22:23:34 +0200
Subject: Improve Python highlights

- Highlight types in attributes as @type (**Foo**.a)
- Highlight superclasses @type
- Highlight ^[A-Z] as @type
- Highlight isinstance(*, type) as @type
- Highlight default_parameter and keyword_argument as @parameter
- Highlight *args, **kwargs as @parameter
- Highlight -> as @operator
---
 queries/python/highlights.scm | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

(limited to 'queries/python')

diff --git a/queries/python/highlights.scm b/queries/python/highlights.scm
index 4cd8b8998..0b6e30033 100644
--- a/queries/python/highlights.scm
+++ b/queries/python/highlights.scm
@@ -9,6 +9,9 @@
            (identifier)) @type)
  (match? @type "^[A-Z]"))
 
+((identifier) @type
+ (match? @type "^[A-Z]"))
+
 ((identifier) @constant
  (match? @constant "^[A-Z][A-Z_]*$"))
 
@@ -43,8 +46,30 @@
 (identifier) @variable
 (attribute attribute: (identifier) @property)
 (type (identifier) @type)
+((call 
+  function: (identifier) @isinstance
+  arguments: (argument_list
+    (*)
+    (identifier) @type))
+ (eq? @isinstance "isinstance"))
+
+; Normal parameters
 (parameters
   (identifier) @parameter) 
+; Default parameters
+(keyword_argument
+  name: (identifier) @parameter)
+; Naming parameters on call-site
+(default_parameter
+  name: (identifier) @parameter)
+; Variadic parameters *args, **kwargs
+(parameters
+  (list_splat ; *args
+    (identifier) @parameter)) 
+(parameters
+  (dictionary_splat ; **kwargs
+    (identifier) @parameter)) 
+
 
 ; Literals
 
@@ -68,6 +93,7 @@
 ; Tokens
 
 "-" @operator
+"->" @operator
 "-=" @operator
 "!=" @operator
 "*" @operator
@@ -145,6 +171,9 @@
 
 (class_definition
   name: (identifier) @type)
+(class_definition
+  superclasses: (argument_list 
+    (identifier) @type))
 
 (attribute
     attribute: (identifier) @field)
@@ -157,6 +186,10 @@
     attribute: (identifier) @type)
     (match? @type "^[A-Z][a-z_]+"))
 
+((attribute
+    object: (identifier) @type)
+    (match? @type "^[A-Z][a-z_]+"))
+
 (class_definition
   body: (block
           (expression_statement
-- 
cgit v1.2.3-70-g09d2