119 lines
4.0 KiB
Python
119 lines
4.0 KiB
Python
import requests
|
||
import time
|
||
import hmac
|
||
import hashlib
|
||
import base64
|
||
import urllib.parse
|
||
from loguru import logger
|
||
|
||
|
||
class DingTalkBot:
|
||
"""
|
||
钉钉机器人类，通过webhook和可选的加签token向群聊发送消息提醒
|
||
"""
|
||
|
||
def __init__(self, webhook: str, secret: str = None):
|
||
"""
|
||
:param webhook: 钉钉自定义机器人webhook地址
|
||
:param secret: 加签密钥（可选）
|
||
"""
|
||
self.webhook = webhook
|
||
self.secret = secret
|
||
|
||
def _gen_signed_url(self):
|
||
"""
|
||
如果有加签token，根据钉钉接口算法拼接签名到url
|
||
"""
|
||
if not self.secret:
|
||
return self.webhook
|
||
timestamp = str(round(time.time() * 1000))
|
||
secret_enc = self.secret.encode("utf-8")
|
||
string_to_sign = f"{timestamp}\n{self.secret}"
|
||
string_to_sign_enc = string_to_sign.encode("utf-8")
|
||
hmac_code = hmac.new(
|
||
secret_enc, string_to_sign_enc, digestmod=hashlib.sha256
|
||
).digest()
|
||
sign = urllib.parse.quote_plus(base64.b64encode(hmac_code))
|
||
url = f"{self.webhook}×tamp={timestamp}&sign={sign}"
|
||
return url
|
||
|
||
def send_text(self, content: str, at_mobiles=None, is_at_all=False):
|
||
"""
|
||
发送文本消息
|
||
|
||
:param content: 消息内容
|
||
:param at_mobiles: 需要@的手机号组成的列表，可选
|
||
:param is_at_all: 是否@所有人，默认False
|
||
"""
|
||
at_mobiles = at_mobiles or []
|
||
data = {
|
||
"msgtype": "text",
|
||
"text": {"content": content},
|
||
"at": {"atMobiles": at_mobiles, "isAtAll": is_at_all},
|
||
}
|
||
|
||
url = self._gen_signed_url() if self.secret else self.webhook
|
||
|
||
try:
|
||
response = requests.post(url, json=data, timeout=5)
|
||
response.raise_for_status()
|
||
result = response.json()
|
||
if result.get("errcode", -1) != 0:
|
||
logger.error(f"钉钉消息发送失败: {result}")
|
||
else:
|
||
logger.info("钉钉消息发送成功")
|
||
except Exception as e:
|
||
logger.error(f"钉钉消息发送异常: {e}")
|
||
|
||
def send_markdown(self, title: str, text: str, at_mobiles=None, is_at_all=False):
|
||
"""
|
||
发送markdown消息
|
||
|
||
:param title: 消息标题
|
||
:param text: markdown格式的消息内容
|
||
:param at_mobiles: 需要@的手机号组成的列表，可选
|
||
:param is_at_all: 是否@所有人，默认False
|
||
"""
|
||
at_mobiles = at_mobiles or []
|
||
data = {
|
||
"msgtype": "markdown",
|
||
"markdown": {"title": title, "text": text},
|
||
"at": {"atMobiles": at_mobiles, "isAtAll": is_at_all},
|
||
}
|
||
|
||
url = self._gen_signed_url() if self.secret else self.webhook
|
||
|
||
try:
|
||
response = requests.post(url, json=data, timeout=5)
|
||
response.raise_for_status()
|
||
result = response.json()
|
||
if result.get("errcode", -1) != 0:
|
||
logger.error(f"钉钉markdown消息发送失败: {result}")
|
||
else:
|
||
logger.info("钉钉markdown消息发送成功")
|
||
except Exception as e:
|
||
logger.error(f"钉钉markdown消息发送异常: {e}")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
|
||
webhook = "https://oapi.dingtalk.com/robot/send?access_token=fb70c1561d8beba94b4f11568f4bb15e3ae07ccbdc8ac19676434a9d1cd17546"
|
||
secret = "SEC1ae7cd2f1a6f9da3611af37da3e7d954c1e8533fc073c6c8cc5e5af3b6e5926b"
|
||
|
||
# CTA 群机器人
|
||
# webhook = "https://oapi.dingtalk.com/robot/send?access_token=87c7abfcdd69b699c32da4e4f5981cd2ca6b0445474fc6ffb36f2ed0f6262fbb"
|
||
# secret = "SECf3d6b43f2f8a87ab91feffd052e71ec314fbf57a1842e483fe07af3c0a0e5aa6"
|
||
dingtalk = DingTalkBot(webhook, secret)
|
||
dingtalk.send_text("测试消息")
|
||
|
||
# 测试markdown消息
|
||
markdown_content = """
|
||
## 系统通知
|
||
- **状态**: 正常运行
|
||
- **CPU使用率**: 65%
|
||
- **内存使用率**: 78%
|
||
|
||
> 详细信息请查看监控面板
|
||
"""
|
||
dingtalk.send_markdown("系统状态报告", markdown_content)
|