Unnamed repository; edit this file 'description' to name the repository.
feat(python): add regex injection to functions in the std `re` module (#14254)
| -rw-r--r-- | runtime/queries/python/injections.scm | 12 |
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")) |