voussoir/voussoirkit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix __str__ in case self.token is not a string.

Browse source
This commit is contained in:
Ethan Dalool 2020-09-09 12:55:09 -07:00
parent bab1c2bffc
commit 7052b2c432
1 changed files with 8 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
voussoirkit/expressionmatch.py
View file

@ -59,8 +59,10 @@ class ExpressionTree:
if self.token is None:
return '""'
if self.token not in OPERATORS:
t = self.token
self_token = str(self.token)
if self_token not in OPERATORS:
t = self_token
t = t.replace('"', '\\"')
t = t.replace('(', '\\(')
t = t.replace(')', '\\)')
@ -72,8 +74,8 @@ class ExpressionTree:
child = self.children[0]
childstring = str(child)
if child.token in OPERATORS:
return f'{self.token}({childstring})'
return f'{self.token} {childstring}'
return f'{self_token}({childstring})'
return f'{self_token} {childstring}'
children = []
for child in self.children:
@ -83,9 +85,9 @@ class ExpressionTree:
children.append(childstring)
if len(children) == 1:
return f'{self.token} {children[0]}'
return f'{self_token} {children[0]}'
s = f' {self.token} '
s = f' {self_token} '
s = s.join(children)
return s