60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
import logging
|
|
from typing import Any, Dict
|
|
|
|
from freqtrade.enums import RPCMessageType
|
|
from freqtrade.rpc import RPC
|
|
from freqtrade.rpc.webhook import Webhook
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class Discord(Webhook):
|
|
def __init__(self, rpc: 'RPC', config: Dict[str, Any]):
|
|
# super().__init__(rpc, config)
|
|
self.rpc = rpc
|
|
self.config = config
|
|
self.strategy = config.get('strategy', '')
|
|
self.timeframe = config.get('timeframe', '')
|
|
|
|
self._url = self.config['discord']['webhook_url']
|
|
self._format = 'json'
|
|
self._retries = 1
|
|
self._retry_delay = 0.1
|
|
|
|
def cleanup(self) -> None:
|
|
"""
|
|
Cleanup pending module resources.
|
|
This will do nothing for webhooks, they will simply not be called anymore
|
|
"""
|
|
pass
|
|
|
|
def send_msg(self, msg) -> None:
|
|
logger.info(f"Sending discord message: {msg}")
|
|
|
|
if msg['type'].value in self.config['discord']:
|
|
|
|
msg['strategy'] = self.strategy
|
|
msg['timeframe'] = self.timeframe
|
|
fields = self.config['discord'].get(msg['type'].value)
|
|
color = 0x0000FF
|
|
if msg['type'] in (RPCMessageType.EXIT, RPCMessageType.EXIT_FILL):
|
|
profit_ratio = msg.get('profit_ratio')
|
|
color = (0x00FF00 if profit_ratio > 0 else 0xFF0000)
|
|
|
|
embeds = [{
|
|
'title': f"Trade: {msg['pair']} {msg['type'].value}",
|
|
'color': color,
|
|
'fields': [],
|
|
|
|
}]
|
|
for f in fields:
|
|
for k, v in f.items():
|
|
v = v.format(**msg)
|
|
embeds[0]['fields'].append( # type: ignore
|
|
{'name': k, 'value': v, 'inline': True})
|
|
|
|
# Send the message to discord channel
|
|
payload = {'embeds': embeds}
|
|
self._send_msg(payload)
|