From 5ac1467564fc6cd816dd457b213d8ff1e1b09333 Mon Sep 17 00:00:00 2001
From: ibuler <ibuler@qq.com>
Date: Thu, 19 Mar 2020 15:41:41 +0800
Subject: [PATCH] =?UTF-8?q?[Update]=20=E4=BF=AE=E5=A4=8D=E6=94=B6=E8=97=8F?=
 =?UTF-8?q?=E6=95=B0=E9=87=8F=E4=B8=8D=E5=AF=B9=E7=9A=84=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/perms/utils/asset_permission.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/apps/perms/utils/asset_permission.py b/apps/perms/utils/asset_permission.py
index 1905bfe62..792b842af 100644
--- a/apps/perms/utils/asset_permission.py
+++ b/apps/perms/utils/asset_permission.py
@@ -315,10 +315,13 @@ class AssetPermissionUtil(AssetPermissionUtilCacheMixin):
             identifier=node_key, tag=node_value,
             parent=user_tree.root,
         )
+        node = user_tree.get_node(node_key)
         assets_id = FavoriteAsset.get_user_favorite_assets_id(self.object)
         all_valid_assets = user_tree.all_valid_assets(user_tree.root)
         valid_assets_id = set(assets_id) & all_valid_assets
         user_tree.set_assets(node_key, valid_assets_id)
+        # 必须设置这个，否则看不到个数
+        node.data['all_assets'] = None
 
     def set_user_tree_to_local(self, user_tree):
         self._user_tree = user_tree