Update pages/monitor_playback.py

pages/monitor_playback.py

@@ -11,7 +11,7 @@ from collections import deque
 from dotenv import load_dotenv
 
 # --- 0. 页面配置 ---
-st.set_page_config(page_title="影城空场防空转监控 Pro", page_icon="🔋", layout="wide")
+st.set_page_config(page_title="影城空场防空转监控", page_icon="💸", layout="wide")
 
 try:
     from streamlit_autorefresh import st_autorefresh
@@ -21,12 +21,12 @@ except ImportError:
 # --- 1. 配置与常量 ---
 load_dotenv()
 
-PUSHOVER_TOKEN = os.getenv("PUSHOVER_TOKEN", "")
-PUSHOVER_USER = os.getenv("PUSHOVER_USER", "")
+PUSHOVER_TOKEN = os.getenv("PUSHOVER_TOKEN")
+PUSHOVER_USER = os.getenv("PUSHOVER_USER")
 # 使用 NTFY 配置
-NTFY_TOPIC = os.getenv("ntfy_code", "gzhdhdtz")
+NTFY_TOPIC = os.getenv("ntfy_code")
 
-CINEMA_ID = os.getenv("CINEMA_ID", "44001291")
+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")
@@ -46,6 +46,7 @@ TOKEN_FILE = os.path.join(ROOT_DIR, 'token_data.json')
 
 class DailyStats:
     """每日运行数据统计"""
+
     def __init__(self):
         self.reset()
         self.yesterday_stats = {
@@ -57,10 +58,10 @@ class DailyStats:
         }
 
     def reset(self):
-        self.zero_sessions = 0    # 0票总场次
-        self.triggers = 0         # 触发通知次数
-        self.notify_fails = 0     # 发送通知失败次数
-        self.api_fails = 0        # API获取失败次数
+        self.zero_sessions = 0  # 0票总场次
+        self.triggers = 0  # 触发通知次数
+        self.notify_fails = 0  # 发送通知失败次数
+        self.api_fails = 0  # API获取失败次数
 
     def snapshot_as_yesterday(self, date_str):
         self.yesterday_stats = {
@@ -72,6 +73,7 @@ class DailyStats:
         }
         self.reset()
 
+
 # 全局统计实例
 daily_stats = DailyStats()
 
@@ -80,12 +82,14 @@ def get_beijing_now():
     utc_now = datetime.now(timezone.utc)
     return utc_now.astimezone(timezone(timedelta(hours=8))).replace(tzinfo=None)
 
+
 def get_business_date():
     now = get_beijing_now()
     if now.time() < dt_time(6, 0):
         return (now - timedelta(days=1)).strftime("%Y-%m-%d")
     return now.strftime("%Y-%m-%d")
 
+
 def parse_show_datetime(date_str, time_str):
     try:
         show_t = datetime.strptime(time_str, "%H:%M").time()
@@ -96,11 +100,13 @@ def parse_show_datetime(date_str, time_str):
     except:
         return None
 
+
 def extract_hall_number_raw(hall_name):
     """仅提取数字ID，用于API映射"""
     match = re.search(r'(\d+)', str(hall_name))
     return match.group(1) if match else str(hall_name)
 
+
 def format_hall_name(hall_name):
     """格式化影厅名称：【和成天下1号厅】 -> 1号厅"""
     match = re.search(r'(\d+)号', str(hall_name))
@@ -154,7 +160,7 @@ class NotificationManager:
                         daily_stats.notify_fails += 1
 
                 self.ntfy_queue.task_done()
-                
+
             except Exception as e:
                 print(f"❌ NTFY 线程异常: {e}")
                 time.sleep(5)
@@ -168,7 +174,7 @@ class NotificationManager:
     def send_daily_report(self, today_date):
         """发送每日统计报告"""
         y_stats = daily_stats.yesterday_stats
