44 lines
1.1 KiB
Python
44 lines
1.1 KiB
Python
from smnp.note.model import Note
|
|
|
|
class Node:
|
|
def __init__(self, parent, pos):
|
|
self.children = []
|
|
self.parent = parent
|
|
self.pos = pos
|
|
for child in self.children:
|
|
child.parent = self
|
|
|
|
def __repr__(self):
|
|
return self.__str__()
|
|
|
|
def __len__(self):
|
|
return len(self.children)
|
|
|
|
def __getitem__(self, index):
|
|
return self.children[index]
|
|
|
|
def append(self, node):
|
|
node.parent = self
|
|
self.children.append(node)
|
|
|
|
def pop(self, index):
|
|
return self.children.pop(index)
|
|
|
|
def print(self):
|
|
print(self._print(0))
|
|
|
|
def _print(self, level):
|
|
string = f"{pad(level)}{self.__class__.__name__}({self.parent.__class__.__name__}):\n"
|
|
for child in self.children:
|
|
if isinstance(child, str) or isinstance(child, int) or isinstance(child, Note):
|
|
string += pad(level + 1) + f"'{child}'\n"
|
|
else:
|
|
string += child._print(level + 1)
|
|
return string
|
|
|
|
def __str__(self):
|
|
return self._print(0)
|
|
|
|
|
|
def pad(level):
|
|
return " " * level |