Moved graph views into model viewsets

netbox/circuits/api/views.py

@@ -1,13 +1,17 @@
 from django.shortcuts import get_object_or_404
 
+from rest_framework.decorators import detail_route
 from rest_framework.mixins import (
     CreateModelMixin, DestroyModelMixin, ListModelMixin, RetrieveModelMixin, UpdateModelMixin,
 )
+from rest_framework.response import Response
 from rest_framework.viewsets import GenericViewSet, ModelViewSet
 
 from circuits.models import Provider, CircuitTermination, CircuitType, Circuit
 from circuits.filters import CircuitFilter
 
+from extras.models import Graph, GRAPH_TYPE_PROVIDER
+from extras.api.serializers import GraphSerializer
 from extras.api.views import CustomFieldModelViewSet
 from utilities.api import WritableSerializerMixin
 from . import serializers
@@ -21,6 +25,13 @@ class ProviderViewSet(CustomFieldModelViewSet):
     queryset = Provider.objects.all()
     serializer_class = serializers.ProviderSerializer
 
+    @detail_route()
+    def graphs(self, request, pk=None):
+        provider = get_object_or_404(Provider, pk=pk)
+        queryset = Graph.objects.filter(type=GRAPH_TYPE_PROVIDER)
+        serializer = GraphSerializer(queryset, many=True, context={'graphed_object': provider})
+        return Response(serializer.data)
+
 
 #
 #  Circuit Types