Simplify DB object management with Django CRUD helpers
This commit is contained in:
@@ -0,0 +1,203 @@
|
||||
import uuid
|
||||
|
||||
from django.http import Http404, HttpResponseBadRequest
|
||||
from django.urls import reverse
|
||||
from django.views.generic.detail import DetailView
|
||||
from django.views.generic.edit import CreateView, DeleteView, UpdateView
|
||||
from django.views.generic.list import ListView
|
||||
from rest_framework.parsers import FormParser
|
||||
|
||||
from core.util import logs
|
||||
|
||||
log = logs.get_logger(__name__)
|
||||
|
||||
|
||||
class ObjectList(ListView):
|
||||
allowed_types = ["modal", "widget", "window", "page"]
|
||||
window_content = "window-content/objects.html"
|
||||
list_template = None
|
||||
|
||||
model = None
|
||||
context_object_name = "objects"
|
||||
page_title = None
|
||||
page_subtitle = None
|
||||
|
||||
title = "Objects"
|
||||
title_singular = "Object"
|
||||
|
||||
submit_url_name = None
|
||||
|
||||
# copied from BaseListView
|
||||
def get(self, request, *args, **kwargs):
|
||||
self.object_list = self.get_queryset()
|
||||
allow_empty = self.get_allow_empty()
|
||||
|
||||
type = kwargs.get("type", None)
|
||||
if not type:
|
||||
return HttpResponseBadRequest("No type specified")
|
||||
if type not in self.allowed_types:
|
||||
return HttpResponseBadRequest("Invalid type specified")
|
||||
self.template_name = f"wm/{type}.html"
|
||||
unique = str(uuid.uuid4())[:8]
|
||||
if type == "page":
|
||||
type = "modal"
|
||||
|
||||
if not allow_empty:
|
||||
# When pagination is enabled and object_list is a queryset,
|
||||
# it's better to do a cheap query than to load the unpaginated
|
||||
# queryset in memory.
|
||||
if self.get_paginate_by(self.object_list) is not None and hasattr(
|
||||
self.object_list, "exists"
|
||||
):
|
||||
is_empty = not self.object_list.exists()
|
||||
else:
|
||||
is_empty = not self.object_list
|
||||
if is_empty:
|
||||
raise Http404("Empty list")
|
||||
submit_url = reverse(self.submit_url_name, kwargs={"type": type})
|
||||
context = self.get_context_data()
|
||||
context["title"] = self.title + f" ({type})"
|
||||
context["title_singular"] = self.title_singular
|
||||
context["unique"] = unique
|
||||
context["window_content"] = self.window_content
|
||||
context["list_template"] = self.list_template
|
||||
context["page_title"] = self.page_title
|
||||
context["page_subtitle"] = self.page_subtitle
|
||||
context["type"] = type
|
||||
context["submit_url"] = submit_url
|
||||
return self.render_to_response(context)
|
||||
|
||||
|
||||
class ObjectCreate(CreateView):
|
||||
allowed_types = ["modal", "widget", "window", "page"]
|
||||
window_content = "window-content/object-form.html"
|
||||
parser_classes = [FormParser]
|
||||
|
||||
model = None
|
||||
context_object_name = "objects"
|
||||
submit_url_name = None
|
||||
|
||||
request = None
|
||||
|
||||
def form_valid(self, form):
|
||||
obj = form.save(commit=False)
|
||||
if self.request is None:
|
||||
raise Exception("Request is None")
|
||||
obj.user = self.request.user
|
||||
obj.save()
|
||||
form.save_m2m()
|
||||
context = {"message": "Object created", "class": "success"}
|
||||
return self.render_to_response(context)
|
||||
|
||||
def get(self, request, *args, **kwargs):
|
||||
self.request = request
|
||||
type = kwargs.get("type", None)
|
||||
if not type:
|
||||
return HttpResponseBadRequest("No type specified")
|
||||
if type not in self.allowed_types:
|
||||
return HttpResponseBadRequest("Invalid type specified")
|
||||
self.template_name = f"wm/{type}.html"
|
||||
unique = str(uuid.uuid4())[:8]
|
||||
if type == "page":
|
||||
type = "modal"
|
||||
|
||||
self.object = None
|
||||
submit_url = reverse(self.submit_url_name, kwargs={"type": type})
|
||||
context = self.get_context_data()
|
||||
context["unique"] = unique
|
||||
context["window_content"] = self.window_content
|
||||
context["context_object_name"] = self.context_object_name
|
||||
context["submit_url"] = submit_url
|
||||
context["type"] = type
|
||||
return self.render_to_response(context)
|
||||
|
||||
def post(self, request, *args, **kwargs):
|
||||
self.request = request
|
||||
self.template_name = "partials/notify.html"
|
||||
return super().post(request, *args, **kwargs)
|
||||
|
||||
|
||||
class ObjectRead(DetailView):
|
||||
allowed_types = ["modal", "widget", "window", "page"]
|
||||
window_content = "window-content/object.html"
|
||||
|
||||
model = None
|
||||
context_object_name = "object"
|
||||
|
||||
|
||||
class ObjectUpdate(UpdateView):
|
||||
allowed_types = ["modal", "widget", "window", "page"]
|
||||
window_content = "window-content/object-form.html"
|
||||
parser_classes = [FormParser]
|
||||
|
||||
model = None
|
||||
context_object_name = "objects"
|
||||
submit_url_name = None
|
||||
|
||||
request = None
|
||||
|
||||
def form_valid(self, form):
|
||||
obj = form.save(commit=False)
|
||||
if self.request is None:
|
||||
raise Exception("Request is None")
|
||||
obj.save()
|
||||
form.save_m2m()
|
||||
context = {"message": "Object updated", "class": "success"}
|
||||
return self.render_to_response(context)
|
||||
|
||||
def get(self, request, *args, **kwargs):
|
||||
self.request = request
|
||||
type = kwargs.get("type", None)
|
||||
pk = kwargs.get("pk", None)
|
||||
if not type:
|
||||
return HttpResponseBadRequest("No type specified")
|
||||
if not pk:
|
||||
return HttpResponseBadRequest("No pk specified")
|
||||
if type not in self.allowed_types:
|
||||
return HttpResponseBadRequest("Invalid type specified")
|
||||
self.template_name = f"wm/{type}.html"
|
||||
unique = str(uuid.uuid4())[:8]
|
||||
if type == "page":
|
||||
type = "modal"
|
||||
|
||||
self.object = self.get_object()
|
||||
submit_url = reverse(self.submit_url_name, kwargs={"type": type, "pk": pk})
|
||||
context = self.get_context_data()
|
||||
context["unique"] = unique
|
||||
context["window_content"] = self.window_content
|
||||
context["context_object_name"] = self.context_object_name
|
||||
context["submit_url"] = submit_url
|
||||
context["type"] = type
|
||||
return self.render_to_response(context)
|
||||
|
||||
def post(self, request, *args, **kwargs):
|
||||
self.request = request
|
||||
self.template_name = "partials/notify.html"
|
||||
return super().post(request, *args, **kwargs)
|
||||
|
||||
|
||||
class ObjectDelete(DeleteView):
|
||||
model = None
|
||||
template_name = "partials/notify.html"
|
||||
|
||||
# Overriden to prevent success URL from being used
|
||||
def delete(self, request, *args, **kwargs):
|
||||
"""
|
||||
Call the delete() method on the fetched object and then redirect to the
|
||||
success URL.
|
||||
"""
|
||||
self.object = self.get_object()
|
||||
# success_url = self.get_success_url()
|
||||
self.object.delete()
|
||||
context = {"message": "Object deleted", "class": "success"}
|
||||
return self.render_to_response(context)
|
||||
|
||||
# This will be used in newer Django versions, until then we get a warning
|
||||
def form_valid(self, form):
|
||||
"""
|
||||
Call the delete() method on the fetched object.
|
||||
"""
|
||||
self.object = self.get_object()
|
||||
self.object.delete()
|
||||
context = {"message": "Object deleted", "class": "success"}
|
||||
return self.render_to_response(context)
|
||||
|
||||
@@ -4,27 +4,21 @@ 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 AccountForm
|
||||
from core.models import Account
|
||||
from core.util import logs
|
||||
from core.views import ObjectCreate, ObjectDelete, ObjectList, ObjectUpdate
|
||||
|
||||
log = logs.get_logger(__name__)
|
||||
|
||||
|
||||
def get_accounts(user):
|
||||
accounts = Account.objects.filter(user=user)
|
||||
return accounts
|
||||
|
||||
|
||||
class AccountInfo(LoginRequiredMixin, View):
|
||||
VIEWABLE_FIELDS_MODEL = ["name", "exchange", "api_key", "sandbox"]
|
||||
allowed_types = ["modal", "widget", "window", "page"]
|
||||
window_content = "window-content/account-info.html"
|
||||
|
||||
def get(self, request, type, account_id):
|
||||
def get(self, request, type, pk):
|
||||
"""
|
||||
Get the account details.
|
||||
:param account_id: The id of the account.
|
||||
@@ -34,7 +28,7 @@ class AccountInfo(LoginRequiredMixin, View):
|
||||
template_name = f"wm/{type}.html"
|
||||
unique = str(uuid.uuid4())[:8]
|
||||
try:
|
||||
account = Account.get_by_id(account_id, request.user)
|
||||
account = Account.get_by_id(pk, request.user)
|
||||
except Account.DoesNotExist:
|
||||
message = "Account does not exist"
|
||||
message_class = "danger"
|
||||
@@ -56,7 +50,7 @@ class AccountInfo(LoginRequiredMixin, View):
|
||||
context = {
|
||||
"db_info": account_info,
|
||||
"live_info": live_info,
|
||||
"account_id": account_id,
|
||||
"pk": pk,
|
||||
"type": type,
|
||||
"unique": unique,
|
||||
"window_content": self.window_content,
|
||||
@@ -65,145 +59,43 @@ class AccountInfo(LoginRequiredMixin, View):
|
||||
return render(request, template_name, context)
|
||||
|
||||
|
||||
class Accounts(LoginRequiredMixin, View):
|
||||
allowed_types = ["modal", "widget", "window", "page"]
|
||||
window_content = "window-content/accounts.html"
|
||||
class AccountList(LoginRequiredMixin, ObjectList):
|
||||
list_template = "partials/account-list.html"
|
||||
model = Account
|
||||
context_object_name = "accounts"
|
||||
title = "Accounts"
|
||||
title_singular = "Account"
|
||||
page_title = "List of accounts"
|
||||
|
||||
async def get(self, request, type):
|
||||
if type not in self.allowed_types:
|
||||
return HttpResponseBadRequest
|
||||
template_name = f"wm/{type}.html"
|
||||
unique = str(uuid.uuid4())[:8]
|
||||
accounts = get_accounts(request.user)
|
||||
if type == "page":
|
||||
type = "modal"
|
||||
context = {
|
||||
"title": f"Accounts ({type})",
|
||||
"unique": unique,
|
||||
"window_content": self.window_content,
|
||||
"items": accounts,
|
||||
"type": type,
|
||||
}
|
||||
return render(request, template_name, context)
|
||||
submit_url_name = "account_create"
|
||||
|
||||
|
||||
class AccountAction(LoginRequiredMixin, APIView):
|
||||
allowed_types = ["modal", "widget", "window", "page"]
|
||||
window_content = "window-content/add-account.html"
|
||||
parser_classes = [FormParser]
|
||||
class AccountCreate(LoginRequiredMixin, ObjectCreate):
|
||||
model = Account
|
||||
form_class = AccountForm
|
||||
context_object_name = "accounts"
|
||||
submit_url_name = "account_create"
|
||||
|
||||
def get(self, request, type, account_id=None):
|
||||
"""
|
||||
Get the form for adding or editing a account.
|
||||
:param account_id: The id of the account to edit. Optional.
|
||||
"""
|
||||
if type not in self.allowed_types:
|
||||
return HttpResponseBadRequest
|
||||
template_name = f"wm/{type}.html"
|
||||
unique = str(uuid.uuid4())[:8]
|
||||
if account_id:
|
||||
try:
|
||||
account = Account.objects.get(id=account_id, user=request.user)
|
||||
form = AccountForm(instance=account)
|
||||
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)
|
||||
else:
|
||||
form = AccountForm()
|
||||
if type == "page":
|
||||
type = "modal"
|
||||
context = {
|
||||
"form": form,
|
||||
"account_id": account_id,
|
||||
"type": type,
|
||||
"unique": unique,
|
||||
"window_content": self.window_content,
|
||||
}
|
||||
|
||||
return render(request, template_name, context)
|
||||
# class AccountRead(LoginRequiredMixin, ObjectRead):
|
||||
# model = Account
|
||||
# context_object_name = "accounts"
|
||||
# submit_url_name = "account_read"
|
||||
# fields = (
|
||||
# "name",
|
||||
# "exchange",
|
||||
# "api_key",
|
||||
# "api_secret",
|
||||
# "sandbox",
|
||||
# )
|
||||
|
||||
def put(self, request, type, account_id=None):
|
||||
"""
|
||||
Add or edit a account.
|
||||
:param account_id: The id of the account to edit. Optional.
|
||||
"""
|
||||
if type not in self.allowed_types:
|
||||
return HttpResponseBadRequest
|
||||
message = None
|
||||
message_class = "success"
|
||||
|
||||
if account_id:
|
||||
try:
|
||||
form = AccountForm(
|
||||
request.data,
|
||||
instance=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,
|
||||
}
|
||||
return render(request, self.template_name, context)
|
||||
else:
|
||||
form = AccountForm(request.data)
|
||||
if form.is_valid():
|
||||
account = form.save(commit=False)
|
||||
account.user = request.user
|
||||
account.save()
|
||||
if account_id:
|
||||
message = f"Account {account_id} edited successfully"
|
||||
else:
|
||||
message = f"Account {account.id} added successfully"
|
||||
else:
|
||||
message = "Error adding account"
|
||||
message_class = "danger"
|
||||
class AccountUpdate(LoginRequiredMixin, ObjectUpdate):
|
||||
model = Account
|
||||
form_class = AccountForm
|
||||
context_object_name = "accounts"
|
||||
submit_url_name = "account_update"
|
||||
|
||||
accounts = get_accounts(request.user)
|
||||
|
||||
context = {
|
||||
"items": accounts,
|
||||
"type": type,
|
||||
}
|
||||
if message:
|
||||
context["message"] = message
|
||||
context["class"] = message_class
|
||||
template_name = "partials/account-list.html"
|
||||
return render(request, template_name, context)
|
||||
|
||||
def delete(self, request, type, account_id):
|
||||
"""
|
||||
Delete a account.
|
||||
:param account_id: The id of the account to delete.
|
||||
"""
|
||||
if type not in self.allowed_types:
|
||||
return HttpResponseBadRequest
|
||||
message = None
|
||||
message_class = "success"
|
||||
try:
|
||||
account = Account.objects.get(id=account_id, user=request.user)
|
||||
account.delete()
|
||||
message = "Account deleted successfully"
|
||||
except Account.DoesNotExist:
|
||||
message = "Error deleting account"
|
||||
message_class = "danger"
|
||||
|
||||
accounts = get_accounts(request.user)
|
||||
|
||||
context = {
|
||||
"items": accounts,
|
||||
"type": type,
|
||||
}
|
||||
if message:
|
||||
context["message"] = message
|
||||
context["class"] = message_class
|
||||
|
||||
template_name = "partials/account-list.html"
|
||||
return render(request, template_name, context)
|
||||
class AccountDelete(LoginRequiredMixin, ObjectDelete):
|
||||
model = Account
|
||||
|
||||
@@ -18,17 +18,19 @@ def get_callbacks(user, hook=None):
|
||||
|
||||
class Callbacks(LoginRequiredMixin, View):
|
||||
allowed_types = ["modal", "widget", "window", "page"]
|
||||
window_content = "window-content/callbacks.html"
|
||||
window_content = "window-content/objects.html"
|
||||
list_template = "partials/callback-list.html"
|
||||
page_title = "List of received callbacks"
|
||||
|
||||
async def get(self, request, type, hook_id=None):
|
||||
async def get(self, request, type, pk=None):
|
||||
if type not in self.allowed_types:
|
||||
return HttpResponseBadRequest
|
||||
template_name = f"wm/{type}.html"
|
||||
unique = str(uuid.uuid4())[:8]
|
||||
|
||||
if hook_id:
|
||||
if pk:
|
||||
try:
|
||||
hook = Hook.objects.get(id=hook_id, user=request.user)
|
||||
hook = Hook.objects.get(id=pk, user=request.user)
|
||||
except Hook.DoesNotExist:
|
||||
message = "Hook does not exist."
|
||||
message_class = "danger"
|
||||
@@ -48,7 +50,9 @@ class Callbacks(LoginRequiredMixin, View):
|
||||
"title": f"Callbacks ({type})",
|
||||
"unique": unique,
|
||||
"window_content": self.window_content,
|
||||
"items": callbacks,
|
||||
"list_template": self.list_template,
|
||||
"object_list": callbacks,
|
||||
"type": type,
|
||||
"page_title": self.page_title,
|
||||
}
|
||||
return render(request, template_name, context)
|
||||
|
||||
@@ -1,13 +1,10 @@
|
||||
import re
|
||||
import uuid
|
||||
|
||||
import orjson
|
||||
from django.conf import settings
|
||||
from django.contrib.auth.mixins import LoginRequiredMixin
|
||||
from django.http import HttpResponse, HttpResponseBadRequest
|
||||
from django.shortcuts import render
|
||||
from django.views import View
|
||||
from rest_framework.parsers import FormParser, JSONParser
|
||||
from rest_framework.parsers import JSONParser
|
||||
from rest_framework.views import APIView
|
||||
from serde import ValidationError
|
||||
|
||||
@@ -16,15 +13,11 @@ from core.lib import market
|
||||
from core.lib.serde import drakdoo_s
|
||||
from core.models import Callback, Hook
|
||||
from core.util import logs
|
||||
from core.views import ObjectCreate, ObjectDelete, ObjectList, ObjectUpdate
|
||||
|
||||
log = logs.get_logger(__name__)
|
||||
|
||||
|
||||
def get_hooks(user):
|
||||
hooks = Hook.objects.filter(user=user)
|
||||
return hooks
|
||||
|
||||
|
||||
def extract_price(message):
|
||||
result = re.findall("\d+\.\d+", message) # noqa
|
||||
if len(result) != 1:
|
||||
@@ -72,8 +65,8 @@ class HookAPI(APIView):
|
||||
"message": hook_resp.message,
|
||||
"period": hook_resp.period,
|
||||
"sent": hook_resp.timestamp.sent,
|
||||
"timestamp_trade": hook_resp.timestamp.trade,
|
||||
"trade": hook_resp.market.exchange,
|
||||
"trade": hook_resp.timestamp.trade,
|
||||
"exchange": hook_resp.market.exchange,
|
||||
"base": hook_resp.market.item,
|
||||
"quote": hook_resp.market.currency,
|
||||
"symbol": symbol,
|
||||
@@ -106,142 +99,32 @@ class HookAPI(APIView):
|
||||
return HttpResponse(orjson.dumps(return_data), content_type="application/json")
|
||||
|
||||
|
||||
class Hooks(LoginRequiredMixin, View):
|
||||
allowed_types = ["modal", "widget", "window", "page"]
|
||||
window_content = "window-content/hooks.html"
|
||||
class HookList(LoginRequiredMixin, ObjectList):
|
||||
# window_content = "window-content/hooks.html"
|
||||
list_template = "partials/hook-list.html"
|
||||
model = Hook
|
||||
context_object_name = "hooks"
|
||||
title = "Hooks"
|
||||
title_singular = "Hook"
|
||||
page_title = "List of active URL endpoints for receiving hooks."
|
||||
page_subtitle = "Add URLs here to receive Drakdoo callbacks. Make then unique!"
|
||||
|
||||
async def get(self, request, type):
|
||||
if type not in self.allowed_types:
|
||||
return HttpResponseBadRequest
|
||||
template_name = f"wm/{type}.html"
|
||||
unique = str(uuid.uuid4())[:8]
|
||||
hooks = get_hooks(request.user)
|
||||
if type == "page":
|
||||
type = "modal"
|
||||
context = {
|
||||
"title": f"Hooks ({type})",
|
||||
"unique": unique,
|
||||
"window_content": self.window_content,
|
||||
"items": hooks,
|
||||
"type": type,
|
||||
}
|
||||
return render(request, template_name, context)
|
||||
submit_url_name = "hook_create"
|
||||
|
||||
|
||||
class HookAction(LoginRequiredMixin, APIView):
|
||||
allowed_types = ["modal", "widget", "window", "page"]
|
||||
window_content = "window-content/add-hook.html"
|
||||
parser_classes = [FormParser]
|
||||
class HookCreate(LoginRequiredMixin, ObjectCreate):
|
||||
model = Hook
|
||||
form_class = HookForm
|
||||
context_object_name = "hooks"
|
||||
submit_url_name = "hook_create"
|
||||
|
||||
def get(self, request, type, hook_id=None):
|
||||
"""
|
||||
Get the form for adding or editing a hook.
|
||||
:param hook_id: The id of the hook to edit. Optional.
|
||||
"""
|
||||
if type not in self.allowed_types:
|
||||
return HttpResponseBadRequest
|
||||
template_name = f"wm/{type}.html"
|
||||
unique = str(uuid.uuid4())[:8]
|
||||
if hook_id:
|
||||
try:
|
||||
hook = Hook.objects.get(id=hook_id, user=request.user)
|
||||
form = HookForm(instance=hook)
|
||||
except Hook.DoesNotExist:
|
||||
message = "Hook 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 = HookForm()
|
||||
if type == "page":
|
||||
type = "modal"
|
||||
context = {
|
||||
"form": form,
|
||||
"hook_id": hook_id,
|
||||
"type": type,
|
||||
"unique": unique,
|
||||
"window_content": self.window_content,
|
||||
}
|
||||
|
||||
return render(request, template_name, context)
|
||||
class HookUpdate(LoginRequiredMixin, ObjectUpdate):
|
||||
model = Hook
|
||||
form_class = HookForm
|
||||
context_object_name = "hooks"
|
||||
submit_url_name = "hook_update"
|
||||
|
||||
def put(self, request, type, hook_id=None):
|
||||
"""
|
||||
Add or edit a hook.
|
||||
:param hook_id: The id of the hook to edit. Optional.
|
||||
"""
|
||||
if type not in self.allowed_types:
|
||||
return HttpResponseBadRequest
|
||||
message = None
|
||||
message_class = "success"
|
||||
|
||||
if hook_id:
|
||||
try:
|
||||
form = HookForm(request.data, instance=Hook.objects.get(id=hook_id))
|
||||
except Hook.DoesNotExist:
|
||||
message = "Hook does not exist"
|
||||
message_class = "danger"
|
||||
context = {
|
||||
"message": message,
|
||||
"class": message_class,
|
||||
}
|
||||
return render(request, self.template_name, context)
|
||||
else:
|
||||
form = HookForm(request.data)
|
||||
if form.is_valid():
|
||||
hook = form.save(commit=False)
|
||||
hook.user = request.user
|
||||
hook.save()
|
||||
if hook_id:
|
||||
message = f"Hook {hook_id} edited successfully"
|
||||
else:
|
||||
message = f"Hook {hook.id} added successfully"
|
||||
else:
|
||||
message = "Error adding hook"
|
||||
message_class = "danger"
|
||||
|
||||
hooks = get_hooks(request.user)
|
||||
|
||||
context = {
|
||||
"items": hooks,
|
||||
"type": type,
|
||||
}
|
||||
if message:
|
||||
context["message"] = message
|
||||
context["class"] = message_class
|
||||
template_name = "partials/hook-list.html"
|
||||
return render(request, template_name, context)
|
||||
|
||||
def delete(self, request, type, hook_id):
|
||||
"""
|
||||
Delete a hook.
|
||||
:param hook_id: The id of the hook to delete.
|
||||
"""
|
||||
if type not in self.allowed_types:
|
||||
return HttpResponseBadRequest
|
||||
message = None
|
||||
message_class = "success"
|
||||
try:
|
||||
hook = Hook.objects.get(id=hook_id, user=request.user)
|
||||
hook.delete()
|
||||
message = "Hook deleted successfully"
|
||||
except Hook.DoesNotExist:
|
||||
message = "Error deleting hook"
|
||||
message_class = "danger"
|
||||
|
||||
hooks = get_hooks(request.user)
|
||||
|
||||
context = {
|
||||
"items": hooks,
|
||||
"type": type,
|
||||
}
|
||||
if message:
|
||||
context["message"] = message
|
||||
context["class"] = message_class
|
||||
|
||||
template_name = "partials/hook-list.html"
|
||||
return render(request, template_name, context)
|
||||
class HookDelete(LoginRequiredMixin, ObjectDelete):
|
||||
model = Hook
|
||||
|
||||
@@ -37,7 +37,10 @@ def get_positions(user, account_id=None):
|
||||
|
||||
class Positions(LoginRequiredMixin, View):
|
||||
allowed_types = ["modal", "widget", "window", "page"]
|
||||
window_content = "window-content/positions.html"
|
||||
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:
|
||||
@@ -51,8 +54,11 @@ class Positions(LoginRequiredMixin, View):
|
||||
"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)
|
||||
|
||||
|
||||
@@ -1,15 +1,9 @@
|
||||
import uuid
|
||||
|
||||
from django.contrib.auth.mixins import LoginRequiredMixin
|
||||
from django.http import HttpResponseBadRequest
|
||||
from django.shortcuts import render
|
||||
from django.views import View # , CreateView, UpdateView, DeleteView
|
||||
from rest_framework.parsers import FormParser
|
||||
from rest_framework.views import APIView
|
||||
|
||||
from core.forms import StrategyForm
|
||||
from core.models import Strategy
|
||||
from core.util import logs
|
||||
from core.views import ObjectCreate, ObjectDelete, ObjectList, ObjectUpdate
|
||||
|
||||
# from django.urls import reverse
|
||||
|
||||
@@ -17,172 +11,30 @@ from core.util import logs
|
||||
log = logs.get_logger(__name__)
|
||||
|
||||
|
||||
def get_strategies(user):
|
||||
strategies = Strategy.objects.filter(user=user)
|
||||
return strategies
|
||||
class StrategyList(LoginRequiredMixin, ObjectList):
|
||||
list_template = "partials/strategy-list.html"
|
||||
model = Strategy
|
||||
context_object_name = "strategies"
|
||||
title = "Strategies"
|
||||
title_singular = "Strategy"
|
||||
page_title = "List of strategies"
|
||||
|
||||
submit_url_name = "strategy_create"
|
||||
|
||||
|
||||
class Strategies(LoginRequiredMixin, View):
|
||||
allowed_types = ["modal", "widget", "window", "page"]
|
||||
window_content = "window-content/strategies.html"
|
||||
|
||||
async def get(self, request, type):
|
||||
if type not in self.allowed_types:
|
||||
return HttpResponseBadRequest
|
||||
template_name = f"wm/{type}.html"
|
||||
unique = str(uuid.uuid4())[:8]
|
||||
strategies = get_strategies(request.user)
|
||||
if type == "page":
|
||||
type = "modal"
|
||||
context = {
|
||||
"title": f"Accounts ({type})",
|
||||
"unique": unique,
|
||||
"window_content": self.window_content,
|
||||
"items": strategies,
|
||||
"type": type,
|
||||
}
|
||||
return render(request, template_name, context)
|
||||
class StrategyCreate(LoginRequiredMixin, ObjectCreate):
|
||||
model = Strategy
|
||||
form_class = StrategyForm
|
||||
context_object_name = "strategies"
|
||||
submit_url_name = "strategy_create"
|
||||
|
||||
|
||||
# class AddStrategy(CreateView):
|
||||
# model = Strategy
|
||||
# form_class = StrategyForm
|
||||
# template_name = "window-content/add-strategy.html"
|
||||
# success_url = reverse("strategies")
|
||||
|
||||
# def form_valid(self, form):
|
||||
# form.instance.user = self.request.user
|
||||
# return super().form_valid(form)
|
||||
|
||||
# def get_context_data(self, **kwargs):
|
||||
# context = super().get_context_data(**kwargs)
|
||||
# context["title"] = "Add Strategy"
|
||||
# context["window_content"] = "window-content/add_strategy.html"
|
||||
# return context
|
||||
class StrategyUpdate(LoginRequiredMixin, ObjectUpdate):
|
||||
model = Strategy
|
||||
form_class = StrategyForm
|
||||
context_object_name = "strategies"
|
||||
submit_url_name = "strategy_update"
|
||||
|
||||
|
||||
class StrategiesAction(LoginRequiredMixin, APIView):
|
||||
allowed_types = ["modal", "widget", "window", "page"]
|
||||
window_content = "window-content/add-strategy.html"
|
||||
parser_classes = [FormParser]
|
||||
|
||||
def get(self, request, type, strategy_id=None):
|
||||
"""
|
||||
Get the form for adding or editing a strategy.
|
||||
:param strategy_id: The id of the strategy to edit. Optional.
|
||||
"""
|
||||
if type not in self.allowed_types:
|
||||
return HttpResponseBadRequest
|
||||
template_name = f"wm/{type}.html"
|
||||
unique = str(uuid.uuid4())[:8]
|
||||
if strategy_id:
|
||||
try:
|
||||
account = Strategy.objects.get(id=strategy_id)
|
||||
form = StrategyForm(instance=account)
|
||||
except Strategy.DoesNotExist:
|
||||
message = "Strategy 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 = StrategyForm()
|
||||
if type == "page":
|
||||
type = "modal"
|
||||
context = {
|
||||
"form": form,
|
||||
"strategy_id": strategy_id,
|
||||
"type": type,
|
||||
"unique": unique,
|
||||
"window_content": self.window_content,
|
||||
}
|
||||
|
||||
return render(request, template_name, context)
|
||||
|
||||
def put(self, request, type, strategy_id=None):
|
||||
"""
|
||||
Add or edit a account.
|
||||
:param account_id: The id of the strategy to edit. Optional.
|
||||
"""
|
||||
if type not in self.allowed_types:
|
||||
return HttpResponseBadRequest
|
||||
message = None
|
||||
message_class = "success"
|
||||
|
||||
if strategy_id:
|
||||
try:
|
||||
form = StrategyForm(
|
||||
request.data,
|
||||
instance=Strategy.objects.get(
|
||||
id=strategy_id, account__user=request.user
|
||||
),
|
||||
)
|
||||
except Strategy.DoesNotExist:
|
||||
message = "Strategy does not exist"
|
||||
message_class = "danger"
|
||||
context = {
|
||||
"message": message,
|
||||
"class": message_class,
|
||||
}
|
||||
return render(request, self.template_name, context)
|
||||
else:
|
||||
form = StrategyForm(request.data)
|
||||
if form.is_valid():
|
||||
hooks = list(form.cleaned_data.get("hooks"))
|
||||
strategy = form.save(commit=False)
|
||||
strategy.user = request.user
|
||||
strategy.hooks.set(hooks)
|
||||
strategy.save()
|
||||
print("HOOKS SET", strategy.hooks)
|
||||
if strategy_id:
|
||||
message = f"Strategy {strategy_id} edited successfully"
|
||||
else:
|
||||
message = f"Strategy {strategy.id} added successfully"
|
||||
else:
|
||||
message = "Error adding strategy"
|
||||
message_class = "danger"
|
||||
|
||||
accounts = get_strategies(request.user)
|
||||
|
||||
context = {
|
||||
"items": accounts,
|
||||
"type": type,
|
||||
}
|
||||
if message:
|
||||
context["message"] = message
|
||||
context["class"] = message_class
|
||||
template_name = "partials/strategy-list.html"
|
||||
return render(request, template_name, context)
|
||||
|
||||
def delete(self, request, type, strategy_id):
|
||||
"""
|
||||
Delete a strategy.
|
||||
:param strategy_id: The id of the strategy to delete.
|
||||
"""
|
||||
if type not in self.allowed_types:
|
||||
return HttpResponseBadRequest
|
||||
message = None
|
||||
message_class = "success"
|
||||
try:
|
||||
strategy = Strategy.objects.get(id=strategy_id, user=request.user)
|
||||
strategy.delete()
|
||||
message = "Strategy deleted successfully"
|
||||
except Strategy.DoesNotExist:
|
||||
message = "Error deleting strategy"
|
||||
message_class = "danger"
|
||||
|
||||
strategies = get_strategies(request.user)
|
||||
|
||||
context = {
|
||||
"items": strategies,
|
||||
"type": strategies,
|
||||
}
|
||||
if message:
|
||||
context["message"] = message
|
||||
context["class"] = message_class
|
||||
|
||||
template_name = "partials/strategy-list.html"
|
||||
return render(request, template_name, context)
|
||||
class StrategyDelete(LoginRequiredMixin, ObjectDelete):
|
||||
model = Strategy
|
||||
|
||||
@@ -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