fisk/core/views/profit.py

from decimal import Decimal as D

from django.contrib.auth.mixins import LoginRequiredMixin
from mixins.views import ObjectList
from two_factor.views.mixins import OTPRequiredMixin

from core.exchanges import GenericAPIError
from core.models import Account
from core.util import logs

log = logs.get_logger(__name__)


class Profit(LoginRequiredMixin, OTPRequiredMixin, ObjectList):
    list_template = "partials/profit-list.html"
    page_title = "Profit by account"
    page_subtitle = None

    context_object_name_singular = "profit"
    context_object_name = "profit"

    list_url_name = "profit"
    list_url_args = ["type"]

    def get_queryset(self, **kwargs):
        items = []
        # Only get enabled accounts
        accounts = Account.objects.filter(user=self.request.user, enabled=True)
        for account in accounts:
            try:
                details = account.client.get_account()
                item = {
                    "account": account,
                    "pl": D(account.initial_balance) - D(details["balance"]),
                    "unrealizedPL": D(details["unrealizedPL"]),
                    "balance": details["balance"],
                    "currency": details["currency"],
                }
                items.append(item)
            except GenericAPIError:
                continue
        return items