From f9e81a268043cb9955a9c0d39ff929a838989b6b Mon Sep 17 00:00:00 2001
From: Jan Hoenig <hrominium@gmail.com>
Date: Sat, 17 Dec 2016 14:25:16 +0100
Subject: [PATCH] repr changes
---
astnodes.py | 10 +++++++++-
1 file changed, 9 insertions(+), 1 deletion(-)
diff --git a/astnodes.py b/astnodes.py
index 4a04a2a10..8856fd421 100644
--- a/astnodes.py
+++ b/astnodes.py
@@ -393,6 +393,10 @@ class TemporaryMemoryFree(Node):
# TODO everything which is not Atomic expression: Pow)
+
+_expr_dict = {'Add': ' + ', 'Mul': ' * ', 'Pow': '**'}
+
+
class Expr(Node):
def __init__(self, args, parent=None):
super(Expr, self).__init__(parent)
@@ -425,6 +429,9 @@ class Expr(Node):
def undefinedSymbols(self):
return set() # Todo fix for symbol analysis
+ def __repr__(self):
+ return _expr_dict[self.__class__.__name__].join(repr(arg) for arg in self.args) # TODO test this
+
class Mul(Expr):
pass
@@ -439,6 +446,7 @@ class Pow(Expr):
class Indexed(Expr):
- pass
+ def __repr__(self):
+ return '%s[%s]' % (self.args[0], self.args[1])
--
GitLab