Add simple constant folding

dflook · dflook · commit 523364ba4c4c · 2023-08-16T16:55:13.000+01:00
diff --git a/src/python_minifier/transforms/constant_folding.py b/src/python_minifier/transforms/constant_folding.py
@@ -19,9 +19,9 @@ def visit_BinOp(self, node):
         node.right = self.visit(node.right)
 
         # Check this is a constant expression that could be folded
-        if not is_ast_node(node.left, (ast.Num, ast.Str, ast.Bytes, ast.NameConstant)):
+        if not is_ast_node(node.left, (ast.Num, ast.Str, 'Bytes', 'NameConstant')):
             return node
-        if not is_ast_node(node.right, (ast.Num, ast.Str, ast.Bytes, ast.NameConstant)):
+        if not is_ast_node(node.right, (ast.Num, ast.Str, 'Bytes', 'NameConstant')):
             return node
 
         expression_printer = ExpressionPrinter()
