Phase in new beta/vm view for creating vms + orders + bills

uncloud_django_based/uncloud/uncloud_vm/views.py

@@ -179,6 +179,46 @@ class VMProductViewSet(ProductViewSet):
         return Response(VMProductSerializer(vm, context={'request': request}).data)
 
 
+class NicoVMProductViewSet(ProductViewSet):
+    permission_classes = [permissions.IsAuthenticated]
+    serializer_class = NicoVMProductSerializer
+
+    def get_queryset(self):
+        obj = VMProduct.objects.filter(owner=self.request.user)
+
+        return obj
+
+    # Use a database transaction so that we do not get half-created structure
+    # if something goes wrong.
+    @transaction.atomic
+    def create(self, request):
+        # Extract serializer data.
+        serializer = self.get_serializer(data=request.data)
+        serializer.is_valid(raise_exception=True)
+        order_recurring_period = serializer.validated_data.pop("recurring_period")
+        order_billing_address = serializer.validated_data.pop("billing_address")
+
+        # Create base order.
+        order = Order(
+                recurring_period=order_recurring_period,
+                billing_address=order_billing_address,
+                owner=request.user,
+                starting_date=timezone.now()
+                )
+        order.save()
+
+        # Create disk image.
+        disk = VMDiskProduct(owner=request.user, order=order,
+                **serializer.validated_data.pop("primary_disk"))
+
+        # Create VM.
+        vm = serializer.save(owner=request.user, order=order, primary_disk=disk)
+        disk.vm = vm
+        disk.save()
+
+        return Response(VMProductSerializer(vm, context={'request': request}).data)
+
+
 ###
 # Admin stuff.