lychiyu/freqtrade_origin
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2024-11-14 20:23:57 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

chore: slightly refactor telegram loudness control

Browse Source
This commit is contained in:
Matthias 2024-11-06 19:50:18 +01:00
parent 899db2de78
commit 03eb079a5e
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
freqtrade/rpc/telegram.py
View File

@ -547,14 +547,14 @@ class Telegram(RPCHandler):
return None return None
return message return message
def send_msg(self, msg: RPCSendMsg) -> None: def _message_loudness(
"""Send a message to telegram channel""" self, msg_type: RPCMessageType, exit_reason: str
) -> Literal["off", "silent", "on"]:
default_noti = "on" default_noti = "on"
msg_type = msg["type"] msg_type = msg_type
noti = "" noti = ""
if msg["type"] == RPCMessageType.EXIT: if msg_type == RPCMessageType.EXIT:
sell_noti = ( sell_noti = (
self._config["telegram"].get("notification_settings", {}).get(str(msg_type), {}) self._config["telegram"].get("notification_settings", {}).get(str(msg_type), {})
) )
@ -562,7 +562,7 @@ class Telegram(RPCHandler):
if isinstance(sell_noti, str): if isinstance(sell_noti, str):
noti = sell_noti noti = sell_noti
else: else:
noti = sell_noti.get(str(msg["exit_reason"]), default_noti) noti = sell_noti.get(str(exit_reason), default_noti)
else: else:
noti = ( noti = (
self._config["telegram"] self._config["telegram"]
@ -570,8 +570,14 @@ class Telegram(RPCHandler):
.get(str(msg_type), default_noti) .get(str(msg_type), default_noti)
) )
return noti
def send_msg(self, msg: RPCSendMsg) -> None:
"""Send a message to telegram channel"""
noti = self._message_loudness(msg["type"], msg.get("exit_reason", ""))
if noti == "off": if noti == "off":
logger.info(f"Notification '{msg_type}' not sent.") logger.info(f"Notification '{msg['type']}' not sent.")
# Notification disabled # Notification disabled
return return