Implement profit view and fix auto refresh

app/urls.py

@@ -28,6 +28,7 @@ from core.views import (
     hooks,
     limits,
     positions,
+    profit,
     signals,
     strategies,
     trades,
@@ -137,6 +138,7 @@ urlpatterns = [
         trades.TradeDeleteAll.as_view(),
         name="trade_delete_all",
     ),
+    path("profit/<str:type>/", profit.Profit.as_view(), name="profit"),
     path("positions/<str:type>/", positions.Positions.as_view(), name="positions"),
     path(
         "positions/<str:type>/<str:account_id>/",

core/templates/base.html

@@ -202,6 +202,18 @@
             Home
           </a>
           {% if user.is_authenticated %}
+
+            <div class="navbar-item has-dropdown is-hoverable">
+              <a class="navbar-link">
+                View
+              </a>
+              <div class="navbar-dropdown">
+                <a class="navbar-item" href="{% url 'profit' type='page' %}">
+                  Profit by account
+                </a>
+              </div>
+            </div>
+
             <div class="navbar-item has-dropdown is-hoverable">
               <a class="navbar-link">
                 Manage
@@ -215,8 +227,6 @@
                 </a>
               </div>
             </div>
-          {% endif %}
-          {% if user.is_authenticated %}
             <div class="navbar-item has-dropdown is-hoverable">
               <a class="navbar-link">
                 Setup

core/templates/index.html

@@ -4,7 +4,7 @@
 {% block outer_content %}
 
   <div class="grid-stack" id="grid-stack-main">
-    <div class="grid-stack-item" gs-w="5" gs-h="14" gs-y="0" gs-x="1">
+    <!-- <div class="grid-stack-item" gs-w="5" gs-h="14" gs-y="0" gs-x="1">
       <div class="grid-stack-item-content">
         <nav class="panel">
           <p class="panel-heading" style="padding: .2em; line-height: .5em;">
@@ -16,7 +16,7 @@
           </article>
         </nav>
       </div>
-    </div>
+    </div> -->
     <div class="grid-stack-item" gs-w="4" gs-h="25" gs-y="0" gs-x="6">
       <div class="grid-stack-item-content">
         <nav class="panel">
@@ -31,22 +31,6 @@
       </div>
     </div>
 
-    <!-- <div class="grid-stack-item" gs-w="5" gs-h="14" gs-y="14" gs-x="1">
-      <div class="grid-stack-item-content">
-        <nav class="panel">
-          <p class="panel-heading" style="padding: .2em; line-height: .5em;">
-            <i class="fa-solid fa-arrows-up-down-left-right has-text-grey-light"></i>
-            Offset
-          </p>
-          <article class="panel-block is-active">
-            {# include 'window-content/offset.html' #}
-          </article>
-        </nav>
-      </div>
-    </div> -->
-  </div>
-
-
     <script>
       var grid = GridStack.init({
         cellHeight: 20,
@@ -109,8 +93,10 @@
         }
       });
     </script>
-  <script>
+    <div
-
+      hx-headers='{"X-CSRFToken": "{{ csrf_token }}"}'
-  </script>
+      hx-get="{% url 'profit' type='widget' %}"
+      hx-target="#widgets-here"
+      hx-trigger="load"></div>
 
 {% endblock %}

core/templates/partials/position-list.html

@@ -4,7 +4,7 @@
   hx-target="#{{ context_object_name }}-table"
   id="{{ context_object_name }}-table"
   hx-swap="outerHTML"
-  hx-trigger="{{ context_object_name_singular }}Event from:body"
+  hx-trigger="{{ context_object_name_singular }}Event from:body, every 5s"
   hx-get="{{ list_url }}">
   <thead>
     <th>account</th>

core/templates/partials/profit-list.html

@@ -0,0 +1,30 @@
+{% include 'partials/notify.html' %}
+
+<table
+  class="table is-fullwidth is-hoverable"
+  hx-target="#{{ context_object_name }}-table"
+  id="{{ context_object_name }}-table"
+  hx-swap="outerHTML"
+  hx-trigger="{{ context_object_name_singular }}Event from:body, every 3s"
+  hx-get="{{ list_url }}">
+  <thead>
+    <th>id</th>
+    <th>name</th>
+    <th>P/L</th>
+    <th>balance</th>
+    <th>currency</th>
+  </thead>
+  {% for item in object_list %}
+  <tr class="
+  {% if item.pl > 0 %}has-background-success-light
+  {% elif item.pl < 0 %}has-background-danger-light
+  {% endif %}">
+      <td>{{ item.account.id }}</td>
+      <td>{{ item.account.name }}</td>
+      <td>{{ item.pl }}</td>
+      <td>{{ item.balance }}</td>
+      <td>{{ item.currency }}</td>
+    </tr>
+  {% endfor %}
+
+</table>

core/templates/wm/widget.html

@@ -1,5 +1,5 @@
 <div id="widget">
-  <div id="widget-{{ unique }}" class="grid-stack-item" {% block widget_options %}gs-w="10" gs-h="1" gs-y="10" gs-x="1"{% endblock %}>
+  <div id="widget-{{ unique }}" class="grid-stack-item" {% block widget_options %}gs-w="5" gs-h="1" gs-y="0" gs-x="1"{% endblock %}>
     <div class="grid-stack-item-content">
 
       <nav class="panel">

core/views/__init__.py

@@ -153,7 +153,9 @@ class ObjectList(RestrictedViewMixin, ObjectNameMixin, ListView):
 
         # Return partials for HTMX
         if self.request.htmx:
-            if orig_type == "page":
+            if request.headers["HX-Target"] == self.context_object_name + "-table":
+                self.template_name = self.list_template
+            elif orig_type == "page":
                 self.template_name = self.list_template
             else:
                 context["window_content"] = self.list_template

core/views/accounts.py

@@ -22,7 +22,7 @@ class AccountInfo(LoginRequiredMixin, OTPRequiredMixin, View):
         "api_key",
         "sandbox",
         "supported_symbols",
-        "instruments",
+        # "instruments",
     ]
     allowed_types = ["modal", "widget", "window", "page"]
     window_content = "window-content/account-info.html"

core/views/positions.py

@@ -61,6 +61,8 @@ class Positions(LoginRequiredMixin, OTPRequiredMixin, View):
     list_template = "partials/position-list.html"
     page_title = "Live positions from all exchanges"
     page_subtitle = "Manual trades are editable under 'Bot Trades' tab."
+    context_object_name_singular = "position"
+    context_object_name = "positions"
 
     def get(self, request, type, account_id=None):
         if type not in self.allowed_types:
@@ -89,12 +91,14 @@ class Positions(LoginRequiredMixin, OTPRequiredMixin, View):
             "page_title": self.page_title,
             "page_subtitle": self.page_subtitle,
             "list_url": list_url,
-            "context_object_name_singular": "position",
+            "context_object_name_singular": self.context_object_name_singular,
-            "context_object_name": "positions",
+            "context_object_name": self.context_object_name,
         }
         # Return partials for HTMX
         if self.request.htmx:
