From 485f08e25cfa36a904b7018ba567d26b4b8300b9 Mon Sep 17 00:00:00 2001
From: Nico Schottelius <nico@nico-notebook.schottelius.org>
Date: Tue, 25 May 2021 19:55:33 +0200
Subject: [PATCH] Cleanup views

---
 uncloud/urls.py  | 29 -----------------------------
 uncloud/views.py |  9 +++++++++
 2 files changed, 9 insertions(+), 29 deletions(-)

diff --git a/uncloud/urls.py b/uncloud/urls.py
index bf3672c..6ecfd7a 100644
--- a/uncloud/urls.py
+++ b/uncloud/urls.py
@@ -25,35 +25,6 @@ router = routers.DefaultRouter()
 # Beta endpoints
 router.register(r'beta/vm', vmviews.NicoVMProductViewSet, basename='nicovmproduct')
 
-# VM
-# router.register(r'v1/vm/snapshot', vmviews.VMSnapshotProductViewSet, basename='vmsnapshotproduct')
-# router.register(r'v1/vm/diskimage', vmviews.VMDiskImageProductViewSet, basename='vmdiskimageproduct')
-# router.register(r'v1/vm/disk', vmviews.VMDiskProductViewSet, basename='vmdiskproduct')
-# router.register(r'v1/vm/vm', vmviews.VMProductViewSet, basename='vmproduct')
-
-# Services
-# router.register(r'v1/service/matrix', serviceviews.MatrixServiceProductViewSet, basename='matrixserviceproduct')
-# router.register(r'v1/service/generic', serviceviews.GenericServiceProductViewSet, basename='genericserviceproduct')
-
-
-# Pay
-
-# router.register(r'v1/my/bill', payviews.BillViewSet, basename='bill')
-# router.register(r'v1/my/order', payviews.OrderViewSet, basename='order')
-# router.register(r'v1/my/payment-method', payviews.PaymentMethodViewSet, basename='payment-method')
-
-# admin/staff urls
-# router.register(r'v1/admin/bill', payviews.AdminBillViewSet, basename='admin/bill')
-# router.register(r'v1/admin/order', payviews.AdminOrderViewSet, basename='admin/order')
-# router.register(r'v1/admin/vmhost', vmviews.VMHostViewSet)
-# router.register(r'v1/admin/vmcluster', vmviews.VMClusterViewSet)
-
-# User/Account
-# router.register(r'v1/my/user', authviews.UserViewSet, basename='user')
-# router.register(r'v1/admin/user', authviews.AdminUserViewSet, basename='useradmin')
-# router.register(r'v1/user/register', authviews.AccountManagementViewSet, basename='user/register')
-
-
 ################################################################################
 # v2
 
diff --git a/uncloud/views.py b/uncloud/views.py
index 198abd0..37542bb 100644
--- a/uncloud/views.py
+++ b/uncloud/views.py
@@ -1,4 +1,13 @@
 from django.views.generic.base import TemplateView
 
+from uncloud_pay.selectors import get_balance_for_user
+
 class UncloudIndex(TemplateView):
     template_name = "uncloud/index.html"
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        print(context)
+        if self.request.user.is_authenticated:
+            context['balance'] = get_balance_for_user(self.request.user)
+        return context