63 lines
1.9 KiB
Python
63 lines
1.9 KiB
Python
"""企微 Webhook 通知服务：同步失败告警、Token 超限通知等。"""
|
||
import requests
|
||
from flask import current_app
|
||
|
||
|
||
def send_wecom_webhook(title: str, content: str):
|
||
"""通过企微群机器人 Webhook 发送通知。
|
||
|
||
Args:
|
||
title: 消息标题
|
||
content: 消息内容（支持 Markdown）
|
||
"""
|
||
webhook_url = current_app.config.get("WECOM_WEBHOOK_URL", "")
|
||
if not webhook_url:
|
||
current_app.logger.warning("WECOM_WEBHOOK_URL 未配置，跳过通知")
|
||
return
|
||
|
||
payload = {
|
||
"msgtype": "markdown",
|
||
"markdown": {
|
||
"content": f"## {title}\n\n{content}",
|
||
},
|
||
}
|
||
|
||
try:
|
||
resp = requests.post(webhook_url, json=payload, timeout=10)
|
||
result = resp.json()
|
||
if result.get("errcode") != 0:
|
||
current_app.logger.error(f"企微 Webhook 发送失败: {result}")
|
||
except Exception as e:
|
||
current_app.logger.error(f"企微 Webhook 请求异常: {e}")
|
||
|
||
|
||
def notify_sync_failure(datasource_name: str, error_msg: str):
|
||
"""数据源同步失败通知。"""
|
||
send_wecom_webhook(
|
||
"⚠️ 数据源同步失败",
|
||
f"**数据源**: {datasource_name}\n"
|
||
f"**错误**: {error_msg}\n"
|
||
f"**时间**: 请检查数据源配置并重试",
|
||
)
|
||
|
||
|
||
def notify_token_limit_reached(username: str, token_usage: int, limit: int):
|
||
"""Token 超限通知。"""
|
||
send_wecom_webhook(
|
||
"⚠️ Token 消耗超限",
|
||
f"**用户**: {username}\n"
|
||
f"**已用**: {token_usage:,} tokens\n"
|
||
f"**限额**: {limit:,} tokens\n"
|
||
f"**建议**: 联系管理员调整配额",
|
||
)
|
||
|
||
|
||
def notify_proposal_generated(username: str, customer_name: str):
|
||
"""方案生成完成通知。"""
|
||
send_wecom_webhook(
|
||
"✅ 推荐方案已生成",
|
||
f"**操作人**: {username}\n"
|
||
f"**客户**: {customer_name}\n"
|
||
f"**状态**: 方案已就绪，请查看",
|
||
)
|