-            if orig_type == "page":
+            if request.headers["HX-Target"] == self.context_object_name + "-table":
+                self.template_name = self.list_template
+            elif orig_type == "page":
                 self.template_name = self.list_template
             else:
                 context["window_content"] = self.list_template

core/views/profit.py

@@ -0,0 +1,74 @@
+import uuid
+
+from django.contrib.auth.mixins import LoginRequiredMixin
+from django.http import HttpResponseBadRequest
+from django.shortcuts import render
+from django.urls import reverse
+from django.views import View
+from rest_framework.parsers import FormParser
+from two_factor.views.mixins import OTPRequiredMixin
+
+from core.exchanges import GenericAPIError
+from core.models import Account, Trade
+from core.util import logs
+
+log = logs.get_logger(__name__)
+
+def get_profit(user):
+    items = []
+    accounts = Account.objects.filter(user=user)
+    for account in accounts:
+        try:
+            details = account.client.get_account()
+            pl = details["pl"]
+            item = {"account": account, "pl": float(pl), "balance": details["balance"], "currency": details["currency"]}
+            items.append(item)
+        except GenericAPIError:
+            continue
+    return items
+
+class Profit(LoginRequiredMixin, OTPRequiredMixin, View):
+    allowed_types = ["modal", "widget", "window", "page"]
+    window_content = "window-content/objects.html"
+    list_template = "partials/profit-list.html"
+    page_title = "Profit by account"
+    page_subtitle = None
+    context_object_name_singular = "profit"
+    context_object_name = "profit"
+
+    def get(self, request, type):
+        if type not in self.allowed_types:
+            return HttpResponseBadRequest
+        self.template_name = f"wm/{type}.html"
+        unique = str(uuid.uuid4())[:8]
+        items = get_profit(request.user)
+
+        orig_type = type
+        if type == "page":
+            type = "modal"
+        cast = {
+            "type": orig_type,
+        }
+        list_url = reverse("profit", kwargs={**cast})
+        context = {
+            "title": f"Profit ({type})",
+            "unique": unique,
+            "window_content": self.window_content,
+            "list_template": self.list_template,
+            "object_list": items,
+            "type": type,
+            "page_title": self.page_title,
+            "page_subtitle": self.page_subtitle,
+            "list_url": list_url,
+            "context_object_name_singular": self.context_object_name_singular,
+            "context_object_name": self.context_object_name,
+        }
+        # Return partials for HTMX
+        if self.request.htmx:
+            if request.headers["HX-Target"] == self.context_object_name+"-table":
+                self.template_name = self.list_template
+            elif orig_type == "page":
+                self.template_name = self.list_template
+            else:
+                context["window_content"] = self.list_template
+        return render(request, self.template_name, context)