mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
from django.contrib.auth.models import Group, User
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from rest_framework import serializers
|
|
|
|
from netbox.api import ContentTypeField, WritableNestedSerializer
|
|
from users.models import ObjectPermission
|
|
|
|
__all__ = [
|
|
'NestedGroupSerializer',
|
|
'NestedObjectPermissionSerializer',
|
|
'NestedUserSerializer',
|
|
]
|
|
|
|
|
|
class NestedGroupSerializer(WritableNestedSerializer):
|
|
url = serializers.HyperlinkedIdentityField(view_name='users-api:group-detail')
|
|
|
|
class Meta:
|
|
model = Group
|
|
fields = ['id', 'url', 'display', 'name']
|
|
|
|
|
|
class NestedUserSerializer(WritableNestedSerializer):
|
|
url = serializers.HyperlinkedIdentityField(view_name='users-api:user-detail')
|
|
|
|
class Meta:
|
|
model = User
|
|
fields = ['id', 'url', 'display', 'username']
|
|
|
|
|
|
class NestedObjectPermissionSerializer(WritableNestedSerializer):
|
|
url = serializers.HyperlinkedIdentityField(view_name='users-api:objectpermission-detail')
|
|
object_types = ContentTypeField(
|
|
queryset=ContentType.objects.all(),
|
|
many=True
|
|
)
|
|
groups = serializers.SerializerMethodField(read_only=True)
|
|
users = serializers.SerializerMethodField(read_only=True)
|
|
|
|
class Meta:
|
|
model = ObjectPermission
|
|
fields = ['id', 'url', 'display', 'name', 'enabled', 'object_types', 'groups', 'users', 'actions']
|
|
|
|
def get_groups(self, obj):
|
|
return [g.name for g in obj.groups.all()]
|
|
|
|
def get_users(self, obj):
|
|
return [u.username for u in obj.users.all()]
|