-        
+
         title = "🟢 影城“空转”检查服务就绪"
         msg = (
             f"{today_date}，今日排片数据已加载，开始智能检查。\n\n"
@@ -177,9 +183,8 @@ class NotificationManager:
             f"触发通知次数：{y_stats['triggers']}\n"
             f"发送通知失败次数：{y_stats['notify_fails']}\n"
             f"API获取失败次数：{y_stats['api_fails']}\n"
-            f"（包括排片数据API和获取TMS数据API）"
         )
-        
+
         self.send_pushover(title, msg, priority=0)
         self.ntfy_queue.put((title, msg, 0))
 
@@ -198,7 +203,8 @@ class TicketAPIManager:
             try:
                 with open(TOKEN_FILE, 'r', encoding='utf-8') as f:
                     return json.load(f).get('token')
-            except: pass
+            except:
+                pass
         return None
 
     def login(self):
@@ -216,7 +222,8 @@ class TicketAPIManager:
         try:
             session = requests.Session()
             login_url = 'https://app.bi.piao51.cn/cinema-app/credential/login.action'
-            login_data = {'username': username, 'password': password, 'type': '1', 'resCode': res_code, 'deviceid': device_id, 'dtype': 'ios'}
+            login_data = {'username': username, 'password': password, 'type': '1', 'resCode': res_code,
+                          'deviceid': device_id, 'dtype': 'ios'}
             session.post(login_url, data=login_data, timeout=15)
             resp = session.get('https://app.bi.piao51.cn/cinema-app/security/logined.action', timeout=10)
             info = resp.json()
@@ -234,7 +241,7 @@ class TicketAPIManager:
         token = self.load_token()
         if not token:
             token = self.login()
-            if not token: 
+            if not token:
                 daily_stats.api_fails += 1
                 return None
 
@@ -268,7 +275,7 @@ class TMSAPIManager:
     def get_token(self):
         if self.auth_token and (time.time() - self.last_token_time < 1800):
             return self.auth_token
-        
+
         if not all([TMS_APP_SECRET, TMS_TICKET]): return None
 
         url = f'http://oa.hengdianfilm.com:7080/cinema-api/admin/generateToken?token=hd&murl=?token=hd&murl=ticket={TMS_TICKET}'
@@ -282,7 +289,8 @@ class TMSAPIManager:
                 self.auth_token = data['param']
                 self.last_token_time = time.time()
                 return self.auth_token
-        except: pass
+        except:
+            pass
         daily_stats.api_fails += 1
         return None
 
@@ -296,7 +304,7 @@ class TMSAPIManager:
         url = 'https://tms.hengdianfilm.com/cinema-api/cinema/schedule/server/list'
         session_id = TMS_X_SESSION_ID or ''
         cookies = {'JSESSIONID': session_id}
-        
+
         headers = {
             'accept': 'application/json, text/javascript, */*; q=0.01',
             'content-type': 'application/json; charset=UTF-8',
@@ -318,7 +326,8 @@ class TMSAPIManager:
         }
 
         try:
-            res = requests.post(url, params=params, cookies=cookies, headers=headers, json=json_data, timeout=10, verify=False)
+            res = requests.post(url, params=params, cookies=cookies, headers=headers, json=json_data, timeout=10,
+                                verify=False)
             data = res.json()
             if data.get("RSPCD") == "000000":
                 return data.get("BODY", {}).get("LIST", [])
@@ -337,16 +346,16 @@ class PlaybackMonitor:
         self.status_text = "初始化中..."
         self.next_wakeup_str = "--:--:--"
         self.active_monitors = []
-        
+
         self.cleared_sessions = set()
         self.processed_checks = set()
 
         self.ticket_api = TicketAPIManager(self.log)
         self.tms_api = TMSAPIManager(self.log)
-        
+
         self.current_business_date = None
         self.daily_schedule_cache = None
-        
+
         self.thread = threading.Thread(target=self._run_loop, daemon=True)
         self.thread.start()
 
@@ -365,32 +374,32 @@ class PlaybackMonitor:
                 zero_count += 1
                 start_str = item.get('showStartTime')
                 if not start_str: continue
