From 1f43a00c7a826f8770715f383e9f2dd534a108a2 Mon Sep 17 00:00:00 2001
From: Mark Veidemanis <m@zm.is>
Date: Fri, 10 Feb 2023 23:47:55 +0000
Subject: [PATCH] Add all models to admin site and tweak some documentation

---
 core/admin.py        | 20 ++++++++++++++++++++
 core/views/assets.py |  4 +++-
 2 files changed, 23 insertions(+), 1 deletion(-)

diff --git a/core/admin.py b/core/admin.py
index 55bcde4..1165a4d 100644
--- a/core/admin.py
+++ b/core/admin.py
@@ -13,6 +13,10 @@ from .models import (
     Trade,
     TradingTime,
     User,
+    NotificationSettings,
+    RiskModel,
+    AssetGroup,
+    AssetRestriction,
 )
 
 # admin.site.__class__ = OTPAdminSite
@@ -73,6 +77,18 @@ class StrategyAdmin(admin.ModelAdmin):
     list_display = ("user", "name", "description", "account", "enabled")
 
 
+class NotificationSettingsAdmin(admin.ModelAdmin):
+    list_display = ("user", "ntfy_topic", "ntfy_url")
+
+class RiskModelAdmin(admin.ModelAdmin):
+    list_display = ("user", "name", "description", "max_loss_percent", "max_risk_percent", "max_open_trades", "max_open_trades_per_symbol")
+
+class AssetGroupAdmin(admin.ModelAdmin):
+    list_display = ("user", "name", "description", "account")
+
+class AssetRestrictionAdmin(admin.ModelAdmin):
+    list_display = ("user", "name", "description", "webhook_id", "group")
+
 admin.site.register(User, CustomUserAdmin)
 admin.site.register(Plan)
 admin.site.register(Session)
@@ -84,3 +100,7 @@ admin.site.register(Trade, TradeAdmin)
 admin.site.register(Callback, CallbackAdmin)
 admin.site.register(TradingTime, TradingTimeAdmin)
 admin.site.register(Strategy, StrategyAdmin)
+admin.site.register(NotificationSettings, NotificationSettingsAdmin)
+admin.site.register(RiskModel, RiskModelAdmin)
+admin.site.register(AssetGroup, AssetGroupAdmin)
+admin.site.register(AssetRestriction, AssetRestrictionAdmin)
\ No newline at end of file
diff --git a/core/views/assets.py b/core/views/assets.py
index 0a21f47..391b2ec 100644
--- a/core/views/assets.py
+++ b/core/views/assets.py
@@ -18,7 +18,8 @@ log = logs.get_logger(__name__)
 class AssetGroupList(LoginRequiredMixin, ObjectList):
     list_template = "partials/assetgroup-list.html"
     model = AssetGroup
-    page_title = "List of asset groups for restrictions. Linked to accounts."
+    page_title = "List of asset groups for restrictions. Linked to strategies."
+    page_subtitle = "Asset groups are collections of asset restrictions."
 
     list_url_name = "assetgroups"
     list_url_args = ["type"]
@@ -64,6 +65,7 @@ class AssetRestrictionList(
     list_template = "partials/assetrestriction-list.html"
     model = AssetRestriction
     page_title = "List of asset restrictions. Linked to asset groups."
+    page_subtitle = "Allows API calls to permit or prohibit trading on defined currency pairs."
 
     list_url_name = "assetrestrictions"
     list_url_args = ["type", "group"]