nikdoof/leggen
1
1
Fork 0
mirror of https://github.com/elisiariocouto/leggen.git synced 2026-01-30 13:58:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(notifications/telegram): Escape characters when notifying via Telegram.

Browse Source
This commit is contained in:
Elisiário Couto
2024-09-15 15:52:17 +01:00
parent e7662bc3dd
commit 7efbccfc90
1 changed files with 24 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
leggen/notifications/telegram.py
View File

@@ -7,13 +7,24 @@ from leggen.utils.text import info
def escape_markdown(text: str) -> str: def escape_markdown(text: str) -> str:
return ( return (
str(text) str(text)
.replace("-", "\\-") .replace("_", "\\_")
.replace("#", "\\#") .replace("*", "\\*")
.replace(".", "\\.") .replace("[", "\\[")
.replace("$", "\\$") .replace("]", "\\]")
.replace("+", "\\+")
.replace("(", "\\(") .replace("(", "\\(")
.replace(")", "\\)") .replace(")", "\\)")
.replace("~", "\\~")
.replace("`", "\\`")
.replace(">", "\\>")
.replace("#", "\\#")
.replace("+", "\\+")
.replace("-", "\\-")
.replace("=", "\\=")
.replace("|", "\\|")
.replace("{", "\\{")
.replace("}", "\\}")
.replace(".", "\\.")
.replace("!", "\\!")
) )
@@ -23,13 +34,15 @@ def send_expire_notification(ctx: click.Context, notification: dict):
bot_url = f"https://api.telegram.org/bot{token}/sendMessage" bot_url = f"https://api.telegram.org/bot{token}/sendMessage"
info("Sending expiration notification to Telegram") info("Sending expiration notification to Telegram")
message = "*💲 [Leggen](https://github.com/elisiariocouto/leggen)*\n" message = "*💲 [Leggen](https://github.com/elisiariocouto/leggen)*\n"
message += f"Your account {notification['bank']} ({notification['requisition_id']}) is in {notification['status']} status. Days left: {notification['days_left']}\n" message += escape_markdown(
f"Your account {notification['bank']} ({notification['requisition_id']}) is in {notification['status']} status. Days left: {notification['days_left']}\n"
)
res = requests.post( res = requests.post(
bot_url, bot_url,
json={ json={
"chat_id": chat_id, "chat_id": chat_id,
"text": escape_markdown(message), "text": message,
"parse_mode": "MarkdownV2", "parse_mode": "MarkdownV2",
}, },
) )
@@ -49,15 +62,15 @@ def send_transaction_message(ctx: click.Context, transactions: list):
message += f"{len(transactions)} new transaction matches\n\n" message += f"{len(transactions)} new transaction matches\n\n"
for transaction in transactions: for transaction in transactions:
message += f"*Name*: {transaction['name']}\n" message += f"*Name*: {escape_markdown(transaction['name'])}\n"
message += f"*Value*: {transaction['value']}{transaction['currency']}\n" message += f"*Value*: {escape_markdown(transaction['value'])}{escape_markdown(transaction['currency'])}\n"
message += f"*Date*: {transaction['date']}\n\n" message += f"*Date*: {escape_markdown(transaction['date'])}\n\n"
res = requests.post( res = requests.post(
bot_url, bot_url,
json={ json={
"chat_id": chat_id, "chat_id": chat_id,
"text": escape_markdown(message), "text": message,
"parse_mode": "MarkdownV2", "parse_mode": "MarkdownV2",
}, },
) )