fix: 修复部分已知问题

hyb

2026-01-30 44480e71b27aa9d4cb8441f50c873f1b110e9691

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
import time
import hmac
import hashlib
import base64
import urllib.parse
import requests
import json
 
class DingTalkHelper:
    def __init__(self, access_token, secret):
        self.access_token = access_token
        self.secret = secret
 
    def calculate_sign(self):
        # 获取当前时间戳，单位为毫秒
        timestamp = str(round(time.time() * 1000))
 
        # 使用 HMAC SHA256 加密算法生成签名
        string_to_sign = '{}\n{}'.format(timestamp, self.secret)
        string_to_sign_enc = string_to_sign.encode('utf-8')
        secret_enc = self.secret.encode('utf-8')
 
        # 计算 HMAC 值
        hmac_code = hmac.new(secret_enc, string_to_sign_enc, digestmod=hashlib.sha256).digest()
 
        # 使用 base64 编码并对 URL 进行编码
        sign = urllib.parse.quote_plus(base64.b64encode(hmac_code))
 
        return timestamp, sign
 
    def send_markdown(self, title, text, is_at_all=False):
        timestamp, sign = self.calculate_sign()
        dingtalk_webhook = f'https://oapi.dingtalk.com/robot/send?access_token={self.access_token}&timestamp={timestamp}&sign={sign}'
 
        payload = {
            "msgtype": "markdown",
            "markdown": {
                "title": title,
                "text": text
            },
            "at": {
                "isAtAll": is_at_all
            }
        }
 
        try:
            response = requests.post(dingtalk_webhook, json=payload)
            if response.status_code == 200:
                print("Markdown消息发送成功！")
            else:
                print(f"发送失败，状态码：{response.status_code}")
        except Exception as e:
            print(f"发送异常：{e}")
 
    def send_message(self, message):
        # 计算签名和时间戳
        timestamp, sign = self.calculate_sign()
 
        # 构建钉钉 Webhook URL，拼接上 timestamp 和 sign
        dingtalk_webhook = f'https://oapi.dingtalk.com/robot/send?access_token={self.access_token}&timestamp={timestamp}&sign={sign}'
 
        headers = {
            'Content-Type': 'application/json',
        }
 
        # 构建发送的消息内容
        payload = {
            "msgtype": "text",
            "text": {
                "content": message
            }
        }
 
        try:
            # 发送请求
            response = requests.post(dingtalk_webhook, json=payload, headers=headers)
 
            # 输出返回的状态码和响应内容
            print(f"Response status: {response.status_code}")
            print(f"Response text: {response.text}")
 
            if response.status_code == 200:
                print("Message sent successfully!")
            else:
                print("Failed to send message.")
        except Exception as e:
            print(f"Error occurred: {e}")