mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
77 lines
2.0 KiB
Python
77 lines
2.0 KiB
Python
from rest_framework.routers import APIRootView
|
|
|
|
from circuits import filtersets
|
|
from circuits.models import *
|
|
from dcim.api.views import PassThroughPortMixin
|
|
from extras.api.views import CustomFieldModelViewSet
|
|
from netbox.api.views import ModelViewSet
|
|
from utilities.utils import count_related
|
|
from . import serializers
|
|
|
|
|
|
class CircuitsRootView(APIRootView):
|
|
"""
|
|
Circuits API root view
|
|
"""
|
|
def get_view_name(self):
|
|
return 'Circuits'
|
|
|
|
|
|
#
|
|
# Providers
|
|
#
|
|
|
|
class ProviderViewSet(CustomFieldModelViewSet):
|
|
queryset = Provider.objects.prefetch_related('tags').annotate(
|
|
circuit_count=count_related(Circuit, 'provider')
|
|
)
|
|
serializer_class = serializers.ProviderSerializer
|
|
filterset_class = filtersets.ProviderFilterSet
|
|
|
|
|
|
#
|
|
# Circuit Types
|
|
#
|
|
|
|
class CircuitTypeViewSet(CustomFieldModelViewSet):
|
|
queryset = CircuitType.objects.annotate(
|
|
circuit_count=count_related(Circuit, 'type')
|
|
)
|
|
serializer_class = serializers.CircuitTypeSerializer
|
|
filterset_class = filtersets.CircuitTypeFilterSet
|
|
|
|
|
|
#
|
|
# Circuits
|
|
#
|
|
|
|
class CircuitViewSet(CustomFieldModelViewSet):
|
|
queryset = Circuit.objects.prefetch_related(
|
|
'type', 'tenant', 'provider', 'termination_a', 'termination_z'
|
|
).prefetch_related('tags')
|
|
serializer_class = serializers.CircuitSerializer
|
|
filterset_class = filtersets.CircuitFilterSet
|
|
|
|
|
|
#
|
|
# Circuit Terminations
|
|
#
|
|
|
|
class CircuitTerminationViewSet(PassThroughPortMixin, ModelViewSet):
|
|
queryset = CircuitTermination.objects.prefetch_related(
|
|
'circuit', 'site', 'provider_network', 'cable'
|
|
)
|
|
serializer_class = serializers.CircuitTerminationSerializer
|
|
filterset_class = filtersets.CircuitTerminationFilterSet
|
|
brief_prefetch_fields = ['circuit']
|
|
|
|
|
|
#
|
|
# Provider networks
|
|
#
|
|
|
|
class ProviderNetworkViewSet(CustomFieldModelViewSet):
|
|
queryset = ProviderNetwork.objects.prefetch_related('tags')
|
|
serializer_class = serializers.ProviderNetworkSerializer
|
|
filterset_class = filtersets.ProviderNetworkFilterSet
|