Python: Fixed decorators (#2018)

This adds a missing multiline flag to support more than one decorator.
PrismJS · Aug 29, 2019 · 5b8a16d · 5b8a16d
1 parent 9f72258
commit 5b8a16d
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 6 deletions.
diff --git a/components/prism-python.js b/components/prism-python.js
@@ -44,7 +44,7 @@ Prism.languages.python = {
 		lookbehind: true
 	},
 	'decorator': {
-		pattern: /(^\s*)@\w+(?:\.\w+)*/i,
+		pattern: /(^\s*)@\w+(?:\.\w+)*/im,
 		lookbehind: true,
 		alias: ['annotation', 'punctuation'],
 		inside: {
@@ -61,4 +61,4 @@ Prism.languages.python = {
 
 Prism.languages.python['string-interpolation'].inside['interpolation'].inside.rest = Prism.languages.python;
 
-Prism.languages.py = Prism.languages.python;
+Prism.languages.py = Prism.languages.python;
diff --git a/components/prism-python.min.js b/components/prism-python.min.js
diff --git a/tests/languages/python/decorator_feature.test b/tests/languages/python/decorator_feature.test
@@ -1,19 +1,48 @@
 @some.decorator(foobar)
 def foo(bar):
-    pass
+	pass
+
+@decorator
+def foo(bar):
+	pass
+
+@decorator
+def foo(bar):
+	pass
 
 ----------------------------------------------------
 
 [
 	["decorator", [
 		"@some",
 		["punctuation", "."],
-		"decorator"]
-	],
+		"decorator"
+	]],
 	["punctuation", "("],
 	"foobar",
 	["punctuation", ")"],
+	["keyword", "def"],
+	["function", "foo"],
+	["punctuation", "("],
+	"bar",
+	["punctuation", ")"],
+	["punctuation", ":"],
+	["keyword", "pass"],
 
+	["decorator", [
+		"@decorator"
+	]],
+	["keyword", "def"],
+	["function", "foo"],
+	["punctuation", "("],
+	"bar",
+	["punctuation", ")"],
+	["punctuation", ":"],
+	["keyword", "pass"],
+
+	["decorator", [
+		"@decorator"
+	]],
 	["keyword", "def"],
 	["function", "foo"],
 	["punctuation", "("],