diff --git a/runtime/queries/python/injections.scm b/runtime/queries/python/injections.scm index 2a120d9b5..28334477c 100644 --- a/runtime/queries/python/injections.scm +++ b/runtime/queries/python/injections.scm @@ -3,7 +3,12 @@ object: (identifier) @_re) arguments: (argument_list (string - (string_content) @injection.content)) + (string_start) + [ + (string_content) @injection.content + (interpolation) + ]+ + (string_end))) (#eq? @_re "re") (#set! injection.language "regex")) @@ -12,11 +17,13 @@ object: (identifier) @_re) arguments: (argument_list (concatenated_string - [ - (string - (string_content) @injection.content) - (comment) - ]+)) + (string + (string_start) + [ + (string_content) @injection.content + (interpolation) + ]+ + (string_end)))) (#eq? @_re "re") (#set! injection.language "regex")) diff --git a/tests/query/injections/python/test.py b/tests/query/injections/python/test.py index 1d6a0a465..f6dbaf355 100644 --- a/tests/query/injections/python/test.py +++ b/tests/query/injections/python/test.py @@ -14,5 +14,15 @@ re_test = re.compile( # comment # ^ @comment ) +a.a( + f"