diff options
-rw-r--r-- | tools/qtpy2cpp_lib/visitor.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/tools/qtpy2cpp_lib/visitor.py b/tools/qtpy2cpp_lib/visitor.py index f5476ac10..35d659d64 100644 --- a/tools/qtpy2cpp_lib/visitor.py +++ b/tools/qtpy2cpp_lib/visitor.py @@ -101,7 +101,6 @@ class ConvertVisitor(ast.NodeVisitor, CppFormatter): self._output_file.write(' + ') def visit_Assign(self, node): - self._output_file.write('\n') self.INDENT() line_no = node.lineno if hasattr(node, 'lineno') else -1 for target in node.targets: @@ -162,7 +161,6 @@ class ConvertVisitor(ast.NodeVisitor, CppFormatter): self._output_file.write(" == ") def visit_Expr(self, node): - self._output_file.write('\n') self.INDENT() self.generic_visit(node) self._output_file.write(';\n') @@ -273,6 +271,13 @@ class ConvertVisitor(ast.NodeVisitor, CppFormatter): self.generic_visit(node) self._output_file.write(" >> ") + def visit_Return(self, node): + self.indent_string("return") + if node.value: + self._output_file.write(" ") + self.generic_visit(node) + self._output_file.write(";\n") + def visit_Str(self, node): self.generic_visit(node) self._output_file.write(format_literal(node)) |