aboutsummaryrefslogtreecommitdiffstats
path: root/tools/qtpy2cpp_lib/visitor.py
diff options
context:
space:
mode:
authorFriedemann Kleint <Friedemann.Kleint@qt.io>2022-05-27 12:11:27 +0200
committerFriedemann Kleint <Friedemann.Kleint@qt.io>2022-05-31 21:49:24 +0200
commitf684775146242b284eb890fd446179c925fa91f6 (patch)
tree6329fc29226f386d334f856b1c16f44960493280 /tools/qtpy2cpp_lib/visitor.py
parent65cab4213ff09244842b7b77ff29d52491864186 (diff)
qtpy2cpp: Add more operators
Pick-to: 6.3 Task-number: PYSIDE-1945 Change-Id: I802ded1388918a8f5b72f8f9382a052e09e936ee Reviewed-by: Cristian Maureira-Fredes <cristian.maureira-fredes@qt.io>
Diffstat (limited to 'tools/qtpy2cpp_lib/visitor.py')
-rw-r--r--tools/qtpy2cpp_lib/visitor.py40
1 files changed, 38 insertions, 2 deletions
diff --git a/tools/qtpy2cpp_lib/visitor.py b/tools/qtpy2cpp_lib/visitor.py
index 96ac5357e..57a483191 100644
--- a/tools/qtpy2cpp_lib/visitor.py
+++ b/tools/qtpy2cpp_lib/visitor.py
@@ -89,6 +89,14 @@ class ConvertVisitor(ast.NodeVisitor, CppFormatter):
self.generic_visit(node)
self._output_file.write(')')
+ def visit_BitAnd(self, node):
+ self.generic_visit(node)
+ self._output_file.write(" & ")
+
+ def visit_BitOr(self, node):
+ self.generic_visit(node)
+ self._output_file.write(" | ")
+
def visit_Call(self, node):
self._output_file.write(format_start_function_call(node))
# Manually do visit(), skip the children of func
@@ -110,6 +118,10 @@ class ConvertVisitor(ast.NodeVisitor, CppFormatter):
self.indent_line('};')
del self._class_scope[-1]
+ def visit_Eq(self, node):
+ self.generic_visit(node)
+ self._output_file.write(" == ")
+
def visit_Expr(self, node):
self._output_file.write('\n')
self.INDENT()
@@ -118,7 +130,11 @@ class ConvertVisitor(ast.NodeVisitor, CppFormatter):
def visit_Gt(self, node):
self.generic_visit(node)
- self._output_file.write('>')
+ self._output_file.write(" > ")
+
+ def visit_GtE(self, node):
+ self.generic_visit(node)
+ self._output_file.write(" >= ")
def visit_For(self, node):
# Manually do visit() to get the indentation right.
@@ -173,9 +189,17 @@ class ConvertVisitor(ast.NodeVisitor, CppFormatter):
self.visit(el)
self._output_file.write('}')
+ def visit_LShift(self, node):
+ self.generic_visit(node)
+ self._output_file.write(" << ")
+
def visit_Lt(self, node):
self.generic_visit(node)
- self._output_file.write('<')
+ self._output_file.write(" < ")
+
+ def visit_LtE(self, node):
+ self.generic_visit(node)
+ self._output_file.write(" <= ")
def visit_Mult(self, node):
self.generic_visit(node)
@@ -194,10 +218,22 @@ class ConvertVisitor(ast.NodeVisitor, CppFormatter):
else:
self._output_file.write('true')
+ def visit_Not(self, node):
+ self.generic_visit(node)
+ self._output_file.write("!")
+
+ def visit_NotEq(self, node):
+ self.generic_visit(node)
+ self._output_file.write(" != ")
+
def visit_Num(self, node):
self.generic_visit(node)
self._output_file.write(format_literal(node))
+ def visit_RShift(self, node):
+ self.generic_visit(node)
+ self._output_file.write(" >> ")
+
def visit_Str(self, node):
self.generic_visit(node)
self._output_file.write(format_literal(node))