Update pages/monitor.py

pages/monitor.py

@@ -5,13 +5,17 @@ import json
 import os
 import threading
 import re
+import urllib3
 from datetime import datetime, timedelta, timezone, time as dt_time
 from collections import deque
 from dotenv import load_dotenv
 
-# --- 0. 必须最先执行 ---
+# --- 0. 基础配置 ---
 st.set_page_config(page_title="影城防撤场监控系统", page_icon="🛡️", layout="wide")
 
+# 屏蔽 HTTPS 证书警告
+urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
+
 # 尝试导入自动刷新组件
 try:
     from streamlit_autorefresh import st_autorefresh
@@ -84,7 +88,7 @@ def simplify_hall_name(raw_name):
     return raw_name
 
 
-# --- 3. 微信推送模块 ---
+# --- 3. 微信推送模块 (含自动重试) ---
 class WeChatPusher:
     def __init__(self, app_id, app_secret):
         self.app_id = app_id
@@ -92,9 +96,11 @@ class WeChatPusher:
         self.token = None
         self.token_expires_time = 0
 
-    def get_access_token(self):
-        """获取并缓存微信 Access Token"""
-        if self.token and datetime.now().timestamp() < self.token_expires_time:
+    def get_access_token(self, force_refresh=False):
+        """获取并缓存微信 Access Token
+        :param force_refresh: 是否强制刷新 Token
+        """
+        if not force_refresh and self.token and datetime.now().timestamp() < self.token_expires_time:
             return self.token
 
         if not self.app_id or not self.app_secret:
@@ -116,42 +122,64 @@ class WeChatPusher:
         return None
 
     def send_template_message(self, user_ids, template_id, data_map):
-        """发送模板消息给多个用户"""
-        token = self.get_access_token()
-        if not token:
-            return False
-
-        url = f"https://api.weixin.qq.com/cgi-bin/message/template/send?access_token={token}"
+        """发送模板消息给多个用户 (带 40001 错误重试机制)"""
         
         # 组装微信数据格式
         formatted_data = {}
         for key, value in data_map.items():
-            # 默认深蓝色
             formatted_data[key] = {"value": value, "color": "#173177"}
 
         # 清洗 user_ids
         valid_ids = [uid.strip() for uid in user_ids if uid and uid.strip()]
-        success_flag = True
+        
+        if not valid_ids:
+            return False
+
+        success_any = False
 
         for user_id in valid_ids:
-            payload = {
-                "touser": user_id,
-                "template_id": template_id,
-                "data": formatted_data
-            }
-            try:
-                resp = requests.post(url, json=payload, timeout=10)
-                result = resp.json()
-                if result.get("errcode") != 0:
-                    print(f"❌ 微信发送失败 [{user_id}]: {result}")
-                    success_flag = False
-                else:
-                    print(f"✅ 微信发送成功 [{user_id}]")
-            except Exception as e:
-                print(f"❌ 微信发送异常 [{user_id}]: {e}")
-                success_flag = False
+            # 每个用户最多尝试 2 次（1次正常，1次刷新 Token 后重试）
+            for attempt in range(2):
+                # 如果是重试（attempt > 0），强制刷新 Token
+                token = self.get_access_token(force_refresh=(attempt > 0))
+                
+                if not token:
+                    break # Token 都拿不到，不用重试了
+
+                url = f"https://api.weixin.qq.com/cgi-bin/message/template/send?access_token={token}"
+                payload = {
+                    "touser": user_id,
+                    "template_id": template_id,
+                    "data": formatted_data
+                }
+                
+                try:
+                    resp = requests.post(url, json=payload, timeout=10)
+                    result = resp.json()
+                    errcode = result.get("errcode")
+
+                    if errcode == 0:
+                        # 发送成功
+                        # print(f"✅ 微信发送成功 [{user_id}]")
+                        success_any = True
+                        break # 跳出重试循环，处理下一个用户
+                    
+                    # 40001: Token 无效
+                    # 40014: Token 不合法
+                    # 42001: Token 过期
+                    elif errcode in [40001, 40014, 42001]:
+                        print(f"🔄 微信 Token 失效 ({errcode})，正在刷新重试 [{user_id}]...")
+                        continue # 进入下一次循环，强制刷新 Token
+
+                    else:
+                        print(f"❌ 微信发送失败 [{user_id}]: {result}")
+                        break # 其他错误，不重试
+
+                except Exception as e:
+                    print(f"❌ 微信发送异常 [{user_id}]: {e}")
+                    break
         
-        return success_flag
+        return success_any
 
 
 # --- 4. API 管理模块 ---
@@ -323,6 +351,7 @@ class CinemaMonitor:
                 "keyword4": f"{ticket_count} 张",
                 "remark": "\n请尽快处理。"
             }
+            # 如果所有用户发送都失败才算失败
             if not self.wx_pusher.send_template_message(WX_USER_OPEN_IDS, WX_TEMPLATE_ID_TICKET, data):
                 self.stats["notify_fails"] += 1