janware/jw-python
1
mirror of ssh://git.janware.com/srv/git/janware/proj/jw-python synced 2026-01-15 01:52:56 +01:00
Code Releases Activity

stree.StringTree: Add .parent and .root

Browse source 
Signed-off-by: Jan Lindemann <jan@janware.com>
This commit is contained in:
Jan Lindemann 2025-05-20 13:09:58 +02:00
commit 18117e4590
1 changed files with 15 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tools/python/jwutils/stree/StringTree.py
View file

@ -34,11 +34,13 @@ def cleanup_string(s: str) -> str:
class StringTree: # export class StringTree: # export
def __init__(self, path: str, content: str) -> None: def __init__(self, path: str, content: str, parent: StringTree|None=None) -> None:
slog(DEBUG, f'Constructing StringTree(path="{path}", content="{content}")') slog(DEBUG, f'Constructing StringTree(path="{path}", content="{content}")')
self.__parent = parent
self.children: OrderedDict[str, StringTree] = OrderedDict() self.children: OrderedDict[str, StringTree] = OrderedDict()
self.content: Optional[str] = None self.content: Optional[str] = None
self.__set(path, content) self.__set(path, content)
assert(hasattr(self, "content")) assert(hasattr(self, "content"))
#assert self.content is not None #assert self.content is not None
@ -100,13 +102,13 @@ class StringTree: # export
nibble = components[0] nibble = components[0]
rest = '.'.join(components[1:]) rest = '.'.join(components[1:])
if nibble not in self.children: if nibble not in self.children:
self.children[nibble] = StringTree('', content=nibble) self.children[nibble] = StringTree('', content=nibble, parent=self)
if l > 1: if l > 1:
assert len(rest) > 0 assert len(rest) > 0
return self.children[nibble].__set(rest, content=content) return self.children[nibble].__set(rest, content=content)
# last component, a.k.a. leaf # last component, a.k.a. leaf
if content is not None: if content is not None:
gc = content if isinstance(content, StringTree) else StringTree('', content=content) gc = content if isinstance(content, StringTree) else StringTree('', content=content, parent=self.children[nibble])
# Make sure no existing grand child is updated. It would reside too # Make sure no existing grand child is updated. It would reside too
# far up in the grand child OrderedDict, we need it last # far up in the grand child OrderedDict, we need it last
if gc.content in self.children[nibble].children: if gc.content in self.children[nibble].children:
@ -189,6 +191,16 @@ class StringTree: # export
slog(DEBUG, f'Returning value from children {self.children}') slog(DEBUG, f'Returning value from children {self.children}')
return self.children[next(reversed(self.children))].content # type: ignore return self.children[next(reversed(self.children))].content # type: ignore
@property
def parent(self):
return self.__parent
@property
def root(self):
if self.__parent is None:
return self
return self.__parent.root
def child_list(self, depth_first: bool=True) -> List[StringTree]: def child_list(self, depth_first: bool=True) -> List[StringTree]:
if depth_first == False: if depth_first == False:
raise Exception("tried to retrieve child list with breadth-first search, not yet implemented") raise Exception("tried to retrieve child list with breadth-first search, not yet implemented")