103 lines
3.3 KiB
Python
103 lines
3.3 KiB
Python
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)
|