mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
|
from __future__ import unicode_literals
|
||
|
|
|
||
|
|
from rest_framework.viewsets import ModelViewSet
|
||
|
|
|
||
|
|
from extras.api.views import CustomFieldModelViewSet
|
||
|
|
from utilities.api import WritableSerializerMixin
|
||
|
|
from virtualization.models import Cluster, ClusterGroup, ClusterType, VirtualMachine, VMInterface
|
||
|
|
from . import serializers
|
||
|
|
|
||
|
|
|
||
|
|
#
|
||
|
|
# Clusters
|
||
|
|
#
|
||
|
|
|
||
|
|
class ClusterTypeViewSet(ModelViewSet):
|
||
|
|
queryset = ClusterType.objects.all()
|
||
|
|
serializer_class = serializers.ClusterTypeSerializer
|
||
|
|
|
||
|
|
|
||
|
|
class ClusterGroupViewSet(ModelViewSet):
|
||
|
|
queryset = ClusterGroup.objects.all()
|
||
|
|
serializer_class = serializers.ClusterGroupSerializer
|
||
|
|
|
||
|
|
|
||
|
|
class ClusterViewSet(WritableSerializerMixin, CustomFieldModelViewSet):
|
||
|
|
queryset = Cluster.objects.select_related('type', 'group')
|
||
|
|
serializer_class = serializers.ClusterSerializer
|
||
|
|
write_serializer_class = serializers.WritableClusterSerializer
|
||
|
|
|
||
|
|
|
||
|
|
#
|
||
|
|
# Virtual machines
|
||
|
|
#
|
||
|
|
|
||
|
|
class VirtualMachineViewSet(WritableSerializerMixin, CustomFieldModelViewSet):
|
||
|
|
queryset = VirtualMachine.objects.all()
|
||
|
|
serializer_class = serializers.VirtualMachineSerializer
|
||
|
|
write_serializer_class = serializers.WritableVirtualMachineSerializer
|
||
|
|
|
||
|
|
|
||
|
|
class VMInterfaceViewSet(WritableSerializerMixin, ModelViewSet):
|
||
|
|
queryset = VMInterface.objects.select_related('virtual_machine')
|
||
|
|
serializer_class = serializers.VMInterfaceSerializer
|
||
|
|
write_serializer_class = serializers.WritableVMInterfaceSerializer
|