mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
ade844f7a7
Mitgate invalidation issues by using prefetch_related instead of select_related. Also use invalidated_update instead of just update.
53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
from circuits.models import Circuit
|
|
from dcim.models import Device, Rack, Site
|
|
from extras.api.views import CustomFieldModelViewSet
|
|
from ipam.models import IPAddress, Prefix, VLAN, VRF
|
|
from tenancy import filters
|
|
from tenancy.models import Tenant, TenantGroup
|
|
from utilities.api import FieldChoicesViewSet, ModelViewSet
|
|
from utilities.utils import get_subquery
|
|
from virtualization.models import VirtualMachine
|
|
from . import serializers
|
|
|
|
|
|
#
|
|
# Field choices
|
|
#
|
|
|
|
class TenancyFieldChoicesViewSet(FieldChoicesViewSet):
|
|
fields = ()
|
|
|
|
|
|
#
|
|
# Tenant Groups
|
|
#
|
|
|
|
class TenantGroupViewSet(ModelViewSet):
|
|
queryset = TenantGroup.objects.annotate(
|
|
tenant_count=get_subquery(Tenant, 'group')
|
|
)
|
|
serializer_class = serializers.TenantGroupSerializer
|
|
filterset_class = filters.TenantGroupFilter
|
|
|
|
|
|
#
|
|
# Tenants
|
|
#
|
|
|
|
class TenantViewSet(CustomFieldModelViewSet):
|
|
queryset = Tenant.objects.prefetch_related(
|
|
'group', 'tags'
|
|
).annotate(
|
|
circuit_count=get_subquery(Circuit, 'tenant'),
|
|
device_count=get_subquery(Device, 'tenant'),
|
|
ipaddress_count=get_subquery(IPAddress, 'tenant'),
|
|
prefix_count=get_subquery(Prefix, 'tenant'),
|
|
rack_count=get_subquery(Rack, 'tenant'),
|
|
site_count=get_subquery(Site, 'tenant'),
|
|
virtualmachine_count=get_subquery(VirtualMachine, 'tenant'),
|
|
vlan_count=get_subquery(VLAN, 'tenant'),
|
|
vrf_count=get_subquery(VRF, 'tenant')
|
|
)
|
|
serializer_class = serializers.TenantSerializer
|
|
filterset_class = filters.TenantFilter
|