aviallon/compiler
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

nodes: prevent infinite recursion in repr

Browse source
This commit is contained in:
Antoine Viallon 2023-05-12 01:22:00 +02:00
parent 0c42eabde1
commit c656a98d3e
Signed by: aviallon
GPG key ID: D126B13AB555E16F
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
compiler/nodes.py
View file

@ -18,6 +18,7 @@ class Node:
def __init__(self): def __init__(self):
self.context: semantic.Context | None = None self.context: semantic.Context | None = None
self._repr_guard: bool = False
@abstractmethod @abstractmethod
def _values(self) -> list[Node | Any]: def _values(self) -> list[Node | Any]:
@ -32,11 +33,15 @@ class Node:
return loc return loc
def __repr__(self): def __repr__(self):
if self._repr_guard:
return self.__class__.__name__
self._repr_guard = True
vals = self._values() vals = self._values()
if type(vals) == list: if type(vals) == list:
vals = ", ".join(repr(val) for val in vals) vals = ", ".join(repr(val) for val in vals)
else: else:
vals = repr(vals) vals = repr(vals)
self._repr_guard = False
return f"{self.__class__.__name__}({vals})" return f"{self.__class__.__name__}({vals})"
def pprint(self, depth: int | None = None, indent: str = "\t"): def pprint(self, depth: int | None = None, indent: str = "\t"):