uncloud-mravi/uncloud/uncloud_vm/views.py

from django.shortcuts import render

from django.contrib.auth.models import User
from django.shortcuts import get_object_or_404

from rest_framework import viewsets, permissions
from rest_framework.response import Response

from .models import VMHost, VMProduct
from .serializers import VMHostSerializer, VMProductSerializer, VMSnapshotProductSerializer

class VMHostViewSet(viewsets.ModelViewSet):
    serializer_class = VMHostSerializer
    queryset = VMHost.objects.all()
    permission_classes = [permissions.IsAdminUser]

class VMProductViewSet(viewsets.ModelViewSet):
    permission_classes = [permissions.IsAuthenticated]
    serializer_class = VMProductSerializer

    def get_queryset(self):
        return VMProduct.objects.filter(owner=self.request.user)

    def create(self, request):
        serializer = VMProductSerializer(data=request.data, context={'request': request})
        serializer.is_valid(raise_exception=True)
        serializer.save(owner=request.user)

        return Response(serializer.data)


class VMSnapshotProductViewSet(viewsets.ModelViewSet):
    permission_classes = [permissions.IsAuthenticated]
    serializer_class = VMSnapshotProductSerializer

    def get_queryset(self):
        return VMSnapshotProduct.objects.filter(owner=self.request.user)

    def create(self, request):
        serializer = VMProductSerializer(data=request.data, context={'request': request})
        serializer.is_valid(raise_exception=True)
        serializer.save(owner=request.user)

        return Response(serializer.data)


class VMSnapshotProductView(viewsets.ViewSet):
    permission_classes = [permissions.IsAuthenticated]

    def list(self, request):
        queryset = VMSnapshotProduct.objects.filter(owner=request.user)
        serializer = VMSnapshotProductSerializer(queryset, many=True, context={'request': request})
        return Response(serializer.data)

    def retrieve(self, request, pk=None):
        queryset = VMSnapshotProduct.objects.filter(owner=request.user)
        vm = get_object_or_404(queryset, pk=pk)
        serializer = VMSnapshotProductSerializer(vm, context={'request': request})
        return Response(serializer.data)

    def create(self, request):
        print(request.data)
        serializer = VMSnapshotCreateSerializer(data=request.data)

        serializer.gb_ssd = 12
        serializer.gb_hdd = 120
        print("F")
        serializer.is_valid(raise_exception=True)

        print(serializer)
        print("A")
        serializer.save()
        print("B")


        # snapshot = VMSnapshotProduct(owner=request.user,
        #                              **serialzer.data)

        return Response(serializer.data)