perf: 代码优化

2021-07-28 14:33:02 +08:00 · 2021-07-28 14:33:02 +08:00 · da588ce0ae
parent d0680c3753
commit da588ce0ae
1 changed files with 9 additions and 7 deletions
--- a/apps/perms/utils/asset/user_permission.py
+++ b/apps/perms/utils/asset/user_permission.py
@ -549,7 +549,9 @@ class UserGrantedNodesQueryUtils(UserGrantedUtilsBase):
            return self.get_top_level_nodes()

        nodes = PermNode.objects.none()
-        if key != PermNode.FAVORITE_NODE_KEY:
+        if key == PermNode.FAVORITE_NODE_KEY:
+            return nodes
+
        node = PermNode.objects.get(key=key)
        granted_status = node.get_granted_status(self.user)
        if granted_status == NodeFrom.granted: