Simplify DB object management with Django CRUD helpers
This commit is contained in:
@@ -1,187 +1,40 @@
|
||||
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 rest_framework.views import APIView
|
||||
|
||||
from core.forms import TradeForm
|
||||
from core.models import Account, Trade
|
||||
from core.models import Trade
|
||||
from core.util import logs
|
||||
from core.views import ObjectCreate, ObjectDelete, ObjectList, ObjectUpdate
|
||||
|
||||
log = logs.get_logger(__name__)
|
||||
|
||||
|
||||
def get_trades(user, account=None):
|
||||
if user:
|
||||
trades = Trade.objects.filter(account__user=user)
|
||||
elif account:
|
||||
trades = Trade.objects.filter(account=account, account__user=user)
|
||||
return trades
|
||||
class TradeList(LoginRequiredMixin, ObjectList):
|
||||
list_template = "partials/trade-list.html"
|
||||
model = Trade
|
||||
context_object_name = "trades"
|
||||
title = "Trades"
|
||||
title_singular = "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."
|
||||
|
||||
submit_url_name = "trade_create"
|
||||
|
||||
|
||||
class Trades(LoginRequiredMixin, View):
|
||||
allowed_types = ["modal", "widget", "window", "page"]
|
||||
window_content = "window-content/trades.html"
|
||||
|
||||
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]
|
||||
if account_id:
|
||||
try:
|
||||
trades = Account.objects.get(id=account_id, user=request.user)
|
||||
except Account.DoesNotExist:
|
||||
message = "Account does not exist."
|
||||
message_class = "danger"
|
||||
context = {
|
||||
"message": message,
|
||||
"class": message_class,
|
||||
"type": type,
|
||||
}
|
||||
return render(request, template_name, context)
|
||||
trades = get_trades(request.user, account_id)
|
||||
else:
|
||||
trades = get_trades(request.user)
|
||||
if type == "page":
|
||||
type = "modal"
|
||||
context = {
|
||||
"title": f"Trades ({type})",
|
||||
"unique": unique,
|
||||
"window_content": self.window_content,
|
||||
"items": trades,
|
||||
"type": type,
|
||||
}
|
||||
return render(request, template_name, context)
|
||||
class TradeCreate(LoginRequiredMixin, ObjectCreate):
|
||||
model = Trade
|
||||
form_class = TradeForm
|
||||
context_object_name = "trades"
|
||||
submit_url_name = "trade_create"
|
||||
|
||||
|
||||
class TradeAction(LoginRequiredMixin, APIView):
|
||||
allowed_types = ["modal", "widget", "window", "page"]
|
||||
window_content = "window-content/add-trade.html"
|
||||
parser_classes = [FormParser]
|
||||
class TradeUpdate(LoginRequiredMixin, ObjectUpdate):
|
||||
model = Trade
|
||||
form_class = TradeForm
|
||||
context_object_name = "trades"
|
||||
submit_url_name = "trade_update"
|
||||
|
||||
def get(self, request, type, trade_id=None):
|
||||
"""
|
||||
Get the form for adding or editing a trade.
|
||||
:param trade_id: The id of the trade to edit. Optional.
|
||||
"""
|
||||
if type not in self.allowed_types:
|
||||
return HttpResponseBadRequest
|
||||
template_name = f"wm/{type}.html"
|
||||
unique = str(uuid.uuid4())[:8]
|
||||
if trade_id:
|
||||
try:
|
||||
trade = Trade.objects.get(id=trade_id, account__user=request.user)
|
||||
form = TradeForm(instance=trade)
|
||||
except Trade.DoesNotExist:
|
||||
message = "Trade does not exist"
|
||||
message_class = "danger"
|
||||
context = {
|
||||
"message": message,
|
||||
"message_class": message_class,
|
||||
"window_content": self.window_content,
|
||||
}
|
||||
return render(request, template_name, context)
|
||||
else:
|
||||
form = TradeForm()
|
||||
if type == "page":
|
||||
type = "modal"
|
||||
context = {
|
||||
"form": form,
|
||||
"trade_id": trade_id,
|
||||
"type": type,
|
||||
"unique": unique,
|
||||
"window_content": self.window_content,
|
||||
}
|
||||
|
||||
return render(request, template_name, context)
|
||||
|
||||
def put(self, request, type, trade_id=None):
|
||||
"""
|
||||
Add or edit a trade.
|
||||
:param trade_id: The id of the trade to edit. Optional.
|
||||
"""
|
||||
if type not in self.allowed_types:
|
||||
return HttpResponseBadRequest
|
||||
message = None
|
||||
message_class = "success"
|
||||
|
||||
if trade_id:
|
||||
try:
|
||||
form = TradeForm(
|
||||
request.data,
|
||||
instance=Trade.objects.get(id=trade_id, account__user=request.user),
|
||||
)
|
||||
except Trade.DoesNotExist:
|
||||
message = "Trade does not exist"
|
||||
message_class = "danger"
|
||||
context = {
|
||||
"message": message,
|
||||
"class": message_class,
|
||||
}
|
||||
return render(request, self.template_name, context)
|
||||
else:
|
||||
form = TradeForm(request.data)
|
||||
if form.is_valid():
|
||||
trade = form.save(commit=False)
|
||||
print("PRESAVE TRADE", trade)
|
||||
trade.user = request.user
|
||||
trade.save()
|
||||
success, returned = trade.post()
|
||||
if success:
|
||||
print("SAVED TRADE", trade)
|
||||
if trade_id:
|
||||
message = f"Trade {trade_id} edited successfully"
|
||||
else:
|
||||
message = f"Trade {trade.id} added successfully"
|
||||
else:
|
||||
message = f"Error adding trade: {returned}"
|
||||
message_class = "danger"
|
||||
else:
|
||||
message = "Error adding trade"
|
||||
message_class = "danger"
|
||||
|
||||
trades = get_trades(request.user)
|
||||
|
||||
context = {
|
||||
"items": trades,
|
||||
"type": type,
|
||||
}
|
||||
if message:
|
||||
context["message"] = message
|
||||
context["class"] = message_class
|
||||
template_name = "partials/trade-list.html"
|
||||
return render(request, template_name, context)
|
||||
|
||||
def delete(self, request, type, trade_id):
|
||||
"""
|
||||
Delete a trade.
|
||||
:param trade_id: The id of the trade to delete.
|
||||
"""
|
||||
if type not in self.allowed_types:
|
||||
return HttpResponseBadRequest
|
||||
message = None
|
||||
message_class = "success"
|
||||
try:
|
||||
trade = Trade.objects.get(id=trade_id, account__user=request.user)
|
||||
trade.delete()
|
||||
message = "Trade deleted successfully"
|
||||
except Trade.DoesNotExist:
|
||||
message = "Error deleting trade"
|
||||
message_class = "danger"
|
||||
|
||||
trades = get_trades(request.user)
|
||||
|
||||
context = {
|
||||
"items": trades,
|
||||
"type": type,
|
||||
}
|
||||
if message:
|
||||
context["message"] = message
|
||||
context["class"] = message_class
|
||||
|
||||
template_name = "partials/trade-list.html"
|
||||
return render(request, template_name, context)
|
||||
class TradeDelete(LoginRequiredMixin, ObjectDelete):
|
||||
model = Trade
|
||||
|
||||
Reference in New Issue
Block a user