Misc cleanup

This commit is contained in:
Jeremy Stretch
2019-04-11 10:49:43 -04:00
parent 3d77eed6e2
commit acaf05c090
5 changed files with 44 additions and 8 deletions

View File

@@ -544,7 +544,7 @@ class VirtualChassisViewSet(ModelViewSet):
#
class PowerPanelViewSet(ModelViewSet):
queryset = PowerPanel.objects.all()
queryset = PowerPanel.objects.select_related('site', 'rack_group')
serializer_class = serializers.PowerPanelSerializer
filterset_class = filters.PowerPanelFilter
@@ -553,8 +553,8 @@ class PowerPanelViewSet(ModelViewSet):
# Power feeds
#
class PowerFeedViewSet(ModelViewSet):
queryset = PowerFeed.objects.all()
class PowerFeedViewSet(CustomFieldModelViewSet):
queryset = PowerFeed.objects.select_related('power_panel', 'rack').prefetch_related('tags')
serializer_class = serializers.PowerFeedSerializer
filterset_class = filters.PowerFeedFilter