Simplify wallet code

This commit is contained in:
Matthias 2021-02-27 10:31:21 +01:00
parent f65092459a
commit 324b9dbdff
2 changed files with 4 additions and 6 deletions

View File

@ -115,8 +115,7 @@ class Backtesting:
if self.config.get('enable_protections', False):
self.protections = ProtectionManager(self.config)
self.wallets = Wallets(self.config, self.exchange)
self.wallets._log = False
self.wallets = Wallets(self.config, self.exchange, log=False)
# Get maximum required startup period
self.required_startup = max([strat.startup_candle_count for strat in self.strategylist])

View File

@ -27,15 +27,14 @@ class Wallet(NamedTuple):
class Wallets:
def __init__(self, config: dict, exchange: Exchange, skip_update: bool = False) -> None:
def __init__(self, config: dict, exchange: Exchange, log: bool = True) -> None:
self._config = config
self._log = True
self._log = log
self._exchange = exchange
self._wallets: Dict[str, Wallet] = {}
self.start_cap = config['dry_run_wallet']
self._last_wallet_refresh = 0
if not skip_update:
self.update()
self.update()
def get_free(self, currency: str) -> float:
balance = self._wallets.get(currency)