mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
31 lines
801 B
Python
31 lines
801 B
Python
from __future__ import unicode_literals
|
|
|
|
from rest_framework.viewsets import ModelViewSet
|
|
|
|
from extras.api.views import CustomFieldModelViewSet
|
|
from tenancy import filters
|
|
from tenancy.models import Tenant, TenantGroup
|
|
from utilities.api import WritableSerializerMixin
|
|
from . import serializers
|
|
|
|
|
|
#
|
|
# Tenant Groups
|
|
#
|
|
|
|
class TenantGroupViewSet(ModelViewSet):
|
|
queryset = TenantGroup.objects.all()
|
|
serializer_class = serializers.TenantGroupSerializer
|
|
filter_class = filters.TenantGroupFilter
|
|
|
|
|
|
#
|
|
# Tenants
|
|
#
|
|
|
|
class TenantViewSet(WritableSerializerMixin, CustomFieldModelViewSet):
|
|
queryset = Tenant.objects.select_related('group')
|
|
serializer_class = serializers.TenantSerializer
|
|
write_serializer_class = serializers.WritableTenantSerializer
|
|
filter_class = filters.TenantFilter
|