Move rpc-count calculation to _rpc class

This commit is contained in:
Matthias 2019-04-06 19:58:45 +02:00
parent 0ac80aacd1
commit 7a598f32dc
2 changed files with 10 additions and 7 deletions

View File

@ -459,7 +459,12 @@ class RPC(object):
if self._freqtrade.state != State.RUNNING:
raise RPCException('trader is not running')
return Trade.get_open_trades()
trades = Trade.get_open_trades()
return {
'current': len(trades),
'max': self._config['max_open_trades'],
'total stake': sum((trade.open_rate * trade.amount) for trade in trades)
}
def _rpc_whitelist(self) -> Dict:
""" Returns the currently active whitelist"""

View File

@ -456,12 +456,10 @@ class Telegram(RPC):
:return: None
"""
try:
trades = self._rpc_count()
message = tabulate({
'current': [len(trades)],
'max': [self._config['max_open_trades']],
'total stake': [sum((trade.open_rate * trade.amount) for trade in trades)]
}, headers=['current', 'max', 'total stake'], tablefmt='simple')
counts = self._rpc_count()
message = tabulate({k: [v] for k, v in counts.items()},
headers=['current', 'max', 'total stake'],
tablefmt='simple')
message = "<pre>{}</pre>".format(message)
logger.debug(message)
self._send_msg(message, parse_mode=ParseMode.HTML)