chore: reduce dict lookups, reuse attribute

This commit is contained in:
Matthias
2024-11-26 07:13:05 +01:00
parent b4b6de4e0d
commit 15c1a8ee0b
+7 -7
View File
@@ -252,7 +252,7 @@ class Wallets:
else: else:
tot_profit = Trade.get_total_closed_profit() tot_profit = Trade.get_total_closed_profit()
open_stakes = Trade.total_open_trades_stakes() open_stakes = Trade.total_open_trades_stakes()
available_balance = self.get_free(self._config["stake_currency"]) available_balance = self.get_free(self._stake_currency)
return available_balance - tot_profit + open_stakes return available_balance - tot_profit + open_stakes
def get_total_stake_amount(self): def get_total_stake_amount(self):
@@ -272,9 +272,9 @@ class Wallets:
# Ensure <tradable_balance_ratio>% is used from the overall balance # Ensure <tradable_balance_ratio>% is used from the overall balance
# Otherwise we'd risk lowering stakes with each open trade. # Otherwise we'd risk lowering stakes with each open trade.
# (tied up + current free) * ratio) - tied up # (tied up + current free) * ratio) - tied up
available_amount = ( available_amount = (val_tied_up + self.get_free(self._stake_currency)) * self._config[
val_tied_up + self.get_free(self._config["stake_currency"]) "tradable_balance_ratio"
) * self._config["tradable_balance_ratio"] ]
return available_amount return available_amount
def get_available_stake_amount(self) -> float: def get_available_stake_amount(self) -> float:
@@ -285,7 +285,7 @@ class Wallets:
(<open_trade stakes> + free amount) * tradable_balance_ratio - <open_trade stakes> (<open_trade stakes> + free amount) * tradable_balance_ratio - <open_trade stakes>
""" """
free = self.get_free(self._config["stake_currency"]) free = self.get_free(self._stake_currency)
return min(self.get_total_stake_amount() - Trade.total_open_trades_stakes(), free) return min(self.get_total_stake_amount() - Trade.total_open_trades_stakes(), free)
def _calculate_unlimited_stake_amount( def _calculate_unlimited_stake_amount(
@@ -344,8 +344,8 @@ class Wallets:
if edge: if edge:
stake_amount = edge.stake_amount( stake_amount = edge.stake_amount(
pair, pair,
self.get_free(self._config["stake_currency"]), self.get_free(self._stake_currency),
self.get_total(self._config["stake_currency"]), self.get_total(self._stake_currency),
val_tied_up, val_tied_up,
) )
else: else: