adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Make _check_available_stake_amount() a separate method

Browse Source
This commit is contained in:
hroff-1902 2019-12-28 02:25:43 +03:00
parent 86f2693040
commit 243bcb2368
1 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
freqtrade/freqtradebot.py
View File

@ -206,9 +206,8 @@ class FreqtradeBot:
def get_trade_stake_amount(self, pair) -> Optional[float]: def get_trade_stake_amount(self, pair) -> Optional[float]:
""" """
Check if stake amount can be fulfilled with the available balance Calculate stake amount for the trade
for the stake currency :return: float: Stake amount
:return: float: Stake Amount
""" """
if self.edge: if self.edge:
return self.edge.stake_amount( return self.edge.stake_amount(
@ -220,16 +219,24 @@ class FreqtradeBot:
else: else:
stake_amount = self.config['stake_amount'] stake_amount = self.config['stake_amount']
available_amount = self.wallets.get_free(self.config['stake_currency'])
if stake_amount == constants.UNLIMITED_STAKE_AMOUNT: if stake_amount == constants.UNLIMITED_STAKE_AMOUNT:
open_trades = len(Trade.get_open_trades()) open_trades = len(Trade.get_open_trades())
if open_trades >= self.config['max_open_trades']: if open_trades >= self.config['max_open_trades']:
logger.warning("Can't open a new trade: max number of trades is reached") logger.warning("Can't open a new trade: max number of trades is reached")
return None return None
available_amount = self.wallets.get_free(self.config['stake_currency'])
return available_amount / (self.config['max_open_trades'] - open_trades) return available_amount / (self.config['max_open_trades'] - open_trades)
# Check if stake_amount is fulfilled return self._check_available_stake_amount(stake_amount)
def _check_available_stake_amount(self, stake_amount) -> float:
"""
Check if stake amount can be fulfilled with the available balance
for the stake currency
:return: float: Stake amount
"""
available_amount = self.wallets.get_free(self.config['stake_currency'])
if available_amount < stake_amount: if available_amount < stake_amount:
raise DependencyException( raise DependencyException(
f"Available balance ({available_amount} {self.config['stake_currency']}) is " f"Available balance ({available_amount} {self.config['stake_currency']}) is "