fisk/core/views/accounts.py

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 core.forms import AccountForm
from core.models import Account
from core.util import logs
from core.views import ObjectCreate, ObjectDelete, ObjectList, ObjectUpdate

log = logs.get_logger(__name__)


class AccountInfo(LoginRequiredMixin, View):
    VIEWABLE_FIELDS_MODEL = [
        "name",
        "exchange",
        "api_key",
        "sandbox",
        "supported_symbols",
    ]
    allowed_types = ["modal", "widget", "window", "page"]
    window_content = "window-content/account-info.html"

    def get(self, request, type, pk):
        """
        Get the account details.
        :param account_id: The id of the account.
        """
        if type not in self.allowed_types:
            return HttpResponseBadRequest
        template_name = f"wm/{type}.html"
        unique = str(uuid.uuid4())[:8]
        try:
            account = Account.get_by_id(pk, request.user)
        except Account.DoesNotExist:
            message = "Account does not exist"
            message_class = "danger"
            context = {
                "message": message,
                "message_class": message_class,
                "window_content": self.window_content,
            }
            return render(request, template_name, context)

        live_info = account.client.get_account()
        live_info = live_info
        account_info = account.__dict__
        account_info = {
            k: v for k, v in account_info.items() if k in self.VIEWABLE_FIELDS_MODEL
        }
        account_info["supported_symbols"] = ", ".join(account_info["supported_symbols"])

        if type == "page":
            type = "modal"
        context = {
            "db_info": account_info,
            "live_info": live_info,
            "pk": pk,
            "type": type,
            "unique": unique,
            "window_content": self.window_content,
        }

        return render(request, template_name, context)


class AccountList(LoginRequiredMixin, ObjectList):
    list_template = "partials/account-list.html"
    model = Account
    page_title = "List of accounts"

    list_url_name = "accounts"
    list_url_args = ["type"]

    submit_url_name = "account_create"


class AccountCreate(LoginRequiredMixin, ObjectCreate):
    model = Account
    form_class = AccountForm

    list_url_name = "accounts"
    list_url_args = ["type"]

    submit_url_name = "account_create"


# class AccountRead(LoginRequiredMixin, ObjectRead):
#     model = Account
#     context_object_name = "accounts"
#     submit_url_name = "account_read"
#     fields = (
#             "name",
#             "exchange",
#             "api_key",
#             "api_secret",
#             "sandbox",
#         )


class AccountUpdate(LoginRequiredMixin, ObjectUpdate):
    model = Account
    form_class = AccountForm

    list_url_name = "accounts"
    list_url_args = ["type"]

    submit_url_name = "account_update"


class AccountDelete(LoginRequiredMixin, ObjectDelete):
    model = Account

    list_url_name = "accounts"
    list_url_args = ["type"]