mirror of https://github.com/jumpserver/jumpserver
fix: fix data masking org problem
parent
1a9e56c520
commit
8f0a41b1a8
|
|
@ -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)
|
||||
|
|
|
|||
|
|
@ -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')
|
||||
)
|
||||
|
|
|
|||
Loading…
Reference in New Issue