Add initial generate-bills and charge-negative-balance uncloud-pay

commands

uncloud/uncloud_vm/views.py.orig

@@ -0,0 +1,84 @@
+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
+
+<<<<<<< HEAD
+from .models import VMHost, VMProduct, VMSnapshotProduct
+from uncloud_pay.models import Order
+
+=======
+from uncloud_pay.models import Order, RecurringPeriod
+
+from .models import VMHost, VMProduct
+>>>>>>> Quickly wire vm creation to orders
+from .serializers import VMHostSerializer, VMProductSerializer, VMSnapshotProductSerializer
+
+import datetime
+
+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):
+        # 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)
+        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)
+
+
+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 = VMSnapshotProductSerializer(data=request.data, context={'request': request})
+        serializer.is_valid(raise_exception=True)
+
+        # Create order
+        now = datetime.datetime.now()
+        order = Order(owner=request.user,
+                      creation_date=now,
+                      starting_date=now,
+                      recurring_price=20,
+                      one_time_price=0,
+                      recurring_period="per_month")
+        order.save()
+
+        # FIXME: calculate the gb_* values
+        serializer.save(owner=request.user,
+                        order=order,
+                        gb_ssd=12,
+                        gb_hdd=20)
+
+        return Response(serializer.data)