Issue #9128: Fix validation of class decorators in parser module.
diff --git a/Lib/test/test_parser.py b/Lib/test/test_parser.py
index 071d362..96f14f6 100644
--- a/Lib/test/test_parser.py
+++ b/Lib/test/test_parser.py
@@ -149,6 +149,13 @@
def test_class_defs(self):
self.check_suite("class foo():pass")
self.check_suite("class foo(object):pass")
+ self.check_suite("@class_decorator\n"
+ "class foo():pass")
+ self.check_suite("@class_decorator(arg)\n"
+ "class foo():pass")
+ self.check_suite("@decorator1\n"
+ "@decorator2\n"
+ "class foo():pass")
def test_import_from_statement(self):
self.check_suite("from sys.path import *")
diff --git a/Misc/NEWS b/Misc/NEWS
index 091deab..b7fe3f6 100644
--- a/Misc/NEWS
+++ b/Misc/NEWS
@@ -468,6 +468,8 @@
Library
-------
+- Issue #9128: Fix validation of class decorators in parser module.
+
- Issue #9094: python -m pickletools will now disassemble pickle files
listed in the command line arguments. See output of python -m
pickletools -h for more details.
diff --git a/Modules/parsermodule.c b/Modules/parsermodule.c
index 3e0f530..e60f23d 100644
--- a/Modules/parsermodule.c
+++ b/Modules/parsermodule.c
@@ -2509,14 +2509,15 @@
static int
validate_decorated(node *tree)
{
- int nch = NCH(tree);
- int ok = (validate_ntype(tree, decorated)
- && (nch == 2)
- && validate_decorators(RCHILD(tree, -2))
- && (validate_funcdef(RCHILD(tree, -1))
- || validate_class(RCHILD(tree, -1)))
- );
- return ok;
+ int nch = NCH(tree);
+ int ok = (validate_ntype(tree, decorated)
+ && (nch == 2)
+ && validate_decorators(RCHILD(tree, -2)));
+ if (TYPE(RCHILD(tree, -1)) == funcdef)
+ ok = ok && validate_funcdef(RCHILD(tree, -1));
+ else
+ ok = ok && validate_class(RCHILD(tree, -1));
+ return ok;
}
static int