fix: 修复xpack引入包问题

This commit is contained in:
feng626
2021-10-29 15:29:57 +08:00
committed by Jiangjie.Bai
parent 15cdd44c6c
commit 6e42a9704c

View File

@@ -2,6 +2,7 @@ from django.utils.translation import ugettext as _
from rest_framework import serializers from rest_framework import serializers
from common.drf.serializers import BulkModelSerializer from common.drf.serializers import BulkModelSerializer
from common.drf.serializers import MethodSerializer from common.drf.serializers import MethodSerializer
from jumpserver.utils import has_valid_xpack_license
from ..models import LoginACL from ..models import LoginACL
from .rules import RuleSerializer from .rules import RuleSerializer
@@ -40,12 +41,11 @@ class LoginACLSerializer(BulkModelSerializer):
self.set_action_choices() self.set_action_choices()
def set_action_choices(self): def set_action_choices(self):
from xpack.plugins.license.models import License
action = self.fields.get('action') action = self.fields.get('action')
if not action: if not action:
return return
choices = action._choices choices = action._choices
if not License.has_valid_license(): if not has_valid_xpack_license():
choices.pop(LoginACL.ActionChoices.confirm, None) choices.pop(LoginACL.ActionChoices.confirm, None)
action._choices = choices action._choices = choices