28 lines
973 B
Python
28 lines
973 B
Python
from rest_framework import viewsets, permissions
|
|
from rest_framework.response import Response
|
|
from django.shortcuts import get_object_or_404
|
|
|
|
from .models import VM
|
|
from .serializers import VMSerializer, OpenNebulaVMSerializer
|
|
|
|
|
|
class RawVMViewSet(viewsets.ModelViewSet):
|
|
queryset = VM.objects.all()
|
|
serializer_class = VMSerializer
|
|
permission_classes = [permissions.IsAdminUser]
|
|
|
|
|
|
class VMViewSet(viewsets.ViewSet):
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
|
|
def list(self, request):
|
|
queryset = VM.objects.filter(owner=request.user)
|
|
serializer = OpenNebulaVMSerializer(queryset, many=True, context={'request': request})
|
|
return Response(serializer.data)
|
|
|
|
def retrieve(self, request, pk=None):
|
|
queryset = VM.objects.filter(owner=request.user)
|
|
vm = get_object_or_404(queryset, pk=pk)
|
|
serializer = OpenNebulaVMSerializer(vm, context={'request': request})
|
|
return Response(serializer.data)
|