mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from rest_framework.viewsets import ModelViewSet
|
|
|
|
from dcim.models import Interface
|
|
from extras.api.views import CustomFieldModelViewSet
|
|
from utilities.api import WritableSerializerMixin
|
|
from virtualization import filters
|
|
from virtualization.models import Cluster, ClusterGroup, ClusterType, VirtualMachine
|
|
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
|
|
filter_class = filters.ClusterFilter
|
|
|
|
|
|
#
|
|
# Virtual machines
|
|
#
|
|
|
|
class VirtualMachineViewSet(WritableSerializerMixin, CustomFieldModelViewSet):
|
|
queryset = VirtualMachine.objects.all()
|
|
serializer_class = serializers.VirtualMachineSerializer
|
|
write_serializer_class = serializers.WritableVirtualMachineSerializer
|
|
filter_class = filters.VirtualMachineFilter
|
|
|
|
|
|
class InterfaceViewSet(WritableSerializerMixin, ModelViewSet):
|
|
queryset = Interface.objects.filter(virtual_machine__isnull=False).select_related('virtual_machine')
|
|
serializer_class = serializers.InterfaceSerializer
|
|
write_serializer_class = serializers.WritableInterfaceSerializer
|