import uuid from django.contrib.auth.mixins import LoginRequiredMixin from django.http import HttpResponseBadRequest from django.shortcuts import render from django.views import View from rest_framework.parsers import FormParser from core.lib import trades from core.models import Account from core.util import logs log = logs.get_logger(__name__) def get_positions(user, account_id=None): items = [] accounts = Account.objects.filter(user=user) for account in accounts: positions = account.client.get_all_positions() print("POSITIONS", positions) for item in positions: item = dict(item) item["account_id"] = account.id item["unrealized_pl"] = float(item["unrealized_pl"]) items.append(item) # try: # parsed = ccxt_s.CCXTRoot.from_dict(order) # except ValidationError as e: # log.error(f"Error creating trade: {e}") # return False # self.status = parsed.status # self.response = order return items class Positions(LoginRequiredMixin, View): allowed_types = ["modal", "widget", "window", "page"] window_content = "window-content/objects.html" list_template = "partials/position-list.html" page_title = "Live positions from all exchanges" page_subtitle = "Manual trades are editable under 'Bot Trades' tab." async def get(self, request, type, account_id=None): if type not in self.allowed_types: return HttpResponseBadRequest template_name = f"wm/{type}.html" unique = str(uuid.uuid4())[:8] items = get_positions(request.user, account_id) if type == "page": type = "modal" context = { "title": f"Hooks ({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, } return render(request, template_name, context) class PositionAction(LoginRequiredMixin, View): allowed_types = ["modal", "widget", "window", "page"] window_content = "window-content/view-position.html" parser_classes = [FormParser] async def get(self, request, type, account_id, asset_id): """ Get live information for a trade. """ if type not in self.allowed_types: return HttpResponseBadRequest template_name = f"wm/{type}.html" unique = str(uuid.uuid4())[:8] account = Account.get_by_id(account_id, request.user) success, info = trades.get_position_info(account, asset_id) print("INFO", info) if not success: message = "Position does not exist" message_class = "danger" items = get_positions(request.user, account_id) if type == "page": type = "modal" context = { "title": f"Hooks ({type})", "unique": unique, "window_content": self.window_content, "items": items, "type": type, } if success: context["items"] = info else: context["message"] = message context["class"] = message_class return render(request, template_name, context)