vmhosts, restructure urls, etc.

uncloud/uncloud_vm/views.py

@@ -6,13 +6,24 @@ from django.shortcuts import get_object_or_404
 from rest_framework import viewsets, permissions
 from rest_framework.response import Response
 
-
-from opennebula.models import VM as OpenNebulaVM
-
-from .models import VMHost
-from .serializers import VMHostSerializer
+from .models import VMHost, VMProduct
+from .serializers import VMHostSerializer, VMProductSerializer
 
 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)