Allow escaping line breaks in Python strings

FIameCaster · Apr 4, 2024 · 9d297f5 · 9d297f5
1 parent fe7de05
commit 9d297f5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/package/src/prism/languages/python.js b/package/src/prism/languages/python.js
@@ -18,7 +18,7 @@ inside[rest] = languages.py = languages.python = {
 		greedy: true
 	},
 	'string-interpolation': {
-		pattern: /(?:fr?|rf)(?:("""|''')[\s\S]*?\1|(["'])(?:\\.|(?!\2)[^\\\n])*\2)/gi,
+		pattern: /(?:fr?|rf)(?:("""|''')[\s\S]*?\1|(["'])(?:\\[\s\S]|(?!\2)[^\\\n])*\2)/gi,
 		greedy: true,
 		inside: {
 			'interpolation': {
@@ -31,12 +31,12 @@ inside[rest] = languages.py = languages.python = {
 		}
 	},
 	'triple-quoted-string': {
-		pattern: /(?:[rub]|br|rb)?("""|''')[\s\S]*?\1/gi,
+		pattern: /(?:br?|rb?|u)?("""|''')[\s\S]*?\1/gi,
 		greedy: true,
 		alias: 'string'
 	},
 	'string': {
-		pattern: /(?:[rub]|br|rb)?(["'])(?:\\.|(?!\1)[^\\\n])*\1/gi,
+		pattern: /(?:br?|rb?|u)?(["'])(?:\\[\s\S]|(?!\1)[^\\\n])*\1/gi,
 		greedy: true
 	},
 	'function': {

diff --git a/package/src/prism/tests/languages/python/string_feature.test b/package/src/prism/tests/languages/python/string_feature.test
@@ -3,6 +3,8 @@
 ''
 'fo\'obar'
 "fo\" # comment obar"
+"foo\
+bar"
 
 r"\n"
 b'foo'
@@ -17,6 +19,7 @@ u"foo"
 	["string", "''"],
 	["string", "'fo\\'obar'"],
 	["string", "\"fo\\\" # comment obar\""],
+	["string", "\"foo\\\nbar\""],
 
 	["string", "r\"\\n\""],
 	["string", "b'foo'"],
@@ -26,4 +29,4 @@ u"foo"
 
 ----------------------------------------------------
 
-Checks for strings.
+Checks for strings.