mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
74 lines
2.2 KiB
Python
74 lines
2.2 KiB
Python
from django.db.models import Count, Prefetch
|
|
from django.shortcuts import get_object_or_404
|
|
from rest_framework.decorators import action
|
|
from rest_framework.response import Response
|
|
|
|
from circuits import filters
|
|
from circuits.models import Provider, CircuitTermination, CircuitType, Circuit
|
|
from extras.api.serializers import RenderedGraphSerializer
|
|
from extras.api.views import CustomFieldModelViewSet
|
|
from extras.models import Graph
|
|
from utilities.api import ModelViewSet
|
|
from . import serializers
|
|
|
|
|
|
#
|
|
# Providers
|
|
#
|
|
|
|
class ProviderViewSet(CustomFieldModelViewSet):
|
|
queryset = Provider.objects.prefetch_related('tags').annotate(
|
|
circuit_count=Count('circuits')
|
|
)
|
|
serializer_class = serializers.ProviderSerializer
|
|
filterset_class = filters.ProviderFilterSet
|
|
|
|
@action(detail=True)
|
|
def graphs(self, request, pk):
|
|
"""
|
|
A convenience method for rendering graphs for a particular provider.
|
|
"""
|
|
provider = get_object_or_404(self.queryset, pk=pk)
|
|
queryset = Graph.objects.restrict(request.user).filter(type__model='provider')
|
|
serializer = RenderedGraphSerializer(queryset, many=True, context={'graphed_object': provider})
|
|
return Response(serializer.data)
|
|
|
|
|
|
#
|
|
# Circuit Types
|
|
#
|
|
|
|
class CircuitTypeViewSet(ModelViewSet):
|
|
queryset = CircuitType.objects.annotate(
|
|
circuit_count=Count('circuits')
|
|
)
|
|
serializer_class = serializers.CircuitTypeSerializer
|
|
filterset_class = filters.CircuitTypeFilterSet
|
|
|
|
|
|
#
|
|
# Circuits
|
|
#
|
|
|
|
class CircuitViewSet(CustomFieldModelViewSet):
|
|
queryset = Circuit.objects.prefetch_related(
|
|
Prefetch('terminations', queryset=CircuitTermination.objects.unrestricted().prefetch_related(
|
|
'site', 'connected_endpoint__device'
|
|
)),
|
|
'type', 'tenant', 'provider',
|
|
).prefetch_related('tags')
|
|
serializer_class = serializers.CircuitSerializer
|
|
filterset_class = filters.CircuitFilterSet
|
|
|
|
|
|
#
|
|
# Circuit Terminations
|
|
#
|
|
|
|
class CircuitTerminationViewSet(ModelViewSet):
|
|
queryset = CircuitTermination.objects.prefetch_related(
|
|
'circuit', 'site', 'connected_endpoint__device', 'cable'
|
|
)
|
|
serializer_class = serializers.CircuitTerminationSerializer
|
|
filterset_class = filters.CircuitTerminationFilterSet
|