From ca898ed7b5cd96b5d52e6ed6c39f25391cbc4f43 Mon Sep 17 00:00:00 2001 From: Aaron3S Date: Tue, 18 Apr 2023 19:51:36 +0800 Subject: [PATCH] =?UTF-8?q?perf:=20=E4=BC=98=E5=8C=96=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/ops/serializers/job.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/apps/ops/serializers/job.py b/apps/ops/serializers/job.py index cc690d812..8b157b66f 100644 --- a/apps/ops/serializers/job.py +++ b/apps/ops/serializers/job.py @@ -4,6 +4,7 @@ from django.utils.translation import gettext_lazy as _ from rest_framework import serializers from assets.models import Node, Asset +from perms.models import PermNode from perms.utils.user_perm import UserPermAssetUtil from common.serializers.fields import ReadableHiddenField from ops.mixin import PeriodTaskSerializerMixin @@ -39,9 +40,9 @@ class JobSerializer(BulkOrgResourceModelSerializer, PeriodTaskSerializerMixin): user = self.get_request_user() perm_util = UserPermAssetUtil(user=user) for node_id in node_ids: - if node_id == 'favorite': + if node_id == PermNode.FAVORITE_NODE_KEY: node_assets = perm_util.get_favorite_assets() - elif node_id == 'ungroup': + elif node_id == PermNode.UNGROUPED_NODE_KEY: node_assets = perm_util.get_ungroup_assets() else: node, node_assets = perm_util.get_node_all_assets(node_id)