Use CRUD helper for position list

core/views/positions.py

@@ -3,7 +3,6 @@ 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
@@ -11,24 +10,11 @@ from two_factor.views.mixins import OTPRequiredMixin
 from core.exchanges import GenericAPIError
 from core.models import Account, Trade
 from core.util import logs
+from core.views import ObjectList
 
 log = logs.get_logger(__name__)
 
 
-def get_positions(user, account_id=None):
-    items = []
-    accounts = Account.objects.filter(user=user)
-    for account in accounts:
-        try:
-            positions = account.client.get_all_positions()
-        except GenericAPIError:
-            continue
-
-        for item in positions:
-            items.append(item)
-    return items
-
-
 def annotate_positions(positions, user, return_order_ids=False):
     """
     Annotate positions with trade information.
@@ -55,7 +41,7 @@ def annotate_positions(positions, user, return_order_ids=False):
             pass
 
 
-class Positions(LoginRequiredMixin, OTPRequiredMixin, View):
+class Positions(LoginRequiredMixin, OTPRequiredMixin, ObjectList):
     allowed_types = ["modal", "widget", "window", "page"]
     window_content = "window-content/objects.html"
     list_template = "partials/position-list.html"
@@ -63,47 +49,65 @@ class Positions(LoginRequiredMixin, OTPRequiredMixin, View):
     page_subtitle = "Manual trades are editable under 'Bot Trades' tab."
     context_object_name_singular = "position"
     context_object_name = "positions"
+    widget_options = 'gs-w="12" gs-h="1" gs-y="0" gs-x="0"'
 
-    def get(self, request, type, account_id=None):
-        if type not in self.allowed_types:
-            return HttpResponseBadRequest
-        self.template_name = f"wm/{type}.html"
-        unique = str(uuid.uuid4())[:8]
-        items = get_positions(request.user, account_id)
-        annotate_positions(items, request.user, return_order_ids=False)
-
-        orig_type = type
-        if type == "page":
-            type = "modal"
-        cast = {
-            "type": orig_type,
-        }
+    def get_queryset(self, **kwargs):
+        account_id = kwargs.get("account_id", None)
         if account_id:
-            cast["account_id"] = account_id
-        list_url = reverse("positions", kwargs={**cast})
-        context = {
-            "title": f"Positions ({type})",
-            "unique": unique,
-            "window_content": self.window_content,
-            "list_template": self.list_template,
-            "items": 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,
-            "widget_options": 'gs-w="12" gs-h="1" gs-y="0" gs-x="0"',
-        }
-        # 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)
+            self.extra_context["account_id"] = account_id
+        items = []
+        accounts = Account.objects.filter(user=self.request.user)
+        for account in accounts:
+            try:
+                positions = account.client.get_all_positions()
+            except GenericAPIError:
+                continue
+
+            for item in positions:
+                items.append(item)
+        annotate_positions(items, self.request.user, return_order_ids=False)
+        return items
+
+    # def get(self, request, type, account_id=None):
+    #     if type not in self.allowed_types:
+    #         return HttpResponseBadRequest
+    #     self.template_name = f"wm/{type}.html"
+    #     unique = str(uuid.uuid4())[:8]
+    #     items = get_positions(request.user, account_id)
+    #     annotate_positions(items, request.user, return_order_ids=False)
+
+    #     orig_type = type
+    #     if type == "page":
+    #         type = "modal"
+    #     cast = {
+    #         "type": orig_type,
+    #     }
+    #     if account_id:
+    #         cast["account_id"] = account_id
+    #     list_url = reverse("positions", kwargs={**cast})
+    #     context = {
+    #         "title": f"Positions ({type})",
+    #         "unique": unique,
+    #         "window_content": self.window_content,
+    #         "list_template": self.list_template,
+    #         "items": 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,
+    #         "widget_options": 'gs-w="12" gs-h="1" gs-y="0" gs-x="0"',
+    #     }
+    #     # 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)
 
 
 class PositionAction(LoginRequiredMixin, OTPRequiredMixin, View):