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 two_factor.views.mixins import OTPRequiredMixin from core.exchanges import GenericAPIError from core.forms import TradeForm from core.models import Trade from core.util import logs from core.views import ( ObjectCreate, ObjectDelete, ObjectList, ObjectNameMixin, ObjectUpdate, ) log = logs.get_logger(__name__) class TradeAction(LoginRequiredMixin, OTPRequiredMixin, View): allowed_types = ["modal", "widget", "window", "page"] window_content = "window-content/trade.html" def get(self, request, type, trade_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] db_info = Trade.get_by_id_or_order(trade_id, request.user) if db_info is None: return HttpResponseBadRequest("Trade not found.") if db_info.order_id is not None: try: if db_info.order_id == db_info.response["id"]: db_info.order_id = str(int(db_info.order_id) + 1) db_info.save() live_info = db_info.account.client.get_trade(db_info.order_id) except GenericAPIError as e: live_info = {"error": e} else: live_info = {} if type == "page": type = "modal" context = { "title": f"Trade info ({type})", "unique": unique, "window_content": self.window_content, "type": type, "db_info": db_info.__dict__, "live_info": live_info, } return render(request, template_name, context) class TradeList(LoginRequiredMixin, OTPRequiredMixin, ObjectList): list_template = "partials/trade-list.html" model = Trade page_title = ( "List of bot and manual trades. This may not reflect actual live trades." ) page_subtitle = "Trades deleted here will not be closed on the exchange." list_url_name = "trades" list_url_args = ["type"] submit_url_name = "trade_create" delete_all_url_name = "trade_delete_all" class TradeCreate(LoginRequiredMixin, OTPRequiredMixin, ObjectCreate): model = Trade form_class = TradeForm list_url_name = "trades" list_url_args = ["type"] submit_url_name = "trade_create" def post_save(self, obj): obj.post() log.debug(f"Posting trade {obj}") class TradeUpdate(LoginRequiredMixin, OTPRequiredMixin, ObjectUpdate): model = Trade form_class = TradeForm list_url_name = "trades" list_url_args = ["type"] submit_url_name = "trade_update" class TradeDelete(LoginRequiredMixin, OTPRequiredMixin, ObjectDelete): model = Trade list_url_name = "trades" list_url_args = ["type"] class TradeDeleteAll(LoginRequiredMixin, OTPRequiredMixin, ObjectNameMixin, View): template_name = "partials/notify.html" model = Trade def delete(self, request): """ Delete all trades by the current user """ Trade.objects.filter(user=request.user).delete() context = {"message": "All trades deleted", "class": "success"} response = render(request, self.template_name, context) response["HX-Trigger"] = f"{self.context_object_name_singular}Event" return response