netbox-community-netbox/netbox/circuits/api/views.py

from rest_framework import generics

from circuits.models import Provider, CircuitType, Circuit
from circuits.filters import CircuitFilter

from . import serializers


class ProviderListView(generics.ListAPIView):
    """
    List all providers
    """
    queryset = Provider.objects.all()
    serializer_class = serializers.ProviderSerializer


class ProviderDetailView(generics.RetrieveAPIView):
    """
    Retrieve a single provider
    """
    queryset = Provider.objects.all()
    serializer_class = serializers.ProviderSerializer


class CircuitTypeListView(generics.ListAPIView):
    """
    List all circuit types
    """
    queryset = CircuitType.objects.all()
    serializer_class = serializers.CircuitTypeSerializer


class CircuitTypeDetailView(generics.RetrieveAPIView):
    """
    Retrieve a single circuit type
    """
    queryset = CircuitType.objects.all()
    serializer_class = serializers.CircuitTypeSerializer


class CircuitListView(generics.ListAPIView):
    """
    List circuits (filterable)
    """
    queryset = Circuit.objects.select_related('type', 'provider', 'site', 'interface__device')
    serializer_class = serializers.CircuitSerializer
    filter_class = CircuitFilter


class CircuitDetailView(generics.RetrieveAPIView):
    """
    Retrieve a single circuit
    """
    queryset = Circuit.objects.select_related('type', 'provider', 'site', 'interface__device')
    serializer_class = serializers.CircuitSerializer