From f0bbc75038134d7debb9aed56d85eb2c2702c6b5 Mon Sep 17 00:00:00 2001
From: Matthias <xmatthias@outlook.com>
Date: Sun, 15 Dec 2019 09:48:35 +0100
Subject: [PATCH] Combine dry_run wallet into original Wallets class

---
 freqtrade/freqtradebot.py |  8 ++---
 freqtrade/rpc/telegram.py |  2 +-
 freqtrade/wallets.py      | 64 +++++++++++++++++++++------------------
 3 files changed, 39 insertions(+), 35 deletions(-)

diff --git a/freqtrade/freqtradebot.py b/freqtrade/freqtradebot.py
index df9fd0b17..a242c11ae 100644
--- a/freqtrade/freqtradebot.py
+++ b/freqtrade/freqtradebot.py
@@ -25,7 +25,7 @@ from freqtrade.rpc import RPCManager, RPCMessageType
 from freqtrade.pairlist.pairlistmanager import PairListManager
 from freqtrade.state import State
 from freqtrade.strategy.interface import IStrategy, SellType
-from freqtrade.wallets import Wallets, WalletsDry
+from freqtrade.wallets import Wallets
 
 logger = logging.getLogger(__name__)
 
@@ -65,10 +65,8 @@ class FreqtradeBot:
         persistence.init(self.config.get('db_url', None),
                          clean_open_orders=self.config.get('dry_run', False))
 
-        if self.config['dry_run']:
-            self.wallets = WalletsDry(self.config, self.exchange)
-        else:
-            self.wallets = Wallets(self.config, self.exchange)
+        self.wallets = Wallets(self.config, self.exchange)
+
         self.dataprovider = DataProvider(self.config, self.exchange)
 
         # Attach Dataprovider to Strategy baseclass
diff --git a/freqtrade/rpc/telegram.py b/freqtrade/rpc/telegram.py
index 4d7857f44..e36b46ba7 100644
--- a/freqtrade/rpc/telegram.py
+++ b/freqtrade/rpc/telegram.py
@@ -335,7 +335,7 @@ class Telegram(RPC):
             output = ''
             if self._config['dry_run']:
                 output += (
-                    f"Simulated balances - starting capital: "
+                    f"*Warning:*Simulated balances in Dry Mode.\nStarting capital: "
                     f"`{self._config['dry_run_wallet']}` {self._config['stake_currency']}.\n"
                     )
             for currency in result['currencies']:
diff --git a/freqtrade/wallets.py b/freqtrade/wallets.py
index eb2603776..f8dd0ee2f 100644
--- a/freqtrade/wallets.py
+++ b/freqtrade/wallets.py
@@ -4,7 +4,7 @@
 import logging
 from typing import Dict, NamedTuple, Any
 from freqtrade.exchange import Exchange
-from freqtrade import constants
+from freqtrade.persistence import Trade
 
 logger = logging.getLogger(__name__)
 
@@ -23,6 +23,7 @@ class Wallets:
         self._config = config
         self._exchange = exchange
         self._wallets: Dict[str, Wallet] = {}
+        self.start_cap = config['dry_run_wallet']
 
         self.update()
 
@@ -50,36 +51,12 @@ class Wallets:
         else:
             return 0
 
-    def update(self) -> None:
-
-        balances = self._exchange.get_balances()
-
-        for currency in balances:
-            self._wallets[currency] = Wallet(
-                currency,
-                balances[currency].get('free', None),
-                balances[currency].get('used', None),
-                balances[currency].get('total', None)
-            )
-
-        logger.info('Wallets synced.')
-
-    def get_all_balances(self) -> Dict[str, Any]:
-        return self._wallets
-
-
-class WalletsDry(Wallets):
-
-    def __init__(self, config: dict, exchange: Exchange) -> None:
-        self.start_cap = config['dry_run_wallet']
-        super().__init__(config, exchange)
-
-    def update(self) -> None:
-        """ Update does not do anything in dry-mode..."""
-        from freqtrade.persistence import Trade
+    def _update_dry(self) -> None:
+        """ Update from database in dry-run mode"""
         closed_trades = Trade.get_trades(Trade.is_open.is_(False)).all()
-        print(len(closed_trades))
+
         tot_profit = sum([trade.calc_profit() for trade in closed_trades])
+
         current_stake = self.start_cap + tot_profit
         self._wallets[self._config['stake_currency']] = Wallet(
             self._config['stake_currency'],
@@ -98,3 +75,32 @@ class WalletsDry(Wallets):
                 0,
                 trade.amount
             )
+
+    def _update_live(self) -> None:
+
+        balances = self._exchange.get_balances()
+
+        for currency in balances:
+            self._wallets[currency] = Wallet(
+                currency,
+                balances[currency].get('free', None),
+                balances[currency].get('used', None),
+                balances[currency].get('total', None)
+            )
+
+    def update(self) -> None:
+        if self._config['dry_run']:
+            self._update_dry()
+        else:
+            self._update_live()
+        logger.info('Wallets synced.')
+
+    def get_all_balances(self) -> Dict[str, Any]:
+        return self._wallets
+
+
+class WalletsDry(Wallets):
+
+    def __init__(self, config: dict, exchange: Exchange) -> None:
+
+        super().__init__(config, exchange)