diff options
-rw-r--r-- | util/codegen.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/util/codegen.py b/util/codegen.py index 195a70a..451cf74 100644 --- a/util/codegen.py +++ b/util/codegen.py @@ -447,11 +447,14 @@ class SourceGenerator(NodeVisitor): self.write('}') def visit_BinOp(self, node): - self.write('(') + op = self.BINOP_SYMBOLS[node.op] + if op in ['+', '-']: + self.write('(') self.visit(node.left) - self.write(' %s ' % self.BINOP_SYMBOLS[node.op]) + self.write(' %s ' % op) self.visit(node.right) - self.write(')') + if op in ['+', '-']: + self.write(')') def visit_BoolOp(self, node): self.write('(') |