-                
+
                 show_dt = parse_show_datetime(business_date, start_str)
                 if not show_dt: continue
-                
-                for i in range(3): 
+
+                for i in range(3):
                     check_time = show_dt + timedelta(minutes=i * 5)
                     check_points.append({
-                        'time': check_time, 
-                        'type': 'CHECK', 
-                        'check_index': i, 
+                        'time': check_time,
+                        'type': 'CHECK',
+                        'check_index': i,
                         'data': item
                     })
-        
+
         daily_stats.zero_sessions = zero_count
-        
+
         check_points.sort(key=lambda x: x['time'])
         return check_points
 
     def _run_loop(self):
-        self.log("🚀 智能防空转监控已启动 (09:00唤醒修复版)")
-        
+        self.log("🚀 智能防空转监控已启动")
+
         while True:
             try:
                 now = get_beijing_now()
                 biz_date = get_business_date()
-                
+
                 # --- 初始化与日期变更 ---
                 need_refresh = False
                 if self.daily_schedule_cache is None:
@@ -400,7 +409,7 @@ class PlaybackMonitor:
                 elif self.current_business_date != biz_date:
                     if now.time() >= dt_time(9, 0):
                         need_refresh = True
-                        
+
                         daily_stats.snapshot_as_yesterday(self.current_business_date)
                         self.cleared_sessions.clear()
                         self.processed_checks.clear()
@@ -412,13 +421,13 @@ class PlaybackMonitor:
                         self.daily_schedule_cache = schedule
                         self.current_business_date = biz_date
                         self.log(f"✅ 排片已更新，共 {len(schedule)} 场。")
-                        
+
                         try:
                             d_obj = datetime.strptime(biz_date, "%Y-%m-%d")
                             date_cn = f"{d_obj.year}年{d_obj.month}月{d_obj.day}日"
                         except:
                             date_cn = biz_date
-                        
+
                         notifier.send_daily_report(date_cn)
                     else:
                         self.log("⚠️ 获取排片失败，5分钟后重试")
@@ -428,59 +437,60 @@ class PlaybackMonitor:
 
                 # --- 任务计算 ---
                 check_points = self._get_target_check_points(self.daily_schedule_cache, biz_date)
-                
+
                 next_target = None
                 for cp in check_points:
                     hall_id = extract_hall_number_raw(cp['data']['hallName'])
                     start_str = cp['data']['showStartTime']
                     check_idx = cp['check_index']
                     dedup_key = f"{biz_date}_{hall_id}_{start_str}_{check_idx}"
-                    
+
                     if dedup_key in self.processed_checks:
                         continue
-                        
+
                     if cp['time'] > (now - timedelta(seconds=30)):
                         next_target = cp
                         break
-                
+
                 if next_target:
                     target_time = next_target['time']
-                    
+
                     # === 核心修复：09:00 拦截器 ===
                     # 如果当前时间早于 09:00，且下一场电影晚于 09:00，
                     # 必须在 09:00 醒来处理新的一天初始化，不能一觉睡过头。
                     nine_am_today = datetime.combine(now.date(), dt_time(9, 0))
-                    
+
                     force_wake_for_daily_reset = False
                     if now < nine_am_today and target_time > nine_am_today:
                         target_time = nine_am_today
                         force_wake_for_daily_reset = True
-                        
+
                     sleep_seconds = max(0, (target_time - now).total_seconds())
-                    
-                    self.status_text = f"💤 休眠中 (等待: {target_time.strftime('%H:%M')})"
+
+                    self.status_text = f"💤 休眠中)"
                     self.next_wakeup_str = target_time.strftime('%H:%M:%S')
-                    
+
                     if force_wake_for_daily_reset:
                         self.active_monitors = ["等待 09:00 日报刷新..."]
                     else:
                         raw_hall = next_target['data']['hallName']
                         clean_hall = format_hall_name(raw_hall)
