ywyw2020
diff --git a/‎feapder/setting.py‎
Lines changed: 1 addition & 0 deletions b/‎feapder/setting.py‎
Lines changed: 1 addition & 0 deletions
diff --git a/‎feapder/utils/tools.py‎
Lines changed: 10 additions & 1 deletion b/‎feapder/utils/tools.py‎
Lines changed: 10 additions & 1 deletion
@@ -167,6 +167,7 @@
 DINGDING_WARNING_URL = ""  # 钉钉机器人api
 DINGDING_WARNING_PHONE = ""  # 报警人 支持列表，可指定多个
 DINGDING_WARNING_ALL = False  # 是否提示所有人， 默认为False
+DINGDING_WARNING_SECRET = None  # 加签密钥
 # 飞书报警
 # https://open.feishu.cn/document/ukTMukTMukTM/ucTM5YjL3ETO24yNxkjN#e1cdee9f
 FEISHU_WARNING_URL = ""  # 飞书机器人api
 
@@ -7,6 +7,7 @@
 @author: Boris
 @email: [email protected]
 """
+import hmac
 import asyncio
 import base64
 import calendar
@@ -2466,12 +2467,20 @@ def reach_freq_limit(rate_limit, *key):
 
 
 def dingding_warning(
-    message, message_prefix=None, rate_limit=None, url=None, user_phone=None
+    message, message_prefix=None, rate_limit=None, url=None, user_phone=None, secret=None
 ):
     # 为了加载最新的配置
     rate_limit = rate_limit if rate_limit is not None else setting.WARNING_INTERVAL
     url = url or setting.DINGDING_WARNING_URL
     user_phone = user_phone or setting.DINGDING_WARNING_PHONE
+    secret = secret or setting.DINGDING_WARNING_SECRET
+    if secret:
+        timestamp = str(round(time.time() * 1000))
+        secret_enc = secret.encode('utf-8')
+        string_to_sign_enc = f'{timestamp}\n{secret}'.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"{url}&timestamp={timestamp}&sign={sign}"
 
     if not all([url, message]):
         return