mirror of https://github.com/jumpserver/jumpserver
62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
from django.utils.translation import gettext_lazy as _
|
|
from rest_framework import serializers
|
|
|
|
from common.serializers.fields import ObjectRelatedField
|
|
from users.models import User
|
|
from orgs.serializers import CurrentOrgDefault
|
|
from ..models import RoleBinding, SystemRoleBinding, OrgRoleBinding
|
|
|
|
__all__ = [
|
|
'RoleBindingSerializer', 'OrgRoleBindingSerializer', 'SystemRoleBindingSerializer'
|
|
]
|
|
|
|
|
|
class RoleBindingSerializer(serializers.ModelSerializer):
|
|
user = ObjectRelatedField(
|
|
required=False, queryset=User.objects,
|
|
label=_('User'), attrs=('id', 'name', 'username')
|
|
)
|
|
|
|
class Meta:
|
|
model = RoleBinding
|
|
fields = [
|
|
'id', 'user', 'role', 'scope', 'org', 'org_name',
|
|
]
|
|
read_only_fields = ['scope']
|
|
extra_kwargs = {
|
|
'org_name': {'label': _("Org name")}
|
|
}
|
|
|
|
|
|
class SystemRoleBindingSerializer(RoleBindingSerializer):
|
|
org = None
|
|
|
|
class Meta(RoleBindingSerializer.Meta):
|
|
model = SystemRoleBinding
|
|
|
|
def get_field_names(self, *args):
|
|
names = super().get_field_names(*args)
|
|
return list(set(names) - {'org'})
|
|
|
|
|
|
class OrgRoleBindingSerializer(RoleBindingSerializer):
|
|
org = serializers.PrimaryKeyRelatedField(
|
|
default=CurrentOrgDefault(), label=_("Organization"), read_only=True
|
|
)
|
|
|
|
class Meta(RoleBindingSerializer.Meta):
|
|
model = OrgRoleBinding
|
|
validators = []
|
|
|
|
def validate(self, attrs):
|
|
data = self.initial_data
|
|
many = isinstance(data, list)
|
|
if not many:
|
|
user = attrs.get('user')
|
|
role = attrs.get('role')
|
|
role_bindings = OrgRoleBinding.objects.filter(user=user, role=role)
|
|
|
|
if not self.instance and role_bindings.exists():
|
|
raise serializers.ValidationError({'role': _('Has bound this role')})
|
|
return attrs
|