-                        self.active_monitors = [f"下个任务: {clean_hall} {next_target['data']['showStartTime']} (第{next_target['check_index']+1}次检查)"]
-                    
+                        self.active_monitors = [
+                            f"下个任务: {clean_hall} {next_target['data']['showStartTime']} (第{next_target['check_index'] + 1}次检查)"]
+
                     self.log(f"💤 智能休眠 {int(sleep_seconds)}秒，将在 {self.next_wakeup_str} 唤醒...")
-                    
+
                     time.sleep(sleep_seconds)
-                    
+
                     # 如果是被强制唤醒来做日报刷新的，直接跳过后续的检查逻辑，
                     # continue 会回到循环头部，触发 if now >= 9:00 的初始化逻辑。
                     if force_wake_for_daily_reset:
                         continue
-                    
+
                     # --- 正常唤醒检查 ---
                     self.status_text = "🔥 正在执行检查..."
                     self.log("⏰ 唤醒！正在同步最新票务数据...")
-                    
+
                     latest_schedule = self.ticket_api.fetch_schedule(biz_date)
                     if latest_schedule:
                         self.daily_schedule_cache = latest_schedule
@@ -492,23 +502,24 @@ class PlaybackMonitor:
                     check_now = get_beijing_now()
                     current_targets = []
                     latest_check_points = self._get_target_check_points(latest_schedule, biz_date)
-                    
+
                     for cp in latest_check_points:
                         time_diff = abs((cp['time'] - check_now).total_seconds())
-                        if time_diff < 120: 
+                        if time_diff < 120:
                             current_targets.append(cp)
 
                     if current_targets:
                         self._process_targets(current_targets, biz_date)
-                    
+
                     time.sleep(5)
-                
+
                 else:
                     self.status_text = "🌙 今日监控结束"
                     self.active_monitors = []
-                    tmr_9am = datetime.combine(datetime.strptime(biz_date, "%Y-%m-%d").date() + timedelta(days=1), dt_time(9, 0))
+                    tmr_9am = datetime.combine(datetime.strptime(biz_date, "%Y-%m-%d").date() + timedelta(days=1),
+                                               dt_time(9, 0))
                     seconds_to_tmr = (tmr_9am - now).total_seconds()
-                    
+
                     if seconds_to_tmr > 3600:
                         self.log("暂无目标，休眠 1 小时...")
                         time.sleep(3600)
@@ -526,11 +537,11 @@ class PlaybackMonitor:
             hall_id = extract_hall_number_raw(t['data']['hallName'])
             start_str = t['data']['showStartTime']
             check_idx = t['check_index']
-            
+
             dedup_key = f"{biz_date}_{hall_id}_{start_str}_{check_idx}"
             if dedup_key in self.processed_checks:
                 continue
-                
+
             if hall_id not in halls_to_check:
                 halls_to_check[hall_id] = []
             halls_to_check[hall_id].append(t)
@@ -539,9 +550,9 @@ class PlaybackMonitor:
 
         self.log(f"🔍 核查 {len(halls_to_check)} 个影厅 TMS 状态...")
         display_results = []
-        
+
         for hall_id, target_list in halls_to_check.items():
-            
+
             all_cleared = True
             for target in target_list:
                 start_str = target['data'].get('showStartTime')
@@ -549,37 +560,37 @@ class PlaybackMonitor:
                 if key not in self.cleared_sessions:
                     all_cleared = False
                     break
-            
+
             if all_cleared:
                 for target in target_list:
                     dedup_key = f"{biz_date}_{hall_id}_{target['data']['showStartTime']}_{target['check_index']}"
                     self.processed_checks.add(dedup_key)
-                
+
                 clean_name = format_hall_name(target_list[0]['data']['hallName'])
                 display_results.append(f"{clean_name} ✅ [已跳过 (已确认撤场)]")
                 continue
 
             tms_list = self.tms_api.fetch_hall_schedule_list(hall_id)
-            
+
             for target in target_list:
                 session = target['data']
                 check_idx = target['check_index']
                 hall_name_raw = session.get('hallName')
                 movie_name = session.get('movieName')
