fix: fix data masking org problem

pull/16151/head
Aaron3S 2025-10-15 15:44:51 +08:00 committed by feng626
parent 1a9e56c520
commit 8f0a41b1a8
2 changed files with 5 additions and 9 deletions

View File

@ -1,6 +1,5 @@
from common.api import JMSBulkModelViewSet
from orgs.mixins.api import OrgBulkModelViewSet
from orgs.utils import tmp_to_root_org
from .common import ACLUserFilterMixin
from ..models import DataMaskingRule
from .. import serializers
@ -15,12 +14,8 @@ class DataMaskingRuleFilter(ACLUserFilterMixin):
fields = ('name', 'action')
class DataMaskingRuleViewSet(JMSBulkModelViewSet):
queryset = DataMaskingRule.objects.all()
class DataMaskingRuleViewSet(OrgBulkModelViewSet):
model = DataMaskingRule
filterset_class = DataMaskingRuleFilter
search_fields = ('name',)
serializer_class = serializers.DataMaskingRuleSerializer
def filter_queryset(self, queryset):
with tmp_to_root_org():
return super().filter_queryset(queryset)

View File

@ -3,12 +3,13 @@ from django.utils.translation import gettext_lazy as _
from acls.models import MaskingMethod, DataMaskingRule
from common.serializers.fields import LabeledChoiceField
from common.serializers.mixin import CommonBulkModelSerializer
from orgs.mixins.serializers import BulkOrgResourceModelSerializer
from .base import BaseUserAssetAccountACLSerializer as BaseSerializer
__all__ = ['DataMaskingRuleSerializer']
class DataMaskingRuleSerializer(BaseSerializer, CommonBulkModelSerializer):
class DataMaskingRuleSerializer(BaseSerializer, BulkOrgResourceModelSerializer):
masking_method = LabeledChoiceField(
choices=MaskingMethod.choices, default=MaskingMethod.fixed_char, label=_('Masking Method')
)