mirror of
https://ghfast.top/https://github.com/aeroxw/tick-stock-panel.git
synced 2026-09-12 22:34:18 +08:00
shy3130
f8914c9140
- 推送: 新增企业微信群机器人通道(与飞书并列), 监控告警/复盘报告多渠道分发 - 个股分析: 历史报告改为右侧常驻栏(展示全部股票报告), 进页面自动恢复上次选股 - 价位算法: 成交密集区改用换手率衰减模型(国内主流筹码分布); 缺口回补判定修正(必须完全穿越缺口) - 回测: 回撤止盈改纯峰值口径(与 trailing_stop 一致); 佣金/印花税/滑点合并一行; 建仓口径加问号气泡说明 - 分时图: 自选列表迷你分时图加渐变填充(对齐个股对话框风格) - 图表: 个股分析日K图右侧标签适配浅色主题; 价位标签与下方文字行双向 hover 高亮联动
296 lines
11 KiB
Python
296 lines
11 KiB
Python
"""Webhook 推送适配器 — 把告警事件推送到外部 IM / 量化软件。
|
|
|
|
职责: 把后端产生的告警事件, 通过用户配置的 Webhook 地址推送到外部。
|
|
目前支持飞书群机器人; QMT / ptrade 等量化通道为待定。
|
|
|
|
飞书自定义机器人接入:
|
|
1. 飞书群 → 群设置 → 群机器人 → 添加「自定义机器人」
|
|
2. 复制生成的 Webhook 地址 (形如 https://open.feishu.cn/open-apis/bot/v2/hook/xxx)
|
|
3. (可选) 安全设置 → 启用「签名校验」, 记录签名密钥(secret)
|
|
4. 填入设置页「飞书 Webhook」配置
|
|
|
|
设计: 失败静默降级, 绝不因推送失败阻断告警主流程 (落盘 / SSE 推送)。
|
|
去重不在本层做, 复用 MonitorRuleEngine 的 cooldown。
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import base64
|
|
import hashlib
|
|
import hmac
|
|
import logging
|
|
import time
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# 单次推送最长字符 (飞书单条文本消息上限 30KB, 这里保守截断避免刷屏)
|
|
_MAX_LEN = 500
|
|
|
|
# 卡片消息正文最长字符 (飞书 interactive 卡片上限 30KB, 保守留余量给标题/结构)
|
|
_CARD_MAX_LEN = 28000
|
|
|
|
# 飞书自定义机器人 Webhook 前缀 (用于 URL 合法性校验)
|
|
FEISHU_HOOK_PREFIX = "https://open.feishu.cn/open-apis/bot/v2/hook/"
|
|
|
|
|
|
def _truncate(text: str) -> str:
|
|
"""截断超长文本。"""
|
|
text = (text or "").strip()
|
|
return text[:_MAX_LEN] + ("…" if len(text) > _MAX_LEN else "")
|
|
|
|
|
|
def is_valid_feishu_url(url: str) -> bool:
|
|
"""校验是否为合法的飞书自定义机器人 Webhook 地址。"""
|
|
return bool(url) and url.startswith(FEISHU_HOOK_PREFIX)
|
|
|
|
|
|
def _gen_sign(timestamp: str, secret: str) -> str:
|
|
"""计算飞书自定义机器人签名。
|
|
|
|
算法 (官方): 把 `timestamp + "\\n" + secret` 作为签名字符串 (key),
|
|
用 HmacSHA256 计算空字符串的签名结果, 再 Base64 编码。
|
|
"""
|
|
string_to_sign = f"{timestamp}\n{secret}"
|
|
hmac_code = hmac.new(
|
|
string_to_sign.encode("utf-8"),
|
|
digestmod=hashlib.sha256,
|
|
).digest()
|
|
return base64.b64encode(hmac_code).decode("utf-8")
|
|
|
|
|
|
def _truncate_card(text: str) -> str:
|
|
"""截断卡片正文 (留余量给标题与卡片结构)。"""
|
|
text = (text or "").strip()
|
|
return text[:_CARD_MAX_LEN] + ("…" if len(text) > _CARD_MAX_LEN else "")
|
|
|
|
|
|
def _post_feishu(webhook_url: str, payload: dict, secret: str) -> bool:
|
|
"""发送一次飞书 webhook 请求并判定成败 (供 text / card 共用)。
|
|
|
|
成功响应: HTTP 200 且业务 code=0 (或非 JSON 的 200)。失败静默返回 False。
|
|
"""
|
|
try:
|
|
import httpx
|
|
|
|
# 启用签名校验时, 请求体须带 timestamp + sign (秒级时间戳)
|
|
if secret:
|
|
timestamp = str(int(time.time()))
|
|
payload["timestamp"] = timestamp
|
|
payload["sign"] = _gen_sign(timestamp, secret)
|
|
|
|
resp = httpx.post(webhook_url, json=payload, timeout=5.0)
|
|
# 飞书成功响应: {"code":0,"msg":"success"} (或 StatusCode 200 + Extra)
|
|
if resp.status_code == 200:
|
|
try:
|
|
data = resp.json()
|
|
# code=0 表示飞书业务侧成功; 部分版本无 code 字段则按 msg 判断
|
|
if isinstance(data, dict):
|
|
code = data.get("code", data.get("StatusCode", 0))
|
|
if code == 0:
|
|
return True
|
|
logger.debug("飞书推送业务失败: %s", data)
|
|
return False
|
|
except ValueError:
|
|
# 非 JSON 响应但 HTTP 200, 视为成功
|
|
return True
|
|
logger.debug("飞书推送 HTTP %s: %s", resp.status_code, resp.text[:200])
|
|
return False
|
|
except Exception as e: # noqa: BLE001
|
|
logger.debug("飞书 Webhook 推送失败: %s", e)
|
|
return False
|
|
|
|
|
|
def send_feishu(webhook_url: str, title: str, body: str, secret: str = "") -> bool:
|
|
"""推送一条文本消息到飞书群机器人。
|
|
|
|
Args:
|
|
webhook_url: 飞书自定义机器人 Webhook 地址
|
|
title: 消息标题 (与正文拼接为一条文本)
|
|
body: 消息正文
|
|
secret: 签名密钥 (机器人启用了「签名校验」时必填; 留空则不带签名)
|
|
|
|
Returns:
|
|
True=成功送达, False=失败或 URL 非法。
|
|
失败静默, 不抛异常 (Webhook 是辅助通道, 不能阻断告警主流程)。
|
|
"""
|
|
if not is_valid_feishu_url(webhook_url):
|
|
return False
|
|
|
|
text = _truncate(f"{title}\n{body}".strip())
|
|
if not text:
|
|
return False
|
|
|
|
payload: dict = {"msg_type": "text", "content": {"text": text}}
|
|
return _post_feishu(webhook_url, payload, secret)
|
|
|
|
|
|
def send_feishu_card(webhook_url: str, title: str, subtitle: str, body_md: str, secret: str = "") -> bool:
|
|
"""推送一条 interactive 卡片消息到飞书群机器人 —— 用 lark_md 渲染完整 markdown 报告。
|
|
|
|
飞书「自定义机器人」webhook 不支持文件附件, 但 interactive 卡片的 lark_md 元素
|
|
可渲染 markdown, 能承载完整复盘报告(通常 2-5KB, 远小于卡片 30KB 上限)。
|
|
|
|
Args:
|
|
webhook_url: 飞书自定义机器人 Webhook 地址
|
|
title: 卡片标题 (显示在蓝色 header)
|
|
subtitle: 副标题 (加粗显示, 如日期/情绪标签; 留空则省略)
|
|
body_md: 卡片正文 markdown (报告全文)
|
|
secret: 签名密钥 (启用签名校验时必填)
|
|
|
|
Returns:
|
|
True=成功送达, False=失败或 URL 非法。
|
|
失败静默, 不抛异常 (与 send_feishu 一致, 不阻断告警主流程)。
|
|
"""
|
|
if not is_valid_feishu_url(webhook_url):
|
|
return False
|
|
|
|
body = _truncate_card(body_md)
|
|
elements: list[dict] = []
|
|
if subtitle.strip():
|
|
elements.append({
|
|
"tag": "div",
|
|
"text": {"tag": "lark_md", "content": f"**{subtitle.strip()}**"},
|
|
})
|
|
elements.append({"tag": "hr"})
|
|
elements.append({
|
|
"tag": "div",
|
|
"text": {"tag": "lark_md", "content": body},
|
|
})
|
|
|
|
payload: dict = {
|
|
"msg_type": "interactive",
|
|
"card": {
|
|
"config": {"wide_screen_mode": True},
|
|
"header": {
|
|
"title": {"tag": "plain_text", "content": title},
|
|
"template": "blue",
|
|
},
|
|
"elements": elements,
|
|
},
|
|
}
|
|
return _post_feishu(webhook_url, payload, secret)
|
|
|
|
|
|
# ================================================================
|
|
# 企业微信群机器人
|
|
# ================================================================
|
|
#
|
|
# 与飞书自定义机器人几乎同构: 同样是"群机器人 Webhook + POST JSON"。
|
|
# 关键差异:
|
|
# 1. Webhook 形态: https://qyapi.weixin.qq.com/cgi-bin/webhook/send?key=xxx
|
|
# 2. 无需签名校验 (key 本身即凭证; 企业微信群机器人可选"签名校验"但极少用)
|
|
# 3. Markdown 原生支持 (msg_type=markdown), 不必像飞书那样包进 interactive 卡片
|
|
# 4. 成功响应: {"errcode":0,"errmsg":"ok"}
|
|
#
|
|
# 限制: 每个机器人每分钟最多 20 条消息 (超出会被限流 460min 内不可用),
|
|
# 依赖 MonitorRuleEngine 的 cooldown 去重即可应对告警场景。
|
|
# 企业微信群的消息可在绑定的个人微信接收, 实现"微信推送"体验。
|
|
|
|
WECOM_HOOK_PREFIX = "https://qyapi.weixin.qq.com/cgi-bin/webhook/send"
|
|
|
|
|
|
def is_valid_wecom_url(url: str) -> bool:
|
|
"""校验是否为合法的企业微信群机器人 Webhook 地址。
|
|
|
|
允许两种写法:
|
|
- 完整: https://qyapi.weixin.qq.com/cgi-bin/webhook/send?key=xxx
|
|
- 仅 key: xxx (企业微信群机器人 key 为 36 位 UUID 样式, 保存时自动补全)
|
|
"""
|
|
if not url:
|
|
return False
|
|
if url.startswith(WECOM_HOOK_PREFIX):
|
|
return True
|
|
# 纯 key: 企业微信 key 形如 12345678-1234-1234-1234-1234567890ab (36 位),
|
|
# 但用户可能截断, 放宽到 >= 20 位的无空格无斜杠字符串。
|
|
url = url.strip()
|
|
if " " in url or "/" in url or "?" in url:
|
|
return False
|
|
return len(url) >= 20
|
|
|
|
|
|
def normalize_wecom_url(url: str) -> str:
|
|
"""把纯 key 补全为完整 Webhook URL。已是完整 URL 则原样返回。"""
|
|
url = (url or "").strip()
|
|
if not url:
|
|
return ""
|
|
if url.startswith(WECOM_HOOK_PREFIX):
|
|
return url
|
|
return f"{WECOM_HOOK_PREFIX}?key={url}"
|
|
|
|
|
|
def _post_wecom(webhook_url: str, payload: dict) -> bool:
|
|
"""发送一次企业微信 webhook 请求并判定成败。
|
|
|
|
成功响应: HTTP 200 且 errcode=0。失败静默返回 False。
|
|
"""
|
|
try:
|
|
import httpx
|
|
|
|
resp = httpx.post(webhook_url, json=payload, timeout=5.0)
|
|
if resp.status_code == 200:
|
|
try:
|
|
data = resp.json()
|
|
if isinstance(data, dict):
|
|
# errcode=0 表示成功; 45009=频率限制, 其它非零=业务失败
|
|
if data.get("errcode") == 0:
|
|
return True
|
|
logger.debug("企业微信推送业务失败: %s", data)
|
|
return False
|
|
except ValueError:
|
|
return True
|
|
logger.debug("企业微信推送 HTTP %s: %s", resp.status_code, resp.text[:200])
|
|
return False
|
|
except Exception as e: # noqa: BLE001
|
|
logger.debug("企业微信 Webhook 推送失败: %s", e)
|
|
return False
|
|
|
|
|
|
def send_wecom(webhook_url: str, title: str, body: str) -> bool:
|
|
"""推送一条文本消息到企业微信群机器人。
|
|
|
|
Args:
|
|
webhook_url: 企业微信群机器人 Webhook 地址 (或纯 key, 会自动补全)
|
|
title: 消息标题 (与正文拼接为一条文本)
|
|
body: 消息正文
|
|
|
|
Returns:
|
|
True=成功送达, False=失败或 URL 非法。
|
|
失败静默, 不抛异常 (与 send_feishu 一致)。
|
|
"""
|
|
webhook_url = normalize_wecom_url(webhook_url)
|
|
if not is_valid_wecom_url(webhook_url):
|
|
return False
|
|
|
|
text = _truncate(f"{title}\n{body}".strip())
|
|
if not text:
|
|
return False
|
|
|
|
payload: dict = {"msg_type": "text", "text": {"content": text}}
|
|
return _post_wecom(webhook_url, payload)
|
|
|
|
|
|
def send_wecom_markdown(webhook_url: str, title: str, body_md: str) -> bool:
|
|
"""推送一条 Markdown 消息到企业微信群机器人 —— 承载完整复盘报告。
|
|
|
|
企业微信群机器人原生支持 markdown 类型 (比飞书 interactive 卡片简单),
|
|
支持 # ## **粗体** >引用 - 列表 等基础语法, 单条上限 4096 字节。
|
|
|
|
Args:
|
|
webhook_url: 企业微信群机器人 Webhook 地址 (或纯 key)
|
|
title: 标题 (作为一级标题 ## 拼到正文前)
|
|
body_md: markdown 正文
|
|
|
|
Returns:
|
|
True=成功送达, False=失败或 URL 非法。
|
|
"""
|
|
webhook_url = normalize_wecom_url(webhook_url)
|
|
if not is_valid_wecom_url(webhook_url):
|
|
return False
|
|
|
|
content = f"## {title}\n\n{_truncate_card(body_md)}"
|
|
if not content.strip():
|
|
return False
|
|
|
|
payload: dict = {"msg_type": "markdown", "markdown": {"content": content}}
|
|
return _post_wecom(webhook_url, payload)
|
|
|