mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
96 lines
3.3 KiB
96 lines
3.3 KiB
|
|
import re
|
|
from rest_framework.serializers import ModelSerializer
|
|
from rest_framework import serializers
|
|
from users.models import User, UserGroup
|
|
from assets.models import Asset, Domain, AdminUser, SystemUser, Label
|
|
from assets.const import (
|
|
GENERAL_FORBIDDEN_SPECIAL_CHARACTERS_PATTERN,
|
|
GENERAL_FORBIDDEN_SPECIAL_CHARACTERS_ERROR_MSG
|
|
)
|
|
from perms.models import AssetPermission
|
|
from common.serializers import AdaptedBulkListSerializer
|
|
from .utils import set_current_org, get_current_org
|
|
from .models import Organization
|
|
from .mixins.serializers import OrgMembershipSerializerMixin
|
|
|
|
|
|
class OrgSerializer(ModelSerializer):
|
|
class Meta:
|
|
model = Organization
|
|
list_serializer_class = AdaptedBulkListSerializer
|
|
fields = '__all__'
|
|
read_only_fields = ['created_by', 'date_created']
|
|
|
|
@staticmethod
|
|
def validate_name(name):
|
|
pattern = GENERAL_FORBIDDEN_SPECIAL_CHARACTERS_PATTERN
|
|
res = re.search(pattern, name)
|
|
if res is not None:
|
|
msg = GENERAL_FORBIDDEN_SPECIAL_CHARACTERS_ERROR_MSG
|
|
raise serializers.ValidationError(msg)
|
|
return name
|
|
|
|
|
|
class OrgReadSerializer(ModelSerializer):
|
|
admins = serializers.SlugRelatedField(slug_field='name', many=True, read_only=True)
|
|
auditors = serializers.SlugRelatedField(slug_field='name', many=True, read_only=True)
|
|
users = serializers.SlugRelatedField(slug_field='name', many=True, read_only=True)
|
|
user_groups = serializers.SerializerMethodField()
|
|
assets = serializers.SerializerMethodField()
|
|
domains = serializers.SerializerMethodField()
|
|
admin_users = serializers.SerializerMethodField()
|
|
system_users = serializers.SerializerMethodField()
|
|
labels = serializers.SerializerMethodField()
|
|
perms = serializers.SerializerMethodField()
|
|
|
|
class Meta:
|
|
model = Organization
|
|
fields = '__all__'
|
|
|
|
@staticmethod
|
|
def get_data_from_model(obj, model):
|
|
current_org = get_current_org()
|
|
set_current_org(Organization.root())
|
|
if model == Asset:
|
|
data = [o.hostname for o in model.objects.filter(org_id=obj.id)]
|
|
else:
|
|
data = [o.name for o in model.objects.filter(org_id=obj.id)]
|
|
set_current_org(current_org)
|
|
return data
|
|
|
|
def get_user_groups(self, obj):
|
|
return self.get_data_from_model(obj, UserGroup)
|
|
|
|
def get_assets(self, obj):
|
|
return self.get_data_from_model(obj, Asset)
|
|
|
|
def get_domains(self, obj):
|
|
return self.get_data_from_model(obj, Domain)
|
|
|
|
def get_admin_users(self, obj):
|
|
return self.get_data_from_model(obj, AdminUser)
|
|
|
|
def get_system_users(self, obj):
|
|
return self.get_data_from_model(obj, SystemUser)
|
|
|
|
def get_labels(self, obj):
|
|
return self.get_data_from_model(obj, Label)
|
|
|
|
def get_perms(self, obj):
|
|
return self.get_data_from_model(obj, AssetPermission)
|
|
|
|
|
|
class OrgMembershipAdminSerializer(OrgMembershipSerializerMixin, ModelSerializer):
|
|
class Meta:
|
|
model = Organization.admins.through
|
|
list_serializer_class = AdaptedBulkListSerializer
|
|
fields = '__all__'
|
|
|
|
|
|
class OrgMembershipUserSerializer(OrgMembershipSerializerMixin, ModelSerializer):
|
|
class Meta:
|
|
model = Organization.users.through
|
|
list_serializer_class = AdaptedBulkListSerializer
|
|
fields = '__all__'
|