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.
55 lines
1.8 KiB
55 lines
1.8 KiB
from rest_framework.serializers import ModelSerializer |
|
from rest_framework import serializers |
|
|
|
from .utils import get_current_org |
|
from .models import Organization |
|
|
|
|
|
class ResourceStatisticsSerializer(serializers.Serializer): |
|
users_amount = serializers.IntegerField(required=False) |
|
groups_amount = serializers.IntegerField(required=False) |
|
|
|
assets_amount = serializers.IntegerField(required=False) |
|
nodes_amount = serializers.IntegerField(required=False) |
|
admin_users_amount = serializers.IntegerField(required=False) |
|
system_users_amount = serializers.IntegerField(required=False) |
|
domains_amount = serializers.IntegerField(required=False) |
|
gateways_amount = serializers.IntegerField(required=False) |
|
|
|
applications_amount = serializers.IntegerField(required=False) |
|
asset_perms_amount = serializers.IntegerField(required=False) |
|
app_perms_amount = serializers.IntegerField(required=False) |
|
|
|
|
|
class OrgSerializer(ModelSerializer): |
|
resource_statistics = ResourceStatisticsSerializer(source='resource_statistics_cache', read_only=True) |
|
|
|
class Meta: |
|
model = Organization |
|
fields_mini = ['id', 'name'] |
|
fields_small = fields_mini + [ |
|
'resource_statistics', |
|
'is_default', 'is_root', |
|
'date_created', 'created_by', |
|
'comment', 'created_by', |
|
] |
|
|
|
fields_m2m = [] |
|
fields = fields_small + fields_m2m |
|
read_only_fields = ['created_by', 'date_created'] |
|
|
|
|
|
class CurrentOrgSerializer(ModelSerializer): |
|
class Meta: |
|
model = Organization |
|
fields = ['id', 'name', 'is_default', 'is_root', 'comment'] |
|
|
|
|
|
class CurrentOrgDefault: |
|
requires_context = False |
|
|
|
def __call__(self, *args): |
|
return get_current_org() |
|
|
|
def __repr__(self): |
|
return '%s()' % self.__class__.__name__
|
|
|