Implement more detailed link group withdrawal simulations

2023-05-05 14:39:02 +01:00 · 2023-05-05 14:39:02 +01:00 · 0be1b98072
parent 35607898f0
commit 0be1b98072
7 changed files with 157 additions and 47 deletions
--- a/app/urls.py
+++ b/app/urls.py
@ -272,4 +272,9 @@ urlpatterns = [
        linkgroups.LinkGroupSimulation.as_view(),
        name="linkgroup_simulate",
    ),
+    path(
+        "links/<str:type>/withdraw/<str:pk>/",
+        linkgroups.LinkGroupWithdraw.as_view(),
+        name="linkgroup_withdraw",
+    ),
 ] + static(settings.STATIC_URL, document_root=settings.STATIC_ROOT)
--- a/core/clients/platforms/agora.py
+++ b/core/clients/platforms/agora.py
@ -37,11 +37,12 @@ class AgoraClient(LocalPlatformClient, BaseClient):
        return rtrn

    # TODO: write test before re-enabling adding total_trades
-    async def withdraw_funds(self):
+    async def withdraw_funds(self, checks):
        """
        Withdraw excess funds to our XMR wallets.
        """
        print("CALLING WITHDRAW FUNDS")
+        # checks = self.check_all()
        totals_all = await self.money.get_total()
        if totals_all is False:
            return False
--- a/core/lib/money.py
+++ b/core/lib/money.py
@ -43,20 +43,27 @@ class Money(object):
        )
        self.es = client

-    async def check_all(self, user=None, nordigen=None, agora=None):
+    async def check_all(self, user=None, link_group=None, nordigen=None, agora=None):
        """
        Run all the balance checks that output into ES in another thread.
        """
-        # TODO: pass link group instead
-        if not all([user, nordigen, agora]):
+        if not all([nordigen, agora]):
+            raise Exception
+        if not any([user, link_group]):
            raise Exception

        # I hate circular dependencies
        self.nordigen = nordigen
        self.agora = agora

-        aggregators = Aggregator.objects.filter(user=user, enabled=True)
-        platforms = Platform.objects.filter(user=user, enabled=True)
+        cast = {}
+        if user is not None:
+            cast["user"] = user
+        if link_group is not None:
+            cast["link_group"] = link_group
+
+        aggregators = Aggregator.objects.filter(enabled=True, **cast)
+        platforms = Platform.objects.filter(enabled=True, **cast)

        total = await self.get_total(aggregators, platforms, trades=True)

--- a/core/management/commands/scheduling.py
+++ b/core/management/commands/scheduling.py
@ -5,21 +5,67 @@ from django.core.management.base import BaseCommand

 from core.clients.aggregators.nordigen import NordigenClient
 from core.clients.platforms.agora import AgoraClient
-from core.models import INTERVAL_CHOICES, Aggregator, Platform
+from core.lib.money import Money
+from core.lib.notify import sendmsg
+from core.models import INTERVAL_CHOICES, Aggregator, LinkGroup, Platform, Requisition
 from core.util import logs

 log = logs.get_logger("scheduling")

 INTERVAL_AGGREGATOR = 10
+INTERVAL_WITHDRAWAL = 7200

 INTERVALS_PLATFORM = [x[0] for x in INTERVAL_CHOICES]


+async def withdrawal_job(group=None):
+    money = Money()
+    if group is not None:
+        groups = [group]
+    else:
+        groups = LinkGroup.objects.filter(enabled=True)
+    for group in groups:
+        checks = await money.check_all(
+            user=group.user, nordigen=NordigenClient, agora=AgoraClient
+        )
+        aggregators = Aggregator.objects.filter(
+            user=group.user,
+            link_group=group,
+        )
+        platforms = Platform.objects.filter(
+            user=group.user,
+            link_group=group,
+        )
+        requisitions = Requisition.objects.filter(
+            user=group.user,
+            aggregator__in=aggregators,
+        )
+        pay_list = money.get_pay_list(
+            group,
+            requisitions,
+            platforms,
+            group.user,
+            checks["total_profit"],
+        )
+        collapsed = money.collapse_pay_list(pay_list)
+        if any(collapsed.values()):
+            message = ""
+            print("COLLAPSED", collapsed)
+            for wallet, amount in collapsed.items():
+                print("ITER", wallet, amount)
+                message += f"{wallet}: {amount}\n"
+            print("MESSAGE", message)
+            await sendmsg(
+                group.user,
+                message,
+                title="Your withdrawal is ready!",
+            )
+
+
 async def aggregator_job():
    aggregators = Aggregator.objects.filter(enabled=True)
    for aggregator in aggregators:
        open_trade_currencies = aggregator.trades_currencies
