aboutsummaryrefslogtreecommitdiffstats
path: root/queries/python
diff options
context:
space:
mode:
Diffstat (limited to 'queries/python')
-rw-r--r--queries/python/highlights.scm13
1 files changed, 12 insertions, 1 deletions
diff --git a/queries/python/highlights.scm b/queries/python/highlights.scm
index fe66d7ca3..d25448000 100644
--- a/queries/python/highlights.scm
+++ b/queries/python/highlights.scm
@@ -176,7 +176,18 @@
(escape_sequence) @string.escape
; doc-strings
-(expression_statement (string) @spell)
+
+(module . (expression_statement (string) @string.documentation @spell))
+
+(class_definition
+ body:
+ (block
+ . (expression_statement (string) @string.documentation @spell)))
+
+(function_definition
+ body:
+ (block
+ . (expression_statement (string) @string.documentation @spell)))
; Tokens