CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2025-12-18 13:51:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

ruff format: rpc modules

Browse Source
This commit is contained in:
Matthias
2024-05-12 16:51:11 +02:00
parent cebbe0121e
commit 5f64cc8e76
23 changed files with 1994 additions and 1689 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
freqtrade/rpc/discord.py
View File

@@ -10,18 +10,18 @@ logger = logging.getLogger(__name__)
class Discord(Webhook):
def __init__(self, rpc: 'RPC', config: Config):
def __init__(self, rpc: "RPC", config: Config):
self._config = config
self.rpc = rpc
self.strategy = config.get('strategy', '')
self.timeframe = config.get('timeframe', '')
self.bot_name = config.get('bot_name', '')
self.strategy = config.get("strategy", "")
self.timeframe = config.get("timeframe", "")
self.bot_name = config.get("bot_name", "")
self._url = config['discord']['webhook_url']
self._format = 'json'
self._url = config["discord"]["webhook_url"]
self._format = "json"
self._retries = 1
self._retry_delay = 0.1
self._timeout = self._config['discord'].get('timeout', 10)
self._timeout = self._config["discord"].get("timeout", 10)
def cleanup(self) -> None:
"""
@@ -31,32 +31,31 @@ class Discord(Webhook):
pass
def send_msg(self, msg) -> None:
if (fields := self._config['discord'].get(msg['type'].value)):
if fields := self._config["discord"].get(msg["type"].value):
logger.info(f"Sending discord message: {msg}")
msg['strategy'] = self.strategy
msg['timeframe'] = self.timeframe
msg['bot_name'] = self.bot_name
msg["strategy"] = self.strategy
msg["timeframe"] = self.timeframe
msg["bot_name"] = self.bot_name
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)
title = msg['type'].value
if 'pair' in msg:
if msg["type"] in (RPCMessageType.EXIT, RPCMessageType.EXIT_FILL):
profit_ratio = msg.get("profit_ratio")
color = 0x00FF00 if profit_ratio > 0 else 0xFF0000
title = msg["type"].value
if "pair" in msg:
title = f"Trade: {msg['pair']} {msg['type'].value}"
embeds = [{
'title': title,
'color': color,
'fields': [],
}]
embeds = [
{
"title": title,
"color": color,
"fields": [],
}
]
for f in fields:
for k, v in f.items():
v = v.format(**msg)
embeds[0]['fields'].append(
{'name': k, 'value': v, 'inline': True})
embeds[0]["fields"].append({"name": k, "value": v, "inline": True})
# Send the message to discord channel
payload = {'embeds': embeds}
payload = {"embeds": embeds}
self._send_msg(payload)