Fix proxy env bug in Telegram alert

This commit is contained in:
Gan, Jimmy
2026-02-20 22:21:16 +08:00
parent dd3e8e0d2c
commit 17e7073fab
+3 -4
View File
@@ -1,6 +1,5 @@
from datetime import timedelta from datetime import timedelta
import asyncio import asyncio
import os
import httpx import httpx
from fastapi import APIRouter, Depends, HTTPException, status, Request from fastapi import APIRouter, Depends, HTTPException, status, Request
from pydantic import BaseModel from pydantic import BaseModel
@@ -36,11 +35,11 @@ async def send_failed_login_alert(ip_address: str, username: str, reason: str):
return return
msg = f"🚨 *Dashboard Security Alert* 🚨\nFailed login attempt detected.\n\n👤 User: `{username}`\n🌐 IP: `{ip_address}`\n❌ Reason: {reason}" msg = f"🚨 *Dashboard Security Alert* 🚨\nFailed login attempt detected.\n\n👤 User: `{username}`\n🌐 IP: `{ip_address}`\n❌ Reason: {reason}"
try: try:
client_options = {}
if config.TELEGRAM_PROXY: if config.TELEGRAM_PROXY:
os.environ["HTTP_PROXY"] = config.TELEGRAM_PROXY client_options["proxy"] = config.TELEGRAM_PROXY
os.environ["HTTPS_PROXY"] = config.TELEGRAM_PROXY
async with httpx.AsyncClient() as client: async with httpx.AsyncClient(**client_options) as client:
res = await client.post( res = await client.post(
f"https://api.telegram.org/bot{config.TELEGRAM_BOT_TOKEN}/sendMessage", f"https://api.telegram.org/bot{config.TELEGRAM_BOT_TOKEN}/sendMessage",
data={"chat_id": config.TELEGRAM_CHAT_ID, "text": msg, "parse_mode": "Markdown"}, data={"chat_id": config.TELEGRAM_CHAT_ID, "text": msg, "parse_mode": "Markdown"},