1
0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00

12552 add back add_related_count

This commit is contained in:
Arthur
2023-05-24 11:13:43 -07:00
parent 71a74e410d
commit a480a4c23b
6 changed files with 225 additions and 211 deletions

View File

@@ -99,14 +99,13 @@ class PassThroughPortMixin(object):
#
class RegionViewSet(NetBoxModelViewSet):
queryset = Region.objects.all().prefetch_related('tags')
# queryset = Region.objects.add_related_count(
# Region.objects.all(),
# Site,
# 'region',
# 'site_count',
# cumulative=True
# ).prefetch_related('tags')
queryset = Region.objects.add_related_count(
Region.objects.all(),
Site,
'region',
'site_count',
cumulative=True
).prefetch_related('tags')
serializer_class = serializers.RegionSerializer
filterset_class = filtersets.RegionFilterSet
@@ -116,14 +115,13 @@ class RegionViewSet(NetBoxModelViewSet):
#
class SiteGroupViewSet(NetBoxModelViewSet):
queryset = SiteGroup.objects.all().prefetch_related('tags')
# queryset = SiteGroup.objects.add_related_count(
# SiteGroup.objects.all(),
# Site,
# 'group',
# 'site_count',
# cumulative=True
# ).prefetch_related('tags')
queryset = SiteGroup.objects.add_related_count(
SiteGroup.objects.all(),
Site,
'group',
'site_count',
cumulative=True
).prefetch_related('tags')
serializer_class = serializers.SiteGroupSerializer
filterset_class = filtersets.SiteGroupFilterSet
@@ -152,20 +150,19 @@ class SiteViewSet(NetBoxModelViewSet):
#
class LocationViewSet(NetBoxModelViewSet):
queryset = Location.objects.all().prefetch_related('site', 'tags')
# queryset = Location.objects.add_related_count(
# Location.objects.add_related_count(
# Location.objects.all(),
# Device,
# 'location',
# 'device_count',
# cumulative=True
# ),
# Rack,
# 'location',
# 'rack_count',
# cumulative=True
# ).prefetch_related('site', 'tags')
queryset = Location.objects.add_related_count(
Location.objects.add_related_count(
Location.objects.all(),
Device,
'location',
'device_count',
cumulative=True
),
Rack,
'location',
'rack_count',
cumulative=True
).prefetch_related('site', 'tags')
serializer_class = serializers.LocationSerializer
filterset_class = filtersets.LocationFilterSet