Allow expressions in decorators (PEP 614)

dflook · dflook · commit ccf27a0bbc7a · 2020-10-15T09:12:06.000+01:00
diff --git a/src/python_minifier/module_printer.py b/src/python_minifier/module_printer.py
@@ -484,7 +484,7 @@ def visit_FunctionDef(self, node, is_async=False):
 
         for d in node.decorator_list:
             self.code += '@'
-            self.visit(d)
+            self._expression(d)
             self.newline()
 
         if is_async:
@@ -513,7 +513,7 @@ def visit_ClassDef(self, node):
 
         for d in node.decorator_list:
             self.code += '@'
-            self.visit(d)
+            self._expression(d)
             self.newline()
 
         first = True
