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

Remove separate config-context API endpoints; include rendered config context when retrieving a single device/VM

This commit is contained in:
Jeremy Stretch
2018-07-03 12:29:20 -04:00
parent 6ddbd79fe6
commit 25b36d6d42
4 changed files with 42 additions and 14 deletions

View File

@ -1,9 +1,5 @@
from __future__ import unicode_literals
from django.shortcuts import get_object_or_404
from rest_framework.decorators import action
from rest_framework.response import Response
from dcim.models import Interface
from extras.api.views import CustomFieldModelViewSet
from utilities.api import FieldChoicesViewSet, ModelViewSet
@ -50,13 +46,15 @@ class ClusterViewSet(CustomFieldModelViewSet):
class VirtualMachineViewSet(CustomFieldModelViewSet):
queryset = VirtualMachine.objects.all()
serializer_class = serializers.VirtualMachineSerializer
filter_class = filters.VirtualMachineFilter
@action(detail=True, url_path='config-context')
def config_context(self, request, pk):
device = get_object_or_404(VirtualMachine, pk=pk)
return Response(device.get_config_context())
def get_serializer_class(self):
"""
Include rendered config context when retrieving a single VirtualMachine.
"""
if self.action == 'retrieve':
return serializers.VirtualMachineWithConfigContextSerializer
return serializers.VirtualMachineSerializer
class InterfaceViewSet(ModelViewSet):