Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'runtime/queries/python/injections.scm')
-rw-r--r--runtime/queries/python/injections.scm12
1 files changed, 12 insertions, 0 deletions
diff --git a/runtime/queries/python/injections.scm b/runtime/queries/python/injections.scm
index 321c90ad..19a22c39 100644
--- a/runtime/queries/python/injections.scm
+++ b/runtime/queries/python/injections.scm
@@ -1,2 +1,14 @@
((comment) @injection.content
(#set! injection.language "comment"))
+
+; Match all 9 functions in the `re` module from the standard library that
+; that takes a regex pattern as first argument.
+; https://docs.python.org/3/library/re.html#functions
+(call
+ function: (attribute
+ object: (identifier) @_module (#eq? @_module "re")
+ attribute: (identifier) @_function (#any-of? @_function "compile" "search" "match" "fullmatch" "sub" "subn" "findall" "finditer" "split"))
+ arguments: (argument_list
+ . (string
+ (string_content) @injection.content))
+ (#set! injection.language "regex"))