fix: 修复资产列表有时的bug

32ed3854 · ibuler · 老广 · b51eb426 · 32ed3854
Commit 32ed3854 authored 4 years ago by ibuler Committed by 老广 4 years ago
Hide whitespace changes
Inline Side-by-side

Showing

with 4 additions and 0 deletions
+4 -0
--- a/apps/assets/utils.py
+++ b/apps/assets/utils.py
@@ -125,6 +125,8 @@ class TreeService(Tree):

    def assets(self, nid):
        node = self.get_node(nid)
+        if not node:
+            return set()
        return node.data.get("assets", set())

    def valid_assets(self, nid):
@@ -132,6 +134,8 @@ class TreeService(Tree):

    def all_assets(self, nid):
        node = self.get_node(nid)
+        if not node:
+            return set()
        if node.data is None:
            node.data = {}
        all_assets = node.data.get("all_assets")