Finish implementing risk models

core/forms.py

@@ -7,6 +7,7 @@ from .models import (
     Account,
     Hook,
     NotificationSettings,
+    RiskModel,
     Signal,
     Strategy,
     Trade,
@@ -110,19 +111,14 @@ class SignalForm(RestrictedFormMixin, ModelForm):
 class AccountForm(RestrictedFormMixin, ModelForm):
     class Meta:
         model = Account
-        fields = (
-            "name",
-            "exchange",
-            "api_key",
-            "api_secret",
-            "sandbox",
-        )
+        fields = ("name", "exchange", "api_key", "api_secret", "sandbox", "risk_model")
         help_texts = {
             "name": "Name of the account. Informational only.",
             "exchange": "The exchange to use for this account.",
             "api_key": "The API key or username for the account.",
             "api_secret": "The API secret or password/token for the account.",
             "sandbox": "Whether to use the sandbox/demo or not.",
+            "risk_model": "The risk model to use for this account.",
         }
 
 
@@ -298,3 +294,24 @@ class NotificationSettingsForm(RestrictedFormMixin, ModelForm):
             "ntfy_topic": "The topic to send notifications to.",
             "ntfy_url": "Custom NTFY server. Leave blank to use the default server.",
         }
+
+
+class RiskModelForm(RestrictedFormMixin, ModelForm):
+    class Meta:
+        model = RiskModel
+        fields = (
+            "name",
+            "description",
+            "max_loss_percent",
+            "max_risk_percent",
+            "max_open_trades",
+            "max_open_trades_per_symbol",
+        )
+        help_texts = {
+            "name": "Name of the risk model. Informational only.",
+            "description": "Description of the risk model. Informational only.",
+            "max_loss_percent": "The maximum percent of the account balance that can be lost before we cease trading.",
+            "max_risk_percent": "The maximum percent of the account balance that can be risked on all open trades.",
+            "max_open_trades": "The maximum number of open trades.",
+            "max_open_trades_per_symbol": "The maximum number of open trades per symbol.",
+        }