From 10a932b20ce7394f6b1662dada07fd762876bf50 Mon Sep 17 00:00:00 2001
From: Bai <baijiangjie@gmail.com>
Date: Thu, 1 Jan 2026 22:12:41 +0800
Subject: [PATCH] perf: modify TreeNode id and TreeLeaf id

---
 apps/assets/tree/x1/node_tree.py | 1 +
 apps/assets/tree/x1/tree.py      | 4 +++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/apps/assets/tree/x1/node_tree.py b/apps/assets/tree/x1/node_tree.py
index e3c0b93a2..159150dfb 100644
--- a/apps/assets/tree/x1/node_tree.py
+++ b/apps/assets/tree/x1/node_tree.py
@@ -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,
diff --git a/apps/assets/tree/x1/tree.py b/apps/assets/tree/x1/tree.py
index 6e2cbfed9..1ff7491da 100644
--- a/apps/assets/tree/x1/tree.py
+++ b/apps/assets/tree/x1/tree.py
@@ -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