mirror of
https://github.com/netbox-community/netbox.git
synced 2025-12-20 19:26:37 +00:00
b3f25a400b
* Enable dynamic field inclusion for REST API serializers * Recurse through nested serializer when resolving prefetches * Remove obsolete calls to prefetch_related() for API views * Remove support for brief_prefetch_fields viewset attribute * Rename query parameter * Fixes #15133: Fix FHRP group representation on assignments endpoint under brief mode (#15134) * Fixes #15133: Fix FHRP group representation on assignments endpoint under brief mode * Update API test * Restore get_queryset() on BriefModeMixin, minus prefetch logic * get_prefetches_for_serializer() should reference serializer field source if set
87 lines
2.6 KiB
Python
87 lines
2.6 KiB
Python
from rest_framework.routers import APIRootView
|
|
|
|
from circuits.models import Circuit
|
|
from dcim.models import Device, Rack, Site
|
|
from ipam.models import IPAddress, Prefix, VLAN, VRF
|
|
from netbox.api.viewsets import NetBoxModelViewSet, MPTTLockedMixin
|
|
from tenancy import filtersets
|
|
from tenancy.models import *
|
|
from utilities.utils import count_related
|
|
from virtualization.models import VirtualMachine, Cluster
|
|
from . import serializers
|
|
|
|
|
|
class TenancyRootView(APIRootView):
|
|
"""
|
|
Tenancy API root view
|
|
"""
|
|
def get_view_name(self):
|
|
return 'Tenancy'
|
|
|
|
|
|
#
|
|
# Tenants
|
|
#
|
|
|
|
class TenantGroupViewSet(MPTTLockedMixin, NetBoxModelViewSet):
|
|
queryset = TenantGroup.objects.add_related_count(
|
|
TenantGroup.objects.all(),
|
|
Tenant,
|
|
'group',
|
|
'tenant_count',
|
|
cumulative=True
|
|
)
|
|
serializer_class = serializers.TenantGroupSerializer
|
|
filterset_class = filtersets.TenantGroupFilterSet
|
|
|
|
|
|
class TenantViewSet(NetBoxModelViewSet):
|
|
queryset = Tenant.objects.annotate(
|
|
circuit_count=count_related(Circuit, 'tenant'),
|
|
device_count=count_related(Device, 'tenant'),
|
|
ipaddress_count=count_related(IPAddress, 'tenant'),
|
|
prefix_count=count_related(Prefix, 'tenant'),
|
|
rack_count=count_related(Rack, 'tenant'),
|
|
site_count=count_related(Site, 'tenant'),
|
|
virtualmachine_count=count_related(VirtualMachine, 'tenant'),
|
|
vlan_count=count_related(VLAN, 'tenant'),
|
|
vrf_count=count_related(VRF, 'tenant'),
|
|
cluster_count=count_related(Cluster, 'tenant')
|
|
)
|
|
serializer_class = serializers.TenantSerializer
|
|
filterset_class = filtersets.TenantFilterSet
|
|
|
|
|
|
#
|
|
# Contacts
|
|
#
|
|
|
|
class ContactGroupViewSet(MPTTLockedMixin, NetBoxModelViewSet):
|
|
queryset = ContactGroup.objects.add_related_count(
|
|
ContactGroup.objects.all(),
|
|
Contact,
|
|
'group',
|
|
'contact_count',
|
|
cumulative=True
|
|
)
|
|
serializer_class = serializers.ContactGroupSerializer
|
|
filterset_class = filtersets.ContactGroupFilterSet
|
|
|
|
|
|
class ContactRoleViewSet(NetBoxModelViewSet):
|
|
queryset = ContactRole.objects.all()
|
|
serializer_class = serializers.ContactRoleSerializer
|
|
filterset_class = filtersets.ContactRoleFilterSet
|
|
|
|
|
|
class ContactViewSet(NetBoxModelViewSet):
|
|
queryset = Contact.objects.all()
|
|
serializer_class = serializers.ContactSerializer
|
|
filterset_class = filtersets.ContactFilterSet
|
|
|
|
|
|
class ContactAssignmentViewSet(NetBoxModelViewSet):
|
|
queryset = ContactAssignment.objects.all()
|
|
serializer_class = serializers.ContactAssignmentSerializer
|
|
filterset_class = filtersets.ContactAssignmentFilterSet
|