mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-29 03:56:43 +08:00
136 lines
4.9 KiB
Python
136 lines
4.9 KiB
Python
"""钉钉自定义机器人 Webhook 客户端。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import base64
|
|
import hashlib
|
|
import hmac
|
|
import time
|
|
from typing import Optional
|
|
from urllib.parse import parse_qsl, urlencode, urlsplit, urlunsplit
|
|
|
|
from app.adapters.network.http import RequestUtils
|
|
from app.runtime.log import logger
|
|
|
|
|
|
class DingTalk:
|
|
"""通过钉钉自定义机器人 Webhook 发送 Markdown 通知。"""
|
|
|
|
def __init__(
|
|
self,
|
|
DINGTALK_WEBHOOK: Optional[str] = None,
|
|
DINGTALK_SECRET: Optional[str] = None,
|
|
**kwargs,
|
|
) -> None:
|
|
"""保存 Webhook 与可选加签密钥，网络请求延迟到实际发送时执行。"""
|
|
self._webhook = (DINGTALK_WEBHOOK or "").strip()
|
|
self._secret = (DINGTALK_SECRET or "").strip()
|
|
self._req = RequestUtils(content_type="application/json", timeout=30)
|
|
|
|
def get_state(self) -> bool:
|
|
"""检查 Webhook 是否为可发送请求的 HTTP(S) 地址。"""
|
|
if not self._webhook:
|
|
return False
|
|
parsed = urlsplit(self._webhook)
|
|
return parsed.scheme in {"http", "https"} and bool(parsed.netloc)
|
|
|
|
def build_request_url(self, timestamp: Optional[int] = None) -> str:
|
|
"""按钉钉加签协议向 Webhook 查询参数追加时间戳与签名。"""
|
|
if not self._secret:
|
|
return self._webhook
|
|
timestamp = timestamp if timestamp is not None else int(time.time() * 1000)
|
|
string_to_sign = f"{timestamp}\n{self._secret}".encode("utf-8")
|
|
digest = hmac.new(
|
|
self._secret.encode("utf-8"),
|
|
string_to_sign,
|
|
digestmod=hashlib.sha256,
|
|
).digest()
|
|
signature = base64.b64encode(digest).decode("utf-8")
|
|
parsed = urlsplit(self._webhook)
|
|
query = dict(parse_qsl(parsed.query, keep_blank_values=True))
|
|
query.update({"timestamp": str(timestamp), "sign": signature})
|
|
return urlunsplit(
|
|
(parsed.scheme, parsed.netloc, parsed.path, urlencode(query), parsed.fragment)
|
|
)
|
|
|
|
@staticmethod
|
|
def build_markdown(
|
|
title: Optional[str],
|
|
text: Optional[str] = None,
|
|
image: Optional[str] = None,
|
|
link: Optional[str] = None,
|
|
) -> tuple[str, str]:
|
|
"""将 MoviePilot 通知字段转换为钉钉 Markdown 标题与正文。"""
|
|
raw_title = str(title or "").strip()
|
|
title_lines = raw_title.splitlines()
|
|
markdown_title = title_lines[0].strip() if title_lines else ""
|
|
markdown_title = markdown_title or "MoviePilot 通知"
|
|
|
|
content_parts = []
|
|
if raw_title:
|
|
content_parts.append(f"### {raw_title}")
|
|
if text:
|
|
content_parts.append(str(text).strip())
|
|
if image:
|
|
content_parts.append(f"")
|
|
if link:
|
|
content_parts.append(f"[查看详情]({link})")
|
|
if not content_parts:
|
|
content_parts.append(markdown_title)
|
|
return markdown_title, "\n\n".join(part for part in content_parts if part)
|
|
|
|
def send_msg(
|
|
self,
|
|
title: Optional[str],
|
|
text: Optional[str] = None,
|
|
image: Optional[str] = None,
|
|
userid: Optional[str] = None,
|
|
link: Optional[str] = None,
|
|
) -> bool:
|
|
"""向机器人所在群发送一条 Markdown 消息并校验钉钉业务返回码。"""
|
|
if not title and not text:
|
|
logger.warning("钉钉通知标题和内容不能同时为空")
|
|
return False
|
|
if not self.get_state():
|
|
logger.error("钉钉自定义机器人 Webhook 配置不完整")
|
|
return False
|
|
|
|
markdown_title, markdown_text = self.build_markdown(
|
|
title=title,
|
|
text=text,
|
|
image=image,
|
|
link=link,
|
|
)
|
|
payload = {
|
|
"msgtype": "markdown",
|
|
"markdown": {
|
|
"title": markdown_title,
|
|
"text": markdown_text,
|
|
},
|
|
}
|
|
response = self._req.post_res(url=self.build_request_url(), json=payload)
|
|
if response is None:
|
|
logger.error("钉钉自定义机器人请求失败")
|
|
return False
|
|
try:
|
|
if response.status_code != 200:
|
|
logger.error(f"钉钉自定义机器人返回 HTTP {response.status_code}")
|
|
return False
|
|
try:
|
|
result = response.json()
|
|
except ValueError:
|
|
logger.error("钉钉自定义机器人返回了无法解析的响应")
|
|
return False
|
|
if not isinstance(result, dict):
|
|
logger.error("钉钉自定义机器人返回了非对象响应")
|
|
return False
|
|
if result.get("errcode") == 0:
|
|
return True
|
|
logger.error(
|
|
"钉钉自定义机器人返回错误："
|
|
f"{result.get('errcode')}-{result.get('errmsg') or '未知错误'}"
|
|
)
|
|
return False
|
|
finally:
|
|
response.close()
|