Quickly wire vm creation to orders

uncloud/uncloud_vm/views.py

@@ -7,8 +7,7 @@ from rest_framework import viewsets, permissions
 from rest_framework.response import Response
 
 from .models import VMHost, VMProduct, VMSnapshotProduct
-from uncloud_pay.models import Order
-
+from uncloud_pay.models import Order, RecurringPeriod
 from .serializers import VMHostSerializer, VMProductSerializer, VMSnapshotProductSerializer
 
 import datetime
@@ -27,9 +26,23 @@ class VMProductViewSet(viewsets.ModelViewSet):
         return VMProduct.objects.filter(owner=self.request.user)
 
     def create(self, request):
+        # Create base order.
+        order = Order.objects.create(
+                recurring_period=RecurringPeriod.PER_MONTH,
+                recurring_price=0,
+                one_time_price=0,
+                owner=request.user
+                )
+
+        # Create VM.
         serializer = VMProductSerializer(data=request.data, context={'request': request})
         serializer.is_valid(raise_exception=True)
-        serializer.save(owner=request.user)
+        vm = serializer.save(owner=request.user, order=order)
+
+        # FIXME: commit everything (VM + order) at once.
+        order.recurring_price = vm.recurring_price(order.recurring_period)
+        order.one_time_price = 0
+        order.save()
 
         return Response(serializer.data)