-                ticket_start_str = session.get('showStartTime') 
+                ticket_start_str = session.get('showStartTime')
                 ticket_start_dt = parse_show_datetime(biz_date, ticket_start_str)
-                
+
                 unique_key = f"{biz_date}_{hall_id}_{ticket_start_str}"
                 dedup_key = f"{biz_date}_{hall_id}_{ticket_start_str}_{check_idx}"
-                
+
                 if dedup_key in self.processed_checks: continue
                 if unique_key in self.cleared_sessions:
                     self.processed_checks.add(dedup_key)
                     continue
 
                 hall_name_clean = format_hall_name(hall_name_raw)
-                status_desc = f"{hall_name_clean} {ticket_start_str} (第{check_idx+1}/3次)"
-                
+                status_desc = f"{hall_name_clean} {ticket_start_str} (第{check_idx + 1}/3次)"
+
                 if tms_list is not None:
                     match_found = False
                     for tms_item in tms_list:
@@ -590,16 +601,16 @@ class PlaybackMonitor:
                         if abs((t_start_dt - ticket_start_dt).total_seconds()) < 1800:
                             match_found = True
                             break
-                    
+
                     if match_found:
                         status_desc += " ⚠️ [空转! TMS未撤]"
                         self.log(f"🚨 发现空转: {hall_name_clean}《{movie_name}》")
-                        
-                        title = f"💸 发现{hall_name_clean}空场空转！"
+
+                        title = f"💸 发现 {hall_name_clean} 空场“空转”！"
                         msg = (
                             f"{hall_name_clean} {ticket_start_str}《{movie_name}》\n"
                             f"无人购票但服务器有排程，未撤场，或许正在播放，请检查。\n"
-                            f"第 {check_idx+1}/3 次检查发现，仅停止播放未撤排程下次检查依然会触发通知。"
+                            f"第 {check_idx + 1}/3 次检查发现，仅停止播放未撤排程下次检查依然会触发通知。"
                         )
                         notifier.send_alert(title, msg, priority=1)
                     else:
@@ -608,7 +619,7 @@ class PlaybackMonitor:
                 else:
                     status_desc += " ❓ [TMS查询失败]"
                     daily_stats.api_fails += 1
-                
+
                 self.processed_checks.add(dedup_key)
                 display_results.append(status_desc)
 
@@ -620,25 +631,25 @@ class PlaybackMonitor:
 def get_monitor():
     return PlaybackMonitor()
 
+
 def main():
     monitor = get_monitor()
-    
+
     if st_autorefresh:
         st_autorefresh(interval=30 * 1000, key="pb_refresh")
 
-    st.title("🔋 影厅空场防空转监控 Pro")
-    st.info("原理：基于票务开场时间，每 5 分钟查询 TMS 服务器排期列表。若 0 票场次仍在 TMS 列表中（未撤场），则视为耗电空转并报警。")
+    st.title("💸 影厅空场防空转监控")
 
     remaining_zero_count = 0
     remaining_total_count = 0
     if monitor.daily_schedule_cache:
         now = get_beijing_now()
         biz_date = monitor.current_business_date or get_business_date()
-        
+
         for item in monitor.daily_schedule_cache:
             start_str = item.get('showStartTime')
             if not start_str: continue
-            
+
             show_dt = parse_show_datetime(biz_date, start_str)
             if show_dt and show_dt > now:
                 remaining_total_count += 1
@@ -656,11 +667,11 @@ def main():
     st.divider()
 
     col_logs, col_mon = st.columns([3, 2])
-    
+
     with col_logs:
         st.subheader("📜 运行日志")
         st.text_area("Logs", "\n".join(list(monitor.logs)), height=450, disabled=True)
-    
+
     with col_mon:
         st.subheader("🕵️ 最近一次检查结果")
         if monitor.active_monitors:
@@ -674,5 +685,6 @@ def main():
         else:
             st.caption("等待下一次唤醒...")
 
+
 if __name__ == "__main__":
     main()