Implement link group detail screen with profit simulation

core/views/aggregators.py

@@ -14,7 +14,7 @@ from two_factor.views.mixins import OTPRequiredMixin
 
 from core.clients.aggregators.nordigen import NordigenClient
 from core.forms import AggregatorForm
-from core.models import Aggregator
+from core.models import Aggregator, Requisition
 from core.util import logs
 from core.views.helpers import synchronize_async_helper
 
@@ -95,6 +95,16 @@ class ReqsList(LoginRequiredMixin, OTPRequiredMixin, ObjectList):
 
         run = synchronize_async_helper(NordigenClient(aggregator))
         reqs = synchronize_async_helper(run.get_requisitions())
+        for req in reqs:
+            # Add in Requisition object
+            requisition_id = req["id"]
+            requisition = Requisition.objects.filter(
+                user=self.request.user,
+                aggregator=aggregator,
+                requisition_id=requisition_id,
+            ).first()
+            if requisition:
+                req["requisition"] = requisition
         return reqs
 
 

core/views/linkgroups.py

@@ -1,9 +1,107 @@
 from django.contrib.auth.mixins import LoginRequiredMixin
-from mixins.views import ObjectCreate, ObjectDelete, ObjectList, ObjectUpdate
+from django.http import HttpResponse
+from mixins.views import (
+    ObjectCreate,
+    ObjectDelete,
+    ObjectList,
+    ObjectRead,
+    ObjectUpdate,
+)
+from rest_framework import status
 from two_factor.views.mixins import OTPRequiredMixin
 
 from core.forms import LinkGroupForm
-from core.models import LinkGroup
+from core.models import Aggregator, LinkGroup, Platform, Requisition
+
+
+class LinkGroupInfo(LoginRequiredMixin, OTPRequiredMixin, ObjectRead):
+    context_object_name_singular = "linkgroup"
+    context_object_name = "linkgroups"
+    detail_template = "partials/linkgroup-info.html"
+
+    def get_object(self, **kwargs):
+        pk = self.kwargs.get("pk")
+        linkgroup = LinkGroup.objects.filter(
+            user=self.request.user,
+            id=pk,
+        ).first()
+        if not linkgroup:
+            return HttpResponse(status=status.HTTP_404_NOT_FOUND)
+        return linkgroup
+
+    def get_context_data(self):
+        context = super().get_context_data()
+
+        aggregators = Aggregator.objects.filter(
+            user=self.request.user,
+            link_group=self.object,
+        )
+        platforms = Platform.objects.filter(
+            user=self.request.user,
+            link_group=self.object,
+        )
+        requisitions = Requisition.objects.filter(
+            user=self.request.user,
+            aggregator__in=aggregators,
+        )
+        context["aggregators"] = aggregators
+        context["platforms"] = platforms
+        context["requisitions"] = requisitions
+        context["linkgroup"] = self.object
+
+        payees = self.object.payees()
+
+        simulation = {}
+        profit = 1000
+        profit_platform = profit * (self.object.platform_owner_cut_percentage / 100)
+        profit_requisition = profit * (
+            self.object.requisition_owner_cut_percentage / 100
+        )
+        profit_operator = profit * (self.object.operator_cut_percentage / 100)
+
+        pay_list = {}
+
+        platform_pay_list = []
+        for payee in payees["platform"]:
+            cast = {
+                "name": payee.name,
+                "address": payee.address,
+                "amount": profit_platform / len(payees["platform"]),
+                "max": profit_platform,
+            }
+            if payee not in pay_list:
+                pay_list[payee] = {}
+            if "amount" in pay_list[payee]:
+                pay_list[payee]["amount"] += cast["amount"]
+            else:
+                pay_list[payee] = dict(cast)
+            platform_pay_list.append(cast)
+
+        requisition_pay_list = []
+        for payee in payees["requisition"]:
+            cast = {
+                "name": payee.name,
+                "address": payee.address,
+                "amount": profit_requisition / len(payees["requisition"]),
+                "max": profit_requisition,
+            }
+            if payee not in pay_list:
+                pay_list[payee] = {}
+            if "amount" in pay_list[payee]:
+                pay_list[payee]["amount"] += cast["amount"]
+            else:
+                pay_list[payee] = dict(cast)
+            requisition_pay_list.append(cast)
+
+        simulation[("Platform", profit_platform)] = platform_pay_list
+        simulation[("Requisition", profit_requisition)] = requisition_pay_list
+        simulation[("Operator", profit_operator)] = []
+
+        context["pay_list"] = pay_list
+
+        context["simulation"] = simulation
+
+        return context
 
 
 class LinkGroupList(LoginRequiredMixin, OTPRequiredMixin, ObjectList):