116 lines
4.3 KiB
Python
116 lines
4.3 KiB
Python
"""
|
|
This module contains class to manage RPC communications (Telegram, Slack, ...)
|
|
"""
|
|
import logging
|
|
from typing import Any, Dict, List
|
|
|
|
from freqtrade.persistence import Trade
|
|
from freqtrade.rpc import RPC, RPCMessageType
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class RPCManager(object):
|
|
"""
|
|
Class to manage RPC objects (Telegram, Slack, ...)
|
|
"""
|
|
def __init__(self, freqtrade) -> None:
|
|
""" Initializes all enabled rpc modules """
|
|
self.registered_modules: List[RPC] = []
|
|
|
|
# Enable telegram
|
|
if freqtrade.config['telegram'].get('enabled', False):
|
|
logger.info('Enabling rpc.telegram ...')
|
|
from freqtrade.rpc.telegram import Telegram
|
|
self.registered_modules.append(Telegram(freqtrade))
|
|
|
|
# Enable Webhook
|
|
if freqtrade.config.get('webhook', {}).get('enabled', False):
|
|
logger.info('Enabling rpc.webhook ...')
|
|
from freqtrade.rpc.webhook import Webhook
|
|
self.registered_modules.append(Webhook(freqtrade))
|
|
|
|
def cleanup(self) -> None:
|
|
""" Stops all enabled rpc modules """
|
|
logger.info('Cleaning up rpc modules ...')
|
|
while self.registered_modules:
|
|
mod = self.registered_modules.pop()
|
|
logger.debug('Cleaning up rpc.%s ...', mod.name)
|
|
mod.cleanup()
|
|
del mod
|
|
|
|
def send_msg(self, msg: Dict[str, Any]) -> None:
|
|
"""
|
|
Send given message to all registered rpc modules.
|
|
A message consists of one or more key value pairs of strings.
|
|
e.g.:
|
|
{
|
|
'status': 'stopping bot'
|
|
}
|
|
"""
|
|
logger.info('Sending rpc message: %s', msg)
|
|
for mod in self.registered_modules:
|
|
logger.debug('Forwarding message to rpc.%s', mod.name)
|
|
mod.send_msg(msg)
|
|
|
|
def startup_messages(self, config, pairlist) -> None:
|
|
if config.get('dry_run', False):
|
|
self.send_msg({
|
|
'type': RPCMessageType.WARNING_NOTIFICATION,
|
|
'status': 'Dry run is enabled. All trades are simulated.'
|
|
})
|
|
stake_currency = config['stake_currency']
|
|
stake_amount = config['stake_amount']
|
|
minimal_roi = config['minimal_roi']
|
|
stoploss = config['stoploss']
|
|
trailing_stop = config['trailing_stop']
|
|
ticker_interval = config['ticker_interval']
|
|
exchange_name = config['exchange']['name']
|
|
strategy_name = config.get('strategy', '')
|
|
self.send_msg({
|
|
'type': RPCMessageType.CUSTOM_NOTIFICATION,
|
|
'status': f'*Exchange:* `{exchange_name}`\n'
|
|
f'*Stake per trade:* `{stake_amount} {stake_currency}`\n'
|
|
f'*Minimum ROI:* `{minimal_roi}`\n'
|
|
f'*{"Trailing " if trailing_stop else ""}Stoploss:* `{stoploss}`\n'
|
|
f'*Ticker Interval:* `{ticker_interval}`\n'
|
|
f'*Strategy:* `{strategy_name}`'
|
|
})
|
|
self.send_msg({
|
|
'type': RPCMessageType.STATUS_NOTIFICATION,
|
|
'status': f'Searching for {stake_currency} pairs to buy and sell '
|
|
f'based on {pairlist.short_desc()}'
|
|
})
|
|
|
|
def notify_sell(self, trade: Trade, config, current_rate: float):
|
|
profit_trade = trade.calc_profit(rate=trade.close_rate_requested)
|
|
|
|
profit_percent = trade.calc_profit_percent(trade.close_rate_requested)
|
|
gain = "profit" if profit_percent > 0 else "loss"
|
|
|
|
msg = {
|
|
'type': RPCMessageType.SELL_NOTIFICATION,
|
|
'exchange': trade.exchange.capitalize(),
|
|
'pair': trade.pair,
|
|
'gain': gain,
|
|
'limit': trade.close_rate_requested,
|
|
'amount': trade.amount,
|
|
'open_rate': trade.open_rate,
|
|
'current_rate': current_rate,
|
|
'profit_amount': profit_trade,
|
|
'profit_percent': profit_percent,
|
|
'sell_reason': trade.sell_reason
|
|
}
|
|
|
|
# For regular case, when the configuration exists
|
|
if 'stake_currency' in config and 'fiat_display_currency' in config:
|
|
stake_currency = config['stake_currency']
|
|
fiat_currency = config['fiat_display_currency']
|
|
msg.update({
|
|
'stake_currency': stake_currency,
|
|
'fiat_currency': fiat_currency,
|
|
})
|
|
|
|
# Send the message
|
|
self.send_msg(msg)
|