perf: modify TreeNode id and TreeLeaf id

This commit is contained in:
Bai
2026-01-01 22:12:41 +08:00
parent dc88dcdabd
commit 10a932b20c
2 changed files with 4 additions and 1 deletions

View File

@@ -120,6 +120,7 @@ class AssetNodeTree(Tree):
assert parent_key is not None, "TreeLeaf parent key should not be None"
tree_leaf = AssetTreeLeaf(
instance=asset,
id=str(asset.id),
key=None,
parent_key=parent_key,
name=asset.name,

View File

@@ -11,6 +11,7 @@ class TreeNode:
def __init__(self, key, parent_key, name, leaves_amount=0):
## need to serialize fields
self.id = key
self.key = key # leaf key is None, parent_key
self.name = name
self.open = True
@@ -64,8 +65,9 @@ class TreeNode:
class TreeLeaf(TreeNode):
def __init__(self, *args, **kwargs):
def __init__(self, id, *args, **kwargs):
super().__init__(*args, **kwargs)
self.id = id
self.key = None
self.is_leaf = True
self.open = False