python: Highlight decorators with arguments.

zed-industries · Dec 5, 2024 · 9e6416d · 9e6416d
1 parent 24edcaf
commit 9e6416d
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/crates/languages/src/python/highlights.scm b/crates/languages/src/python/highlights.scm
@@ -28,15 +28,18 @@
 
 ; Function calls
 
-(decorator
-  "@" @punctuation.special
-  (identifier) @function.decorator)
-
 (call
   function: (attribute attribute: (identifier) @function.method.call))
 (call
   function: (identifier) @function.call)
 
+(decorator
+  "@" @punctuation.special
+  (identifier) @function.decorator)
+(decorator
+  "@" @punctuation.special
+  (call function: (identifier) @function.decorator.call))
+
 ; Function and class definitions
 
 (function_definition