From c082e5f6a6f5271bf26ba3ed4bc0ec94ed8d2c0b Mon Sep 17 00:00:00 2001
From: Matthias <xmatthias@outlook.com>
Date: Tue, 3 Dec 2024 07:26:47 +0100
Subject: [PATCH] feat: add dry_run_wallet helper

---
 freqtrade/util/__init__.py       |  2 ++
 freqtrade/util/dry_run_wallet.py | 12 ++++++++++++
 2 files changed, 14 insertions(+)
 create mode 100644 freqtrade/util/dry_run_wallet.py

diff --git a/freqtrade/util/__init__.py b/freqtrade/util/__init__.py
index 7a4d4d119..a0b618b11 100644
--- a/freqtrade/util/__init__.py
+++ b/freqtrade/util/__init__.py
@@ -11,6 +11,7 @@ from freqtrade.util.datetime_helpers import (
     format_ms_time,
     shorten_date,
 )
+from freqtrade.util.dry_run_wallet import get_dry_run_wallet
 from freqtrade.util.formatters import decimals_per_coin, fmt_coin, fmt_coin2, round_value
 from freqtrade.util.ft_precise import FtPrecise
 from freqtrade.util.measure_time import MeasureTime
@@ -35,6 +36,7 @@ __all__ = [
     "dt_utc",
     "format_date",
     "format_ms_time",
+    "get_dry_run_wallet",
     "FtPrecise",
     "PeriodicCache",
     "shorten_date",
diff --git a/freqtrade/util/dry_run_wallet.py b/freqtrade/util/dry_run_wallet.py
new file mode 100644
index 000000000..f994f0d4c
--- /dev/null
+++ b/freqtrade/util/dry_run_wallet.py
@@ -0,0 +1,12 @@
+from pytest import Config
+
+
+def get_dry_run_wallet(config: Config) -> int | float:
+    """
+    Return dry-run wallet balance in stake currency from configuration.
+    This setup also supports dictionary mode for dry-run-wallet.
+    """
+    if isinstance(_start_cap := config["dry_run_wallet"], float | int):
+        return _start_cap
+    else:
+        return _start_cap.get("stake_currency")