-        log.debug(f"Currencies of open trades: {open_trade_currencies}")
        if aggregator.service == "nordigen":
            instance = None
            if aggregator.fetch_accounts is True:
@ -74,6 +120,10 @@ class Command(BaseCommand):

        log.debug(f"Scheduling {INTERVAL_AGGREGATOR} second aggregator job")
        scheduler.add_job(aggregator_job, "interval", seconds=INTERVAL_AGGREGATOR)
+
+        log.debug(f"Scheduling {INTERVAL_WITHDRAWAL} second withdrawal job")
+        scheduler.add_job(withdrawal_job, "interval", seconds=INTERVAL_WITHDRAWAL)
+
        for interval in INTERVALS_PLATFORM:
            if interval == 0:
                continue
--- a/core/templates/partials/linkgroup-info-sim.html
+++ b/core/templates/partials/linkgroup-info-sim.html
@ -1,43 +1,24 @@
-<h1 class="title">Simulation for $1000</h1>
-{{ object }}
-<p>
-  Assuming equal throughput for platforms and requisitions.
-  <strong>Note that this is just a simulation, equal throughput is highly unlikely.</strong>
-</p>
+<h1 class="title">Simulation with live data</h1>
+<p>Total profit: {{ total_profit }}</p>
 <div class="columns">
  <div class="column">
    <div class="content">
      <ul>
-        {% for key, list in simulation.items %}
-          <li>
-            {{ key.0 }}: ${{ key.1 }}
+        {% for wallet, list in pay_list.items %}
+          <li>{{ wallet.name }} to <code>{{ wallet.address }}</code>: </li>
          <ul>
-              {% for item in list %}
-                <li>${{ item.amount }} to {{ item.name }} at <code>{{ item.address }}</code></li>
+            {% for amount, item in list %}
+              <li>{{ amount }} for {{ item }}</li>
            {% endfor %}
          </ul>
-          </li>
        {% endfor %}
      </ul>
    </div>
    <h1 class="title is-4">Total for wallets</h1>
    <div class="box">
-      {% for wallet, pay_list in pay_list.items %}
-        {{ wallet }}: ${{ pay_list.amount }}
-        <progress class="progress" value="{{ pay_list.amount }}" max="1000"></progress>
-      {% endfor %}
-    </div>
-  </div>
-
-  <div class="column">
-    <div class="box">
-      {% for key, list in simulation.items %}
-        <strong>{{ key.0 }}: ${{ key.1 }}</strong>
-        <progress class="progress" value="{{ key.1 }}" max="1000"></progress>
-        {% for item in list %}
-          <em>{{ item.name }}: ${{ item.amount }}</em><progress class="progress" value="{{ item.amount }}" max="{{ item.max }}"></progress>
-        {% endfor %}
-
+      {% for wallet, amount in collapsed.items %}
+        {{ wallet }}: ${{ amount }}
+        <progress class="progress" value="{{ amount }}" max="{{ total_profit }}"></progress>
      {% endfor %}
    </div>
  </div>
--- a/core/templates/partials/linkgroup-info.html
+++ b/core/templates/partials/linkgroup-info.html
@ -89,4 +89,47 @@
  </div>
 </div>

-{% include 'partials/linkgroup-info-sim.html' %}
+<h1 class="title">Simulation for $1000</h1>
+<p>
+  Assuming equal throughput for platforms and requisitions.
+  <strong>Note that this is just a simulation, equal throughput is highly unlikely.</strong>
+</p>
+<div class="columns">
+  <div class="column">
+    <div class="content">
+      <ul>
+        {% for key, list in simulation.items %}
+          <li>
+            {{ key.0 }}: ${{ key.1 }}
+            <ul>
+              {% for item in list %}
+                <li>${{ item.amount }} to {{ item.name }} at <code>{{ item.address }}</code></li>
+              {% endfor %}
+            </ul>
+          </li>
+        {% endfor %}
+      </ul>
+    </div>
+    <h1 class="title is-4">Total for wallets</h1>
+    <div class="box">
+      {% for wallet, pay_list in pay_list.items %}
+        {{ wallet }}: ${{ pay_list.amount }}
+        <progress class="progress" value="{{ pay_list.amount }}" max="1000"></progress>
+      {% endfor %}
+    </div>
+  </div>
+
+  <div class="column">
+    <div class="box">
+      {% for key, list in simulation.items %}
+        <strong>{{ key.0 }}: ${{ key.1 }}</strong>
+        <progress class="progress" value="{{ key.1 }}" max="1000"></progress>
+        {% for item in list %}
+          <em>{{ item.name }}: ${{ item.amount }}</em><progress class="progress" value="{{ item.amount }}" max="{{ item.max }}"></progress>
+        {% endfor %}
+
+      {% endfor %}
+    </div>
+  </div>
+
+</div>
--- a/core/views/linkgroups.py
+++ b/core/views/linkgroups.py
@ -1,6 +1,8 @@
 from django.contrib.auth.mixins import LoginRequiredMixin
 from django.http import HttpResponse
+from django.shortcuts import render
 from django.urls import reverse
+from django.views import View
 from mixins.views import (
    ObjectCreate,
    ObjectDelete,
@ -15,6 +17,7 @@ from core.clients.aggregators.nordigen import NordigenClient
 from core.clients.platforms.agora import AgoraClient
 from core.forms import LinkGroupForm
 from core.lib.money import Money
+from core.management.commands.scheduling import withdrawal_job
 from core.models import (
    Aggregator,
    LinkGroup,
@ -46,7 +49,9 @@ class LinkGroupInfo(LoginRequiredMixin, OTPRequiredMixin, ObjectRead):

        self.extra_buttons = [
            {
-                "url": "#",
+                "url": reverse(
+                    "linkgroup_withdraw", kwargs={"pk": self.object.id, "type": "modal"}
+                ),
                "action": "withdraw",
                "method": "get",
                "label": "Withdraw profit",
@ -205,19 +210,17 @@ class LinkGroupSimulation(LoginRequiredMixin, OTPRequiredMixin, ObjectRead):
            return HttpResponse(status=status.HTTP_404_NOT_FOUND)
        money = Money()
        checks = synchronize_async_helper(
-            money.check_all(
-                user=self.request.user, nordigen=NordigenClient, agora=AgoraClient
-            )
+            money.check_all(link_group=pk, nordigen=NordigenClient, agora=AgoraClient)
        )
        print("CHECKS", checks)

        aggregators = Aggregator.objects.filter(
            user=self.request.user,
-            link_group=self.object,
+            link_group=linkgroup,
        )
        platforms = Platform.objects.filter(
            user=self.request.user,
-            link_group=self.object,
+            link_group=linkgroup,
        )
        requisitions = Requisition.objects.filter(
            user=self.request.user,
@ -234,4 +237,24 @@ class LinkGroupSimulation(LoginRequiredMixin, OTPRequiredMixin, ObjectRead):
        print("PAY LIST", pay_list)
        collapsed = money.collapse_pay_list(pay_list)
        print("COLLAPSED", collapsed)
-        return collapsed
+        self.extra_context = {"collapsed": collapsed}
+        self.extra_context["pay_list"] = pay_list
+        self.extra_context["total_profit"] = checks["total_profit"]
+        return pay_list
+
+
+class LinkGroupWithdraw(LoginRequiredMixin, OTPRequiredMixin, View):
+    template_name = "mixins/partials/notify.html"
+
+    def get(self, request, **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)
+        run = synchronize_async_helper(withdrawal_job(linkgroup))
+        print("RUN", run)
+        context = {"class": "success", "message": "Withdrawal run"}
+        return render(request, self.template_name, context)