Update pages/monitor_playback.py

pages/monitor_playback.py

@@ -41,6 +41,7 @@ CINEMA_ID = os.getenv("CINEMA_ID")
 TMS_APP_SECRET = os.getenv("TMS_APP_SECRET")
 TMS_TICKET = os.getenv("TMS_TICKET")
 TMS_X_SESSION_ID = os.getenv("TMS_X_SESSION_ID")
+TMS_THEATER_ID = os.getenv("TMS_THEATER_ID") # 新增：影院ID
 
 HALL_ID_MAP = {
     "1": "79181753", "2": "87350725", "3": "93340931",
@@ -195,21 +196,16 @@ class WeChatPusher:
                     errcode = result.get("errcode")
 
                     if errcode == 0:
-                        # 发送成功
-                        # print(f"✅ 微信发送成功 [{user_id}]")
-                        break # 跳出重试循环，处理下一个用户
+                        break # 成功
                     
-                    # 40001: Token 无效
-                    # 40014: Token 不合法
-                    # 42001: Token 过期
                     elif errcode in [40001, 40014, 42001]:
                         print(f"🔄 微信 Token 失效 ({errcode})，正在刷新重试 [{user_id}]...")
-                        continue # 进入下一次循环，强制刷新 Token
+                        continue # 强制刷新 Token
 
                     else:
                         print(f"❌ 微信发送失败 [{user_id}]: {result}")
                         daily_stats.notify_fails += 1
-                        break # 其他错误（如模板ID错、OpenID错），不重试
+                        break # 其他错误，不重试
 
                 except Exception as e:
                     print(f"❌ 微信发送异常 [{user_id}]: {e}")
@@ -388,6 +384,7 @@ class TMSAPIManager:
 
         if not all([TMS_APP_SECRET, TMS_TICKET]): return None
 
+        # 获取 OA 系统 Token
         url = f'http://oa.hengdianfilm.com:7080/cinema-api/admin/generateToken?token=hd&murl=?token=hd&murl=ticket={TMS_TICKET}'
         headers = {'Content-Type': 'application/json', 'User-Agent': 'Mozilla/5.0'}
         payload = {'appId': 'hd', 'appSecret': TMS_APP_SECRET, 'timeStamp': int(time.time() * 1000)}
@@ -448,6 +445,57 @@ class TMSAPIManager:
             daily_stats.api_fails += 1
             return None
 
+    def trigger_schedule_refresh(self):
+        """触发 TMS 排期刷新"""
+        token = self.get_token()
+        if not token or not TMS_THEATER_ID: return
+
+        try:
+            # 动态生成时间戳
+            now_tm = time.localtime()
+            today_midnight = time.mktime((now_tm.tm_year, now_tm.tm_mon, now_tm.tm_mday, 0, 0, 0, 0, 0, 0))
+            start_date = int(today_midnight * 1000)
+            end_date = start_date + (24 * 60 * 60 * 1000)
+
+            url = 'https://tms.hengdianfilm.com/cinema-api/tms/cmd/show/schedule'
+            session_id = TMS_X_SESSION_ID or ''
+            
+            headers = {
+                'accept': 'application/json, text/javascript, */*; q=0.01',
+                'content-type': 'application/json; charset=UTF-8',
+                'origin': 'https://tms.hengdianfilm.com',
+                'referer': f'https://tms.hengdianfilm.com/hd/index?Scheduling&THEATER_ID={TMS_THEATER_ID}&DATE={start_date}&PAGE_CAPACITY=20&PAGE_INDEX=1',
+                'token': token, # 这里的 token 由 get_token() 动态获取
+                'user-agent': 'Mozilla/5.0',
+                'x-requested-with': 'XMLHttpRequest',
+                'x-sessionid': session_id
+            }
+            
+            params = {
+                'token': 'hd',
+                'murl': 'Scheduling',
+            }
+
+            cookies = {'JSESSIONID': session_id}
+            
+            json_data = {
+                'THEATER_LIST': [
+                    {
+                        'THEATER_ID': int(TMS_THEATER_ID),
+                        'START_DATE': start_date,
+                        'END_DATE': end_date,
+                    },
+                ],
+            }
+
+            # 发送请求，非阻塞，忽略证书错误
+            requests.post(url, params=params, headers=headers, cookies=cookies, json=json_data, timeout=5, verify=False)
+            self.logger("🔄 TMS 排期刷新指令已发送")
+
+        except Exception as e:
+            # 刷新失败不影响主程序
+            self.logger(f"⚠️ TMS 排期刷新失败: {e}")
+
 
 # --- 6. 智能监控主逻辑 ---
 class PlaybackMonitor:
@@ -716,6 +764,10 @@ class PlaybackMonitor:
                         remark = "仅停止播放未撤排程下次检查依然会触发通知。"
                         notifier.send_idle_alert(hall_name_clean, movie_name, ticket_start_str, count_info, remark)
                         
+                        # --- NEW: 第一次发现后，触发TMS刷新 ---
+                        if check_idx == 0:
+                            self.tms_api.trigger_schedule_refresh()
+
                     else:
                         status_desc += " ✅ [正常 (TMS无排期)]"
                         self.cleared_sessions.add(unique_key)