forked from uncloud/uncloud
94 lines
2.9 KiB
Python
94 lines
2.9 KiB
Python
from django.shortcuts import render
|
|
from django.contrib.auth import get_user_model
|
|
from django.contrib.auth.models import Group
|
|
|
|
from rest_framework import viewsets, permissions, generics
|
|
|
|
from rest_framework.views import APIView
|
|
from rest_framework.response import Response
|
|
|
|
from uncloud_vm.models import VMProduct
|
|
from .models import VMSnapshotProduct
|
|
from .serializers import UserSerializer, GroupSerializer, VMSnapshotSerializer, VMSnapshotCreateSerializer
|
|
|
|
|
|
import inspect
|
|
import sys
|
|
import re
|
|
|
|
|
|
# POST /vm/snapshot/ vmuuid=... => create snapshot, returns snapshot uuid
|
|
# GET /vm/snapshot => list
|
|
# DEL /vm/snapshot/<uuid:uuid> => delete
|
|
# create-list -> get, post => ListCreateAPIView
|
|
# del on other!
|
|
class VMSnapshotView(viewsets.ViewSet):
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
|
|
def list(self, request):
|
|
queryset = VMSnapshotProduct.objects.filter(owner=request.user)
|
|
serializer = VMSnapshotSerializer(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 = VMSnapshotSerializer(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)
|
|
|
|
|
|
|
|
# maybe drop or not --- we need something to guide the user!
|
|
# class ProductsViewSet(viewsets.ViewSet):
|
|
# permission_classes = [permissions.IsAuthenticated]
|
|
|
|
# def list(self, request):
|
|
|
|
# clsmembers = []
|
|
# for modules in [ 'uncloud_api.models', 'uncloud_vm.models' ]:
|
|
# clsmembers.extend(inspect.getmembers(sys.modules[modules], inspect.isclass))
|
|
|
|
|
|
# products = []
|
|
# for name, c in clsmembers:
|
|
# # Include everything that ends in Product, but not Product itself
|
|
# m = re.match(r'(?P<pname>.+)Product$', name)
|
|
# if m:
|
|
# products.append({
|
|
# 'name': m.group('pname'),
|
|
# 'description': c.description,
|
|
# 'recurring_period': c.recurring_period,
|
|
# 'pricing_model': c.pricing_model()
|
|
# }
|
|
# )
|
|
|
|
|
|
# return Response(products)
|
|
|
|
|
|
class UserViewSet(viewsets.ModelViewSet):
|
|
serializer_class = UserSerializer
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
|
|
def get_queryset(self):
|
|
return self.request.user
|