小K 发表于 2025-10-13 00:01:01

抖音直播无人值守全天候轮询录制工具2.0

前端代码仅展示部分。请在下方下载完整源码。
先看效果图:

MoonTV 抖音直播监控系统项目总结项目概述
本项目是一个抖音直播监控和录制系统,具有多直播间管理、自动轮询检查、直播录制等功能。前端使用Vue.js构建,后端使用Python Flask框架实现。核心功能1. 多直播间管理
支持同时监控多个直播间的在线状态
自动轮询检查直播间状态(默认60秒间隔,可自定义)
显示直播间详细信息(房间ID、主播名、在线人数等)
2. 直播录制功能
支持手动开始/停止录制
支持开播时自动录制(可选)
录制文件保存在本地
3. 播放器功能
支持FLV直播流播放
页面内嵌式播放器(非弹窗)
支持多个播放器同时播放
播放器默认静音,点击播放后取消静音
播放器标题显示为主播名或房间ID
4. 批量操作
支持多选直播间
批量开始/停止录制
批量暂停/恢复轮询
批量移除直播间
5. 历史记录
记录直播间轮询历史
显示主播名、直播间地址和时间信息
技术架构前端 (douyin-frontend)
框架:Vue.js 3
样式:Tailwind CSS
播放器:flv.js
构建工具:Vue CLI
后端 (douyin-backend)
框架:Python Flask
多线程:threading模块
HTTP请求:requests库
数据存储:JSON文件(saved_rooms.json, rooms_history.json)
主要文件结构
MoonTV-main/
├── douyin-frontend/
│   ├── src/
│   │   ├── App.vue (主应用组件)
│   │   ├── MultiRoomManager.vue (多直播间管理器)
│   │   └── assets/ (静态资源)
│   ├── public/
│   └── package.json
├── douyin-backend/
│   ├── app.py (主应用文件)
│   ├── saved_rooms.json (保存的直播间配置)
│   ├── rooms_history.json (轮询历史记录)
│   └── recordings/ (录制文件目录)
└── docs/
    └── PROJECT_SUMMARY.md (项目说明文档)
复制代码API接口多直播间管理接口
GET /api/multi-poll/status - 获取所有直播间状态
POST /api/multi-poll/add - 添加直播间
POST /api/multi-poll/remove - 移除直播间
POST /api/multi-poll/start-record - 开始录制
POST /api/multi-poll/stop-record - 停止录制
POST /api/multi-poll/pause - 暂停轮询
POST /api/multi-poll/resume - 恢复轮询
GET /api/multi-poll/history - 获取历史记录
重要功能实现细节1. 暂停功能
暂停不仅停止录制,还会停止轮询检查,确保完全暂停直播间监控。2. 播放器实现
使用flv.js库支持FLV直播流播放
页面内嵌式播放器,支持多个播放器同时播放
默认静音状态,点击播放后取消静音
播放器标题显示为主播名或房间ID
3. 数据持久化
直播间配置保存在saved_rooms.json
轮询历史记录保存在rooms_history.json
录制文件保存在recordings目录下
启动方式
打开CMD
CD到项目目录下后端服务
python app.py
复制代码前端服务
cd douyin-frontend
npm install# 首次运行需要安装依赖
npm run serve
复制代码项目特点
开箱即用,无需复杂配置
支持多直播间同时监控
自动录制功能
数据本地持久化存储
历史记录去重功能
支持手机端短链接解析
可获取直播间实时数据(如在线人数等)
使用场景
直播平台观众数据监控
网红经济数据分析系统
直播带货效果评估工具
多平台直播状态监控中心



后端:
from flask import Flask, request, jsonify
from flask_cors import CORS
import requests
import re
import time
import os
import subprocess
import threading
import json
import logging
from datetime import datetime
from functools import wraps

app = Flask(__name__)
CORS(app, resources={r"/*": {"origins": ["http://127.0.0.1:8080", "http://localhost:8080"]}}, supports_credentials=True)

# 配置日志
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__)

# 全局变量
recording_sessions = {}
recording_lock = threading.Lock()

# 新增:多直播间轮询管理
polling_sessions = {}
polling_lock = threading.Lock()

# 异常处理装饰器
def handle_exceptions(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
      try:
            return func(*args, **kwargs)
      except Exception as e:
            logger.error(f"函数 {func.__name__} 执行失败: {str(e)}", exc_info=True)
            return jsonify({
                'success': False,
                'message': f'服务器内部错误: {str(e)}'
            }), 500
    return wrapper

def get_real_stream_url(url, max_retries=3):
    """
    解析抖音直播链接,获取真实的直播流地址
    :param url: 抖音直播链接
    :param max_retries: 最大重试次数
    :return: 直播流地址或 None
    """
   
    # 存储捕获到的直播流地址的变量,放在循环外部以便在所有尝试结束后仍能访问
    captured_stream_urls = []
   
    for attempt in range(max_retries):
      try:
            from playwright.sync_api import sync_playwright
            
            with sync_playwright() as p:
                # 启动浏览器(无头模式)
                browser = p.chromium.launch(headless=True)
                context = browser.new_context(
                  user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
                  viewport={"width": 1920, "height": 1080}
                )
                page = context.new_page()
               
                # 创建一个事件,用于在捕获到直播流地址时通知主线程
                stream_captured_event = threading.Event()
               
                # 处理URL格式
                if not url.startswith("http"):
                  url = f"https://live.douyin.com/{url}"
                  logger.info(f"转换为完整URL: {url}")
               
                # 访问直播页面
                logger.info(f"[尝试{attempt + 1}] 开始访问页面: {url}")
                page.goto(url, timeout=30000, wait_until="domcontentloaded")
               
                # 定义在捕获到直播流地址时的处理函数
                def on_stream_captured(url):
                  logger.info(f"[尝试{attempt + 1}] 成功捕获到直播流地址: {url}")
                  if url not in captured_stream_urls:
                        captured_stream_urls.append(url)
                        logger.info(f"[尝试{attempt + 1}] 已保存直播流地址,当前共 {len(captured_stream_urls)} 个")
                        # 立即设置事件,通知主线程已捕获到直播流地址
                        stream_captured_event.set()
                        logger.info(f"[尝试{attempt + 1}] 已通知主线程捕获到直播流地址")
               
                # 添加网络请求监听函数
                def handle_response(response):
                  try:
                        response_url = response.url
                        if (response_url.endswith('.m3u8') or
                            response_url.endswith('.flv') or
                            ('.flv?' in response_url) or
                            ('.m3u8?' in response_url) or
                            ('douyincdn.com' in response_url and ('stream' in response_url or 'pull' in response_url)) or
                            ('video' in response.headers.get('content-type', '') and not response_url.endswith('.mp4'))):
                            on_stream_captured(response_url)
                  except Exception as e:
                        logger.warning(f"处理响应失败: {e}")
               
                page.on("response", handle_response)
               
                # 直接等待网络请求,最多等待10秒
                max_wait_time = 10
                logger.info(f"[尝试{attempt + 1}] 开始等待直播流地址捕获...")
               
                # 等待事件或超时
                for elapsed_time in range(1, max_wait_time + 1):
                  # 先检查是否已经捕获到直播流地址
                  if captured_stream_urls:
                        logger.info(f"[尝试{attempt + 1}] 检测到已捕获 {len(captured_stream_urls)} 个直播流地址")
                        context.close()
                        return captured_stream_urls# 返回第一个捕获到的地址
                        
                  # 等待事件通知
                  if stream_captured_event.wait(1):# 等待1秒
                        logger.info(f"[尝试{attempt + 1}] 在 {elapsed_time} 秒后收到直播流地址捕获通知")
                        context.close()
                        return captured_stream_urls# 返回第一个捕获到的地址
                     
                  # 每2秒输出一次等待日志
                  if elapsed_time % 2 == 0:
                        logger.info(f"[尝试{attempt + 1}] 等待网络请求中... ({elapsed_time}/{max_wait_time}秒)")
               
                # 等待结束后最后检查一次变量
                if captured_stream_urls:# 变量不为空
                  logger.info(f"[尝试{attempt + 1}] 等待结束后发现 {len(captured_stream_urls)} 个直播流地址")
                  context.close()
                  return captured_stream_urls
                else:
                  logger.warning(f"[尝试{attempt + 1}] 等待结束后仍未捕获到直播流地址")
               
                # 保存页面内容用于调试
                try:
                  with open('debug_page_content.html', 'w', encoding='utf-8') as f:
                        f.write(page.content())
                except Exception as e:
                  logger.warning(f"保存调试文件失败: {e}")
               
                # 最后一次检查是否捕获到直播流地址
                if captured_stream_urls:
                  logger.info(f"[尝试{attempt + 1}] 关闭浏览器前发现已捕获到直播流地址")
                  context.close()
                  return captured_stream_urls
               
                context.close()
                if attempt < max_retries - 1:
                  logger.info(f"第 {attempt + 1} 次尝试失败,准备第 {attempt + 2} 次尝试...")
                  time.sleep(2)# 重试前等待
                     
      except Exception as e:
            logger.error(f"解析直播流地址失败 (尝试 {attempt + 1}): {str(e)}")
            # 即使发生异常,也检查是否已经捕获到直播流地址
            if captured_stream_urls:
                logger.info(f"[尝试{attempt + 1}] 尽管发生异常,但已捕获到直播流地址")
                return captured_stream_urls
               
            if attempt < max_retries - 1:
                time.sleep(2)
            continue
   
    # 最后一次检查是否有捕获到的直播流地址
    if captured_stream_urls:
      logger.info(f"虽然所有 {max_retries} 次尝试报告失败,但已捕获到 {len(captured_stream_urls)} 个直播流地址")
      return captured_stream_urls
         
    logger.error(f"所有 {max_retries} 次尝试均失败,未能捕获到直播流地址")
    return None

def parse_viewer_count(text):
    """
    解析观看人数文本为数字
    例: "32人在线" -> 32, "1.2万人在看" -> 12000, "5000人在看" -> 5000
    """
    try:
      # 移除常见的文字,保留数字和单位
      clean_text = re.sub(r'[人在看观气线众]', '', text)
         
      # 查找数字和单位
      match = re.search(r'(\d+(?:\.\d+)?)\s*([万w])?', clean_text, re.IGNORECASE)
      if match:
            number = float(match.group(1))
            unit = match.group(2)
            
            # 如果有"万"或"w"单位,乘以10000
            if unit and unit.lower() in ['万', 'w']:
                number *= 10000
               
            return int(number)
    except Exception as e:
      logger.debug(f"解析观看人数失败: {e}")
   
    return 0

def get_live_room_info(url, max_retries=3):
    """
    获取直播间详细信息,包括在线人数
    :param url: 抖音直播链接
    :param max_retries: 最大重试次数
    :return: 包含在线人数等信息的字典
    """
   
    room_info = {
      'online_count': 0,
      'is_live': False,
      'stream_url': None,
      'room_title': '',
      'anchor_name': '',
      'room_id': '',
      'viewer_count_text': ''# 显示的观看人数文本(如"1.2万人在看")
    }
   
    for attempt in range(max_retries):
      try:
            from playwright.sync_api import sync_playwright
            
            with sync_playwright() as p:
                browser = p.chromium.launch(headless=True)
                context = browser.new_context(
                  user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
                  viewport={"width": 1920, "height": 1080}
                )
                page = context.new_page()
               
                # 存储捕获的数据
                captured_data = {
                  'stream_urls': [],
                  'api_responses': []
                }
               
                # 处理URL格式
                if not url.startswith("http"):
                  url = f"https://live.douyin.com/{url}"
                     
                logger.info(f"[尝试{attempt + 1}] 开始获取直播间信息: {url}")
               
                # 监听网络请求,捕获API响应
                def handle_response(response):
                  try:
                        response_url = response.url
                        
                        # 捕获直播流地址
                        if (response_url.endswith('.m3u8') or
                            response_url.endswith('.flv') or
                            ('.flv?' in response_url) or
                            ('.m3u8?' in response_url) or
                            ('douyincdn.com' in response_url and ('stream' in response_url or 'pull' in response_url))):
                            captured_data['stream_urls'].append(response_url)
                            logger.info(f"捕获到直播流: {response_url}")
                        
                        # 捕获包含直播间信息的API响应
                        if ('webcast/room/' in response_url or
                            'webcast/web/' in response_url or
                            '/api/live_data/' in response_url or
                            'room_id' in response_url):
                            try:
                              if response.status == 200:
                                    response_json = response.json()
                                    captured_data['api_responses'].append({
                                        'url': response_url,
                                        'data': response_json
                                    })
                                    logger.info(f"捕获到API响应: {response_url}")
                            except Exception as json_error:
                              logger.debug(f"API响应解析失败: {json_error}")
                                 
                  except Exception as e:
                        logger.debug(f"处理响应失败: {e}")
               
                page.on("response", handle_response)
               
                # 访问直播页面
                page.goto(url, timeout=30000, wait_until="domcontentloaded")
               
                # 等待页面加载并捕获网络请求
                time.sleep(5)
               
                # 尝试从页面元素获取信息
                try:
                  # 方法1: 通过页面元素获取在线人数 - 更精确的选择器
                  online_selectors = [
                        '',
                        '',
                        '',
                        '',
                        'span:has-text("在线观众")',
                        'span:has-text("观众")',
                        'div:has-text("在线观众")',
                        '.webcast-chatroom___content span'
                  ]
                     
                  viewer_text = ""
                  # 首先尝试找到"在线观众"相关的元素
                  for selector in online_selectors:
                        try:
                            elements = page.query_selector_all(selector)
                            for element in elements:
                              text = element.inner_text().strip()
                              # 更严格的匹配条件,只要包含"在线观众"或纯数字的
                              if ('在线观众' in text or '观众' in text) and any(c.isdigit() for c in text):
                                    # 提取"在线观众 · 32"这样的格式
                                    import re
                                    match = re.search(r'在线观众[\s·]*([\d,]+)', text)
                                    if match:
                                        viewer_text = f"{match.group(1)}人在线"
                                        logger.info(f"找到在线观众数: {viewer_text}")
                                        break
                                    # 或者提取"观众 32"这样的格式
                                    match = re.search(r'观众[\s·]*([\d,]+)', text)
                                    if match:
                                        viewer_text = f"{match.group(1)}人在线"
                                        logger.info(f"找到观众数: {viewer_text}")
                                        break
                            if viewer_text:
                              break
                        except Exception as e:
                            logger.debug(f"选择器 {selector} 解析失败: {e}")
                     
                  # 如果没找到,尝试从页面内容中提取"在线观众"信息
                  if not viewer_text:
                        page_content = page.content()
                        # 使用正则表达式精确匹配"在线观众 · 数字"格式
                        patterns = [
                            r'在线观众[\s·]*([\d,]+)',
                            r'观众[\s·]*([\d,]+)',
                            r'(\d+)\s*人在线',
                            r'(\d+)\s*观看'
                        ]
                        
                        for pattern in patterns:
                            matches = re.findall(pattern, page_content)
                            if matches:
                              # 取第一个匹配的数字
                              count_str = matches.replace(',', '')# 移除千分位逗号
                              try:
                                    count = int(count_str)
                                    viewer_text = f"{count}人在线"
                                    logger.info(f"通过正则表达式获取到观众数: {viewer_text}")
                                    break
                              except ValueError:
                                    continue
                     
                  # 解析人数文本为数字
                  if viewer_text:
                        room_info['viewer_count_text'] = viewer_text
                        online_count = parse_viewer_count(viewer_text)
                        room_info['online_count'] = online_count
                        
                except Exception as e:
                  logger.warning(f"从页面元素获取在线人数失败: {e}")
               
                # 方法2: 从API响应中提取信息
                for api_resp in captured_data['api_responses']:
                  try:
                        data = api_resp['data']
                        
                        # 抖音API响应结构可能包含以下字段
                        if 'data' in data:
                            room_data = data['data']
                           
                            # 在线人数
                            if 'user_count' in room_data:
                              room_info['online_count'] = max(room_info['online_count'], room_data['user_count'])
                            elif 'stats' in room_data and 'user_count' in room_data['stats']:
                              room_info['online_count'] = max(room_info['online_count'], room_data['stats']['user_count'])
                            elif 'room_view_stats' in room_data:
                              room_info['online_count'] = max(room_info['online_count'], room_data['room_view_stats'].get('display_long', 0))
                           
                            # 直播状态
                            if 'status' in room_data:
                              room_info['is_live'] = room_data['status'] == 2# 2通常表示正在直播
                           
                            # 房间标题
                            if 'title' in room_data:
                              room_info['room_title'] = room_data['title']
                           
                            # 主播名称
                            if 'owner' in room_data and 'nickname' in room_data['owner']:
                              room_info['anchor_name'] = room_data['owner']['nickname']
                           
                            # 房间ID
                            if 'id_str' in room_data:
                              room_info['room_id'] = room_data['id_str']
                                 
                  except Exception as e:
                        logger.debug(f"解析API响应失败: {e}")
               
                # 设置直播流地址
                if captured_data['stream_urls']:
                  room_info['stream_url'] = captured_data['stream_urls']
                  room_info['is_live'] = True
               
                # 如果没有从API获取到在线人数,尝试页面内容检测
                if room_info['online_count'] == 0 and not room_info['viewer_count_text']:
                  try:
                        page_content = page.content()
                        
                        # 使用更精确的正则表达式从页面内容中提取人数
                        patterns = [
                            r'在线观众[\s·]*([\d,]+)',# "在线观众 · 32"
                            r'观众[\s·]*([\d,]+)',      # "观众 32"
                            r'"user_count["\s]*:\s*(\d+)',
                            r'"viewer_count["\s]*:\s*(\d+)',
                        ]
                        
                        for pattern in patterns:
                            matches = re.findall(pattern, page_content, re.IGNORECASE)
                            if matches:
                              try:
                                    count_str = matches.replace(',', '')# 移除千分位逗号
                                    count = int(count_str)
                                    room_info['online_count'] = count
                                    room_info['viewer_count_text'] = f"{count}人在线"
                                    logger.info(f"通过正则表达式获取到人数: {room_info['online_count']}")
                                    break
                              except ValueError:
                                    continue
                                    
                  except Exception as e:
                        logger.warning(f"页面内容解析失败: {e}")
               
                context.close()
               
                # 如果获取到了有效信息就返回
                if room_info['online_count'] > 0 or room_info['stream_url'] or room_info['is_live']:
                  logger.info(f"成功获取直播间信息: 在线人数={room_info['online_count']}, 直播状态={room_info['is_live']}")
                  return room_info
                     
      except Exception as e:
            logger.error(f"获取直播间信息失败 (尝试 {attempt + 1}): {str(e)}")
            if attempt < max_retries - 1:
                time.sleep(2)
            continue
   
    logger.error(f"所有 {max_retries} 次尝试均失败,无法获取直播间信息")
    return room_info

@app.route('/')
@handle_exceptions
def home():
    return jsonify({
      'message': '抖音直播解析后端服务已启动',
      'api': ['/api/parse', '/api/room-info', '/api/monitor', '/api/record/start', '/api/record/stop', '/api/record/status']
    })

@app.route('/api/parse', methods=['POST'])
@handle_exceptions
def parse_live_stream():
    data = request.get_json()
    url = data.get('url')
   
    if not url:
      return jsonify({
            'success': False,
            'message': '无效的直播链接或主播ID'
      })
   
    # 处理不同格式的输入
    processed_url = url.strip()
    logger.info(f"收到解析请求,原始输入: {processed_url}")
   
    # 1. 检查是否是纯数字(主播ID)
    if re.match(r'^\d+$', processed_url):
      logger.info(f"检测到主播ID格式: {processed_url}")
      room_id = processed_url
      full_url = f"https://live.douyin.com/{room_id}"
   
    # 2. 检查是否是完整的抖音直播URL
    elif "douyin.com" in processed_url:
      logger.info(f"检测到抖音URL格式: {processed_url}")
      # 提取房间号
      if "/user/" in processed_url:
            # 用户主页URL
            logger.info("检测到用户主页URL,尝试提取用户ID")
            user_id_match = re.search(r'/user/([^/?]+)', processed_url)
            if user_id_match:
                room_id = user_id_match.group(1)
                full_url = f"https://live.douyin.com/{room_id}"
            else:
                return jsonify({
                  'success': False,
                  'message': '无法从用户主页URL提取用户ID'
                })
      else:
            # 直播间URL
            room_id_match = re.search(r'live\.douyin\.com/([^/?]+)', processed_url)
            if room_id_match:
                room_id = room_id_match.group(1)
                full_url = f"https://live.douyin.com/{room_id}"
            else:
                # 尝试直接使用
                room_id = processed_url
                full_url = processed_url
   
    # 3. 其他格式(可能是短链接或其他标识符)
    else:
      logger.info(f"未识别的URL格式,尝试直接使用: {processed_url}")
      room_id = processed_url
      full_url = processed_url
   
    logger.info(f"处理后的房间ID: {room_id}, 完整URL: {full_url}")
   
    # 调用解析函数获取直播流地址
    real_stream_url = get_real_stream_url(full_url)
   
    if real_stream_url:
      logger.info(f"成功解析直播流地址: {real_stream_url}")
      return jsonify({
            'success': True,
            'streamUrl': real_stream_url,
            'roomId': room_id,
            'fullUrl': full_url
      })
    else:
      logger.warning(f"无法解析直播流地址,输入: {processed_url}")
      return jsonify({
            'success': False,
            'message': '无法解析直播链接,请确认主播是否开播'
      })

# 新增:获取直播间详细信息的API接口
@app.route('/api/room-info', methods=['POST'])
@handle_exceptions
def get_room_info():
    """获取直播间详细信息,包括在线人数"""
    data = request.get_json()
    url = data.get('url')
   
    if not url:
      return jsonify({
            'success': False,
            'message': '无效的直播链接或主播 ID'
      })
   
    # 处理URL格式
    processed_url = url.strip()
    logger.info(f"收到直播间信息请求: {processed_url}")
   
    # URL格式处理逻辑(与parse_live_stream相同)
    if re.match(r'^\d+$', processed_url):
      full_url = f"https://live.douyin.com/{processed_url}"
    elif "douyin.com" in processed_url:
      full_url = processed_url
    else:
      full_url = processed_url
   
    # 获取直播间信息
    room_info = get_live_room_info(full_url)
   
    if room_info['is_live'] or room_info['online_count'] > 0:
      return jsonify({
            'success': True,
            'data': {
                'online_count': room_info['online_count'],
                'viewer_count_text': room_info['viewer_count_text'],
                'is_live': room_info['is_live'],
                'stream_url': room_info['stream_url'],
                'room_title': room_info['room_title'],
                'anchor_name': room_info['anchor_name'],
                'room_id': room_info['room_id']
            }
      })
    else:
      return jsonify({
            'success': False,
            'message': '直播间未开播或无法获取信息',
            'data': room_info
      })

def get_anchor_info(anchor_id, max_retries=2):
    """
    获取主播信息(名字、直播状态等)
    :param anchor_id: 主播ID
    :param max_retries: 最大重试次数
    :return: dict 包含 {"is_live": bool, "name": str, "title": str}
    """
    for attempt in range(max_retries):
      try:
            from playwright.sync_api import sync_playwright
            import random
            
            with sync_playwright() as p:
                # 启动浏览器(无头模式)
                browser = p.chromium.launch(headless=True)
                context = browser.new_context(
                  user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/125.0.0.0 Safari/537.36",
                  extra_http_headers={
                        "Referer": "https://www.douyin.com/",
                        "Accept-Language": "zh-CN,zh;q=0.9"
                  },
                  viewport={"width": 1920, "height": 1080},
                  java_script_enabled=True
                )
                page = context.new_page()
               
                # 随机延迟(1-3秒),模拟人类操作
                time.sleep(random.uniform(1, 3))
               
                # 访问直播间页面
                try:
                  # 处理URL格式,确保不重复添加域名
                  if anchor_id.startswith("https://live.douyin.com/"):
                        url = anchor_id
                        room_id = anchor_id.split("/")[-1]
                  else:
                        url = f"https://live.douyin.com/{anchor_id}"
                        room_id = anchor_id
                     
                  logger.info(f"[尝试{attempt + 1}] 开始访问直播间页面: {url}")
                  page.goto(url, timeout=30000, wait_until="domcontentloaded")
                  logger.info(f"[尝试{attempt + 1}] 成功访问直播间页面")
                except Exception as e:
                  if "Timeout" in str(e):
                        logger.warning(f"[尝试{attempt + 1}] 页面加载超时,继续处理")
                  else:
                        logger.error(f"[尝试{attempt + 1}] 访问直播间页面失败: {e}")
                        context.close()
                        continue
               
                # 等待页面加载
                try:
                  logger.info(f"[尝试{attempt + 1}] 等待页面关键元素加载...")
                  page.wait_for_selector("body", timeout=10000)
                  # 额外等待,确保页面完全加载
                  time.sleep(3)
                except Exception as wait_e:
                  logger.warning(f"[尝试{attempt + 1}] 等待元素失败: {wait_e},继续处理")
               
                # 获取页面内容
                content = page.content()
                logger.info(f"[尝试{attempt + 1}] 页面内容长度: {len(content)} 字符")
               
                # 提取主播信息
                anchor_info = {
                  "is_live": False,
                  "name": f"anchor_{room_id}",# 默认名字
                  "title": ""
                }
               
                # 尝试获取主播名字
                logger.info(f"[尝试{attempt + 1}] 开始尝试获取主播名字...")
               
                # 策略1: 尝试从页面标题获取(优先策略)
                try:
                  title = page.title()
                  logger.info(f"[尝试{attempt + 1}] 页面标题: {title}")
                     
                  # 抖音直播间标题格式分析
                  if title and title != "抖音直播":
                        # 格式1: "主播名字的直播间"
                        if "的直播间" in title:
                            name_from_title = title.split("的直播间").strip()
                            if name_from_title and len(name_from_title) < 50 and name_from_title != room_id:
                              anchor_info["name"] = name_from_title
                              logger.info(f"[尝试{attempt + 1}] 从页面标题获取到主播名字: {name_from_title}")
                        # 格式2: "主播名字 - 抖音直播"
                        elif " - 抖音" in title or " - 直播" in title:
                            parts = title.split(" - ")
                            if len(parts) > 0:
                              potential_name = parts.strip()
                              if potential_name and len(potential_name) < 50 and potential_name != room_id:
                                    anchor_info["name"] = potential_name
                                    logger.info(f"[尝试{attempt + 1}] 从页面标题解析到主播名字: {potential_name}")
                        # 格式3: "主播名字正在直播"
                        elif "正在直播" in title:
                            name_from_title = title.replace("正在直播", "").strip()
                            if name_from_title and len(name_from_title) < 50 and name_from_title != room_id:
                              anchor_info["name"] = name_from_title
                              logger.info(f"[尝试{attempt + 1}] 从'正在直播'标题获取到主播名字: {name_from_title}")
                        # 格式4: 直接使用标题(如果长度合理)
                        elif len(title) < 50 and title != room_id and not any(word in title.lower() for word in ["douyin", "live", "直播"]):
                            anchor_info["name"] = title
                            logger.info(f"[尝试{attempt + 1}] 直接使用页面标题作为主播名字: {title}")
                except Exception as title_e:
                  logger.debug(f"[尝试{attempt + 1}] 从标题获取名字失败: {title_e}")
               
                # 策略2: 尝试从页面元素获取(如果标题没有找到合适的名字)
                if anchor_info["name"] == f"anchor_{room_id}":
                  try:
                        logger.info(f"[尝试{attempt + 1}] 尝试从页面元素获取主播名字...")
                        
                        # 更新的选择器列表
                        name_selectors = [
                            "",
                            "",
                            ".webcast-avatar-info__name",
                            ".live-user-info .name",
                            ".live-user-name",
                            ".user-name",
                            ".anchor-name",
                            "",
                            "h3",
                            ".nickname"
                        ]
                        
                        for selector in name_selectors:
                            try:
                              name_element = page.query_selector(selector)
                              if name_element:
                                    name_text = name_element.inner_text().strip()
                                    if name_text and len(name_text) < 50 and name_text != room_id and not name_text.isdigit():
                                        anchor_info["name"] = name_text
                                        logger.info(f"[尝试{attempt + 1}] 使用选择器 {selector} 获取到主播名字: {name_text}")
                                        break
                            except Exception as sel_e:
                              logger.debug(f"[尝试{attempt + 1}] 选择器 {selector} 失败: {sel_e}")
                              continue
                  except Exception as e:
                        logger.debug(f"[尝试{attempt + 1}] 从页面元素获取名字失败: {e}")
               
                # 策略3: 从页面JSON数据中提取(如果前面都没找到)
                if anchor_info["name"] == f"anchor_{room_id}":
                  try:
                        logger.info(f"[尝试{attempt + 1}] 尝试从页面JSON数据获取主播名字...")
                        
                        content_text = page.content()
                        
                        # 多种JSON字段模式
                        json_patterns = [
                            r'"nickname"\s*:\s*"([^"]+)"',
                            r'"displayName"\s*:\s*"([^"]+)"',
                            r'"userName"\s*:\s*"([^"]+)"',
                            r'"ownerName"\s*:\s*"([^"]+)"',
                            r'"anchorName"\s*:\s*"([^"]+)"',
                            r'"user_name"\s*:\s*"([^"]+)"',
                            r'"anchor_info"[^}]*"nickname"\s*:\s*"([^"]+)"'
                        ]
                        
                        import re as regex_re
                        for pattern in json_patterns:
                            matches = regex_re.findall(pattern, content_text)
                            for match in matches:
                              if match and len(match) < 50 and match != room_id and not match.isdigit():
                                    # 过滤掉明显不是名字的内容
                                    if not any(word in match.lower() for word in ['http', 'www', '.com', 'live', 'stream']):
                                        anchor_info["name"] = match
                                        logger.info(f"[尝试{attempt + 1}] 从页面JSON数据获取到主播名字: {match} (模式: {pattern})")
                                        break
                            if anchor_info["name"] != f"anchor_{room_id}":
                              break
                                 
                  except Exception as content_e:
                        logger.debug(f"[尝试{attempt + 1}] 从页面内容获取名字失败: {content_e}")
               
                # 策略4: 最后的降级处理(使用更友好的默认名字)
                if anchor_info["name"] == f"anchor_{room_id}":
                  # 尝试从room_id中提取可能的用户名部分
                  if len(room_id) > 8:# 如果room_id足够长,尝试截取前8位作为更简洁的标识
                        anchor_info["name"] = f"主播{room_id[:8]}"
                  else:
                        anchor_info["name"] = f"主播{room_id}"
                  logger.info(f"[尝试{attempt + 1}] 使用降级处理的默认名字: {anchor_info['name']}")
               
                # 检查直播状态
                stream_urls = []
                def handle_response(response):
                  url = response.url
                  if ((url.endswith('.flv') or url.endswith('.m3u8')) and
                        not url.endswith('.mp4') and
                        ('pull-' in url or 'douyincdn.com' in url)):
                        stream_urls.append(url)
                        logger.info(f"[尝试{attempt + 1}] 捕获到直播流: {url}")
               
                page.on("response", handle_response)
               
                # 等待更多网络请求
                logger.info(f"[尝试{attempt + 1}] 等待网络请求...")
                time.sleep(3)
               
                # 多种方式检测直播状态
                anchor_info["is_live"] = (
                  "直播中" in content or
                  "正在直播" in content or
                  "live_no_stream" not in content.lower() and "直播" in content or
                  "live" in content.lower() or
                  page.query_selector(".webcast-chatroom___enter-done") is not None or
                  page.query_selector(".live-room") is not None or
                  page.query_selector("video") is not None or
                  page.query_selector("video") is not None or
                  page.query_selector("video") is not None or
                  len(stream_urls) > 0
                )
               
                context.close()
               
                logger.info(f"[尝试{attempt + 1}] 最终获取结果 - 主播名字: {anchor_info['name']}, 直播状态: {'在线' if anchor_info['is_live'] else '离线'}")
               
                return anchor_info
               
      except Exception as e:
            logger.error(f"获取主播信息失败 (尝试 {attempt + 1}): {str(e)}")
            if attempt < max_retries - 1:
                time.sleep(2)
            continue
   
    logger.error(f"所有 {max_retries} 次尝试均失败,返回默认结果")
    # 最终降级处理
    fallback_name = f"主播{anchor_id[:8]}" if len(str(anchor_id)) > 8 else f"主播{anchor_id}"
    return {"is_live": False, "name": fallback_name, "title": ""}

def check_anchor_status(anchor_id, max_retries=2):
    """
    检查主播是否开播
    :param anchor_id: 主播ID
    :param max_retries: 最大重试次数
    :return: True(开播)/False(未开播)
    """
    for attempt in range(max_retries):
      try:
            from playwright.sync_api import sync_playwright
            import random
            
            with sync_playwright() as p:
                # 启动浏览器(无头模式)
                browser = p.chromium.launch(headless=True)
                context = browser.new_context(
                  user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
                  extra_http_headers={
                        "Referer": "https://www.douyin.com/",
                        "Accept-Language": "zh-CN,zh;q=0.9"
                  },
                  viewport={"width": 1920, "height": 1080},
                  java_script_enabled=True
                )
                page = context.new_page()
               
                # 随机延迟(1-3秒),模拟人类操作
                time.sleep(random.uniform(1, 3))
               
                # 访问直播间页面
                try:
                  # 处理URL格式,确保不重复添加域名
                  if anchor_id.startswith("https://live.douyin.com/"):
                        url = anchor_id
                        room_id = anchor_id.split("/")[-1]
                  else:
                        url = f"https://live.douyin.com/{anchor_id}"
                        room_id = anchor_id
                     
                  page.goto(url, timeout=30000, wait_until="domcontentloaded")
                  logger.info(f"成功访问直播间页面: {url}")
                except Exception as e:
                  if "Timeout" in str(e):
                        logger.warning(f"页面加载超时,继续处理")
                  else:
                        logger.error(f"访问直播间页面失败: {e}")
                        context.close()
                        continue
               
                # 等待页面加载
                try:
                  page.wait_for_selector("video, .live-room, .webcast-chatroom", timeout=10000)
                except:
                  logger.warning("未找到关键元素,继续处理")
               
                # 获取页面内容
                content = page.content()
               
                # 检查直播状态
                stream_urls = []
                def handle_response(response):
                  url = response.url
                  if ((url.endswith('.flv') or url.endswith('.m3u8')) and
                        not url.endswith('.mp4') and
                        ('pull-' in url or 'douyincdn.com' in url)):
                        stream_urls.append(url)
                        logger.info(f"捕获到直播流: {url}")
               
                page.on("response", handle_response)
               
                # 等待更多网络请求
                time.sleep(3)
               
                # 多种方式检测直播状态
                is_live = (
                  "直播中" in content or
                  "正在直播" in content or
                  "直播" in content or
                  "live" in content.lower() or
                  page.query_selector(".webcast-chatroom___enter-done") is not None or
                  page.query_selector(".live-room") is not None or
                  page.query_selector("video") is not None or
                  page.query_selector("video") is not None or
                  page.query_selector("video") is not None or
                  len(stream_urls) > 0 or
                  any("live.douyin.com" in url for url in stream_urls)
                )
               
                context.close()
               
                if is_live:
                  logger.info(f"主播 {anchor_id} 正在直播")
                  if stream_urls:
                        logger.info(f"捕获到直播流地址: {stream_urls}")
                else:
                  logger.info(f"主播 {anchor_id} 未开播")
               
                return is_live
               
      except Exception as e:
            logger.error(f"检查主播状态失败 (尝试 {attempt + 1}): {str(e)}")
            if attempt < max_retries - 1:
                time.sleep(2)
            continue
   
    logger.error(f"所有 {max_retries} 次尝试均失败")
    return False

@app.route('/api/monitor', methods=['POST'])
@handle_exceptions
def monitor_live_stream():
    data = request.get_json()
    anchor_id = data.get('anchor_id')
    max_wait_minutes = data.get('max_wait', 5)# 默认最多等待5分钟
    check_interval = data.get('interval', 30)   # 默认每30秒检查一次
   
    logger.info(f"收到监控请求,主播ID: {anchor_id}, 最长等待: {max_wait_minutes}分钟, 轮询地址: https://live.douyin.com/{anchor_id}")
   
    if not anchor_id:
      logger.warning("无效的主播ID")
      return jsonify({
            'success': False,
            'message': '无效的主播ID'
      })
   
    max_checks = (max_wait_minutes * 60) // check_interval
    checks_done = 0
   
    # 轮询检查主播状态
    while checks_done < max_checks:
      checks_done += 1
      logger.info(f"第 {checks_done}/{max_checks} 次检查主播 {anchor_id} 状态")
         
      is_live = check_anchor_status(anchor_id)
      if is_live:
            logger.info(f"主播 {anchor_id} 正在直播,开始解析直播流地址")
            
            # 获取直播流地址
            stream_url = get_real_stream_url(f"https://live.douyin.com/{anchor_id}")
            if stream_url:
                logger.info(f"成功获取直播流地址: {stream_url}")
                return jsonify({
                  'success': True,
                  'status': 'live',
                  'streamUrl': stream_url,
                  'checks_performed': checks_done
                })
            else:
                logger.warning("无法解析直播流地址")
                return jsonify({
                  'success': False,
                  'message': '无法解析直播流地址',
                  'checks_performed': checks_done
                })
      else:
            logger.info(f"主播 {anchor_id} 未开播,等待下一次检查")
            
            # 如果达到最大检查次数,返回未开播状态
            if checks_done >= max_checks:
                logger.info(f"监控超时,主播 {anchor_id} 在 {max_wait_minutes} 分钟内未开播")
                return jsonify({
                  'success': True,
                  'status': 'not_live',
                  'message': f'主播在 {max_wait_minutes} 分钟内未开播',
                  'checks_performed': checks_done
                })
            
            time.sleep(check_interval)
   
    logger.warning("监控循环异常结束")
    return jsonify({
      'success': False,
      'message': '监控异常结束',
      'checks_performed': checks_done
    })

class MultiRoomPoller:
    """多直播间轮询管理器"""
   
    def __init__(self):
      self.polling_rooms = {}# 存储轮询中的直播间
      self.polling_history = []# 存储历史轮询记录
      self.lock = threading.Lock()
      self.running = True
      self.max_history_records = 1000# 最大历史记录数
      self.rooms_file = 'saved_rooms.json'# 本地存储文件
      self.history_file = 'rooms_history.json'# 历史记录文件
         
      # 启动时加载已保存的直播间
      self._load_rooms_from_file()
      self._load_history_from_file()
   
    def add_room(self, room_id, room_url, check_interval=60, auto_record=False):
      """添加直播间到轮询列表"""
      with self.lock:
            if room_id not in self.polling_rooms:
                # 不再在这里添加历史记录,等待轮询线程获取到真实主播名字后再添加
               
                self.polling_rooms = {
                  'room_url': room_url,
                  'room_id': room_id,
                  'check_interval': check_interval,
                  'auto_record': auto_record,
                  'status': 'waiting',# waiting, checking, live, offline, paused
                  'last_check': None,
                  'stream_url': None,
                  'recording_session_id': None,
                  'thread': None,
                  'anchor_name': f'anchor_{room_id}',# 新增:主播名字
                  'live_title': '',# 新增:直播标题
                  'added_time': datetime.now(),# 新增:添加时间
                  'history_added': False,# 新增:标记是否已添加历史记录
                  'online_count': 0,# 新增:在线人数
                  'viewer_count_text': ''# 新增:观看人数文本
                }
               
                # 启动轮询线程
                thread = threading.Thread(
                  target=self._poll_room,
                  args=(room_id,),
                  daemon=True
                )
                thread.start()
                self.polling_rooms['thread'] = thread
               
                logger.info(f"已添加直播间 {room_id} 到轮询列表")
               
                # 保存到本地文件
                self._save_rooms_to_file()
               
                return True
            else:
                logger.warning(f"直播间 {room_id} 已在轮询列表中")
                return False
   
    def remove_room(self, room_id):
      """从轮询列表移除直播间"""
      with self.lock:
            if room_id in self.polling_rooms:
                room_info = self.polling_rooms
                # 记录到历史
                self._add_to_history(
                  room_id,
                  room_info['room_url'],
                  '',
                  '',
                  room_info.get('anchor_name', f'anchor_{room_id}')
                )
               
                # 停止录制(如果正在录制)
                if self.polling_rooms['recording_session_id']:
                  self._stop_recording(room_id)
               
                # 标记线程停止
                self.polling_rooms['status'] = 'stopped'
                del self.polling_rooms
               
                # 保存到本地文件
                self._save_rooms_to_file()
               
                logger.info(f"已从轮询列表移除直播间 {room_id}")
                return True
            return False
   
    def pause_room(self, room_id):
      """暂停指定直播间的轮询"""
      with self.lock:
            if room_id in self.polling_rooms:
                # 如果已经在暂停状态,返回False
                if self.polling_rooms['status'] == 'paused':
                  return False
               
                # 更新状态为暂停
                self.polling_rooms['status'] = 'paused'
                logger.info(f"已暂停直播间 {room_id} 的轮询")
                return True
            return False
   
    def resume_room(self, room_id):
      """恢复指定直播间的轮询"""
      with self.lock:
            if room_id in self.polling_rooms:
                # 如果不在暂停状态,返回False
                if self.polling_rooms['status'] != 'paused':
                  return False
               
                # 更新状态为等待
                self.polling_rooms['status'] = 'waiting'
                logger.info(f"已恢复直播间 {room_id} 的轮询")
                return True
            return False
   
    def _poll_room(self, room_id):
      """单个直播间轮询逻辑"""
      while self.running:
            try:
                with self.lock:
                  if room_id not in self.polling_rooms:
                        break
                     
                  room_info = self.polling_rooms
                  # 检查是否暂停
                  if room_info['status'] == 'paused':
                        # 如果暂停,等待一段时间后继续检查
                        time.sleep(5)
                        continue
                     
                  if room_info['status'] == 'stopped':
                        break
               
                # 更新状态为检查中
                with self.lock:
                  self.polling_rooms['status'] = 'checking'
                  self.polling_rooms['last_check'] = datetime.now()
               
                # 检查直播状态并获取主播信息
                anchor_info = get_anchor_info(room_info['room_id'])
                is_live = anchor_info['is_live']
               
                # 获取直播间详细信息(包括在线人数)
                room_detail_info = {'online_count': 0, 'viewer_count_text': ''}
                if is_live:
                  try:
                        # 调用get_live_room_info获取在线人数信息
                        room_detail_info = get_live_room_info(room_info['room_url'])
                        logger.info(f"直播间 {room_id} 在线人数: {room_detail_info.get('online_count', 0)}")
                  except Exception as e:
                        logger.warning(f"获取直播间 {room_id} 在线人数失败: {e}")
               
                # 更新主播信息和在线人数
                with self.lock:
                  self.polling_rooms['anchor_name'] = anchor_info['name']
                  self.polling_rooms['live_title'] = anchor_info['title']
                  self.polling_rooms['online_count'] = room_detail_info.get('online_count', 0)
                  self.polling_rooms['viewer_count_text'] = room_detail_info.get('viewer_count_text', '')
                     
                  # 如果还没有添加历史记录,现在添加一条记录
                  if not self.polling_rooms.get('history_added', False):
                        self._add_to_history(
                            room_id,
                            room_info['room_url'],
                            '',
                            '',
                            anchor_info['name']
                        )
                        self.polling_rooms['history_added'] = True
               
                if is_live:
                  logger.info(f"检测到直播间 {room_id} 正在直播")
                     
                  # 记录状态变化到历史(如果之前不是直播状态)
                  # 简化版:不记录状态变化
                     
                  # 解析直播流地址
                  stream_url = get_real_stream_url(room_info['room_url'])
                     
                  if stream_url:
                        with self.lock:
                            self.polling_rooms['status'] = 'live'
                            self.polling_rooms['stream_url'] = stream_url
                        
                        # 如果启用自动录制且未在录制
                        if (room_info['auto_record'] and
                            not room_info['recording_session_id']):
                            self._start_recording(room_id, stream_url)
                            # 简化版:不记录自动录制开始
                  else:
                        logger.warning(f"直播间 {room_id} 在线但无法获取流地址")
                        with self.lock:
                            old_status = self.polling_rooms['status']
                            self.polling_rooms['status'] = 'live_no_stream'
                            # 简化版:不记录状态变化
                        
                        # 如果之前在录制,停止录制(直播结束无流)
                        if room_info['recording_session_id']:
                            self._stop_recording(room_id)
                            logger.info(f"直播间 {room_id} 直播结束无流,已停止录制")
                            # 简化版:不记录停止录制
                else:
                  # 直播间离线
                  with self.lock:
                        old_status = self.polling_rooms['status']
                        self.polling_rooms['status'] = 'offline'
                        self.polling_rooms['stream_url'] = None
                        
                        # 简化版:不记录状态变化
                     
                  # 如果之前在录制,停止录制
                  if room_info['recording_session_id']:
                        self._stop_recording(room_id)
                        logger.info(f"直播间 {room_id} 离线,已停止录制")
                        # 简化版:不记录停止录制
               
                # 等待下次检查
                time.sleep(room_info['check_interval'])
               
            except Exception as e:
                logger.error(f"轮询直播间 {room_id} 异常: {str(e)}")
                with self.lock:
                  if room_id in self.polling_rooms:
                        self.polling_rooms['status'] = 'error'
                time.sleep(30)# 出错时等待30秒后重试
   
    def _start_recording(self, room_id, stream_url):
      """启动录制"""
      try:
            # 获取主播名字用于文件命名
            with self.lock:
                anchor_name = self.polling_rooms.get('anchor_name', f'anchor_{room_id}')
            
            # 清理文件名中的非法字符
            safe_anchor_name = re.sub(r'[<>:"/\|?*]', '_', anchor_name)
            
            session_id = f"auto_record_{room_id}_{int(time.time())}"
            timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
            # 使用主播名字命名文件
            output_path = f"recordings/{safe_anchor_name}_{timestamp}.mp4"
            
            # 启动录制线程
            thread = threading.Thread(
                target=record_stream,
                args=(stream_url, output_path, session_id),
                daemon=True
            )
            thread.start()
            
            with self.lock:
                self.polling_rooms['recording_session_id'] = session_id
            
            logger.info(f"已为主播 {anchor_name} (房间 {room_id}) 启动自动录制,会话ID: {session_id},文件: {output_path}")
            
      except Exception as e:
            logger.error(f"启动直播间 {room_id} 录制失败: {str(e)}")
   
    def _stop_recording(self, room_id):
      """停止录制"""
      try:
            with self.lock:
                session_id = self.polling_rooms['recording_session_id']
                if session_id:
                  self.polling_rooms['recording_session_id'] = None
            
            if session_id:
                # 停止录制会话
                with recording_lock:
                  if session_id in recording_sessions:
                        session = recording_sessions
                        if session['process']:
                            session['process'].terminate()
                        session['status'] = 'stopped'
                        session['end_time'] = datetime.now()
               
                logger.info(f"已停止直播间 {room_id} 的录制,会话ID: {session_id}")
               
      except Exception as e:
            logger.error(f"停止直播间 {room_id} 录制失败: {str(e)}")
   
    def get_status(self):
      """获取所有轮询状态"""
      with self.lock:
            # 过滤掉不能JSON序列化的对象(如Thread)
            status = {}
            for room_id, room_info in self.polling_rooms.items():
                status = {
                  'room_url': room_info['room_url'],
                  'room_id': room_info['room_id'],
                  'check_interval': room_info['check_interval'],
                  'auto_record': room_info['auto_record'],
                  'status': room_info['status'],
                  'last_check': room_info['last_check'].isoformat() if room_info['last_check'] else None,
                  'stream_url': room_info['stream_url'],
                  'recording_session_id': room_info['recording_session_id'],
                  'anchor_name': room_info.get('anchor_name', f'anchor_{room_id}'),# 新增:主播名字
                  'live_title': room_info.get('live_title', ''),# 新增:直播标题
                  'added_time': room_info.get('added_time').isoformat() if room_info.get('added_time') else None,# 新增:添加时间
                  'online_count': room_info.get('online_count', 0),# 新增:在线人数
                  'viewer_count_text': room_info.get('viewer_count_text', '')# 新增:观看人数文本
                  # 注意:我们不包含 'thread' 字段,因为它不能JSON序列化
                }
            return status
   
    def _add_to_history(self, room_id, room_url, action, description, anchor_name=None):
      """添加记录到历史(简化版,带去重功能)"""
      # 获取主播名字,优先使用参数,其次从房间信息中获取
      if not anchor_name:
            with self.lock:
                if room_id in self.polling_rooms:
                  anchor_name = self.polling_rooms.get('anchor_name', f'anchor_{room_id}')
                else:
                  anchor_name = f'anchor_{room_id}'
         
      # 检查是否已存在相同的链接(去重)
      existing_urls = {record['room_url'] for record in self.polling_history}
      if room_url in existing_urls:
            logger.info(f"历史记录去重: 链接 {room_url} 已存在,跳过添加")
            return
         
      history_record = {
            'id': f"{room_id}_{int(time.time()*1000)}",# 唯一ID
            'anchor_name': anchor_name,
            'room_url': room_url,
            'timestamp': datetime.now().isoformat(),
            'date': datetime.now().strftime('%Y-%m-%d'),
            'time': datetime.now().strftime('%H:%M:%S')
      }
         
      # 添加到历史列表的开头(最新的在前面)
      self.polling_history.insert(0, history_record)
         
      # 保持历史记录数量在限制内
      if len(self.polling_history) > self.max_history_records:
            self.polling_history = self.polling_history[:self.max_history_records]
         
      # 保存历史记录到文件
      self._save_history_to_file()
         
      logger.info(f"历史记录: {description} (房间 {room_id}),主播: {anchor_name}")
   
    def get_history(self, limit=50, room_id=None, action=None):
      """获取历史记录"""
      with self.lock:
            history = self.polling_history.copy()
         
      # 限制返回数量
      return history[:limit]
   
    def _save_rooms_to_file(self):
      """保存直播间列表到文件"""
      try:
            rooms_data = {}
            for room_id, room_info in self.polling_rooms.items():
                rooms_data = {
                  'room_url': room_info['room_url'],
                  'check_interval': room_info['check_interval'],
                  'auto_record': room_info['auto_record'],
                  'anchor_name': room_info.get('anchor_name', f'anchor_{room_id}'),
                  'added_time': room_info['added_time'].isoformat() if room_info.get('added_time') else datetime.now().isoformat()
                }
            
            with open(self.rooms_file, 'w', encoding='utf-8') as f:
                json.dump(rooms_data, f, ensure_ascii=False, indent=2)
            
            logger.info(f"已保存 {len(rooms_data)} 个直播间到 {self.rooms_file}")
      except Exception as e:
            logger.error(f"保存直播间列表失败: {str(e)}")
   
    def _load_rooms_from_file(self):
      """从文件加载直播间列表"""
      try:
            if os.path.exists(self.rooms_file):
                with open(self.rooms_file, 'r', encoding='utf-8') as f:
                  rooms_data = json.load(f)
               
                for room_id, room_info in rooms_data.items():
                  # 使用加载的数据创建直播间信息
                  self.polling_rooms = {
                        'room_url': room_info['room_url'],
                        'room_id': room_id,
                        'check_interval': room_info.get('check_interval', 60),
                        'auto_record': room_info.get('auto_record', False),
                        'status': 'waiting',
                        'last_check': None,
                        'stream_url': None,
                        'recording_session_id': None,
                        'thread': None,
                        'anchor_name': room_info.get('anchor_name', f'anchor_{room_id}'),
                        'live_title': '',
                        'added_time': datetime.fromisoformat(room_info.get('added_time', datetime.now().isoformat())),
                        'history_added': False,# 加载的房间也需要添加历史记录(如果能获取到真实主播名字)
                        'online_count': room_info.get('online_count', 0),# 新增:在线人数
                        'viewer_count_text': room_info.get('viewer_count_text', '')# 新增:观看人数文本
                  }
                     
                  # 启动轮询线程
                  thread = threading.Thread(
                        target=self._poll_room,
                        args=(room_id,),
                        daemon=True
                  )
                  thread.start()
                  self.polling_rooms['thread'] = thread
               
                logger.info(f"从 {self.rooms_file} 加载了 {len(rooms_data)} 个直播间")
            else:
                logger.info(f"直播间配置文件 {self.rooms_file} 不存在,将创建新文件")
      except Exception as e:
            logger.error(f"加载直播间列表失败: {str(e)}")
   
    def _save_history_to_file(self):
      """保存历史记录到文件"""
      try:
            with open(self.history_file, 'w', encoding='utf-8') as f:
                json.dump(self.polling_history, f, ensure_ascii=False, indent=2)
            
            logger.debug(f"已保存历史记录到 {self.history_file}")
      except Exception as e:
            logger.error(f"保存历史记录失败: {str(e)}")
   
    def _load_history_from_file(self):
      """从文件加载历史记录(带去重功能)"""
      try:
            if os.path.exists(self.history_file):
                with open(self.history_file, 'r', encoding='utf-8') as f:
                  raw_history = json.load(f)
               
                # 去重处理:根据 room_url 去重,保留最新的记录
                seen_urls = set()
                deduped_history = []
               
                for record in raw_history:
                  room_url = record.get('room_url', '')
                  if room_url not in seen_urls:
                        seen_urls.add(room_url)
                        deduped_history.append(record)
                  else:
                        logger.debug(f"去重: 跳过重复链接 {room_url}")
               
                self.polling_history = deduped_history
               
                # 如果去重后数量有变化,保存文件
                if len(deduped_history) != len(raw_history):
                  logger.info(f"历史记录去重: 从 {len(raw_history)} 条去重到 {len(deduped_history)} 条")
                  self._save_history_to_file()
               
                logger.info(f"从 {self.history_file} 加载了 {len(self.polling_history)} 条历史记录")
            else:
                logger.info(f"历史记录文件 {self.history_file} 不存在,将创建新文件")
      except Exception as e:
            logger.error(f"加载历史记录失败: {str(e)}")
   
    def stop_all(self):
      """停止所有轮询"""
      self.running = False
      with self.lock:
            for room_id in list(self.polling_rooms.keys()):
                self.remove_room(room_id)

# 全局轮询管理器实例
multi_poller = MultiRoomPoller()

def record_stream(stream_url, output_path, session_id):
    """
    使用 FFmpeg 录制直播流(支持分段录制)
    :param stream_url: 直播流地址
    :param output_path: 输出文件路径(不含分段序号)
    :param session_id: 录制会话ID
    """
    try:
      logger.info(f"开始录制会话 {session_id}: {stream_url}")
         
      # 创建录制目录
      os.makedirs(os.path.dirname(output_path), exist_ok=True)
         
      # 更新录制会话状态
      with recording_lock:
            recording_sessions = {
                'process': None,
                'output_path': output_path,
                'start_time': datetime.now(),
                'stream_url': stream_url,
                'status': 'recording',
                'segments': [],
                'current_segment': 0
            }
         
      # 生成分段文件名模板
      base_name = output_path.rsplit('.', 1)
      segment_template = f"{base_name}_part%03d.mp4"
         
      logger.info(f"录制会话 {session_id} 输出路径: {output_path}")
      logger.info(f"录制会话 {session_id} 分段模板: {segment_template}")
         
      # 构建 FFmpeg 命令 - 使用正确的分段格式
      if stream_url.endswith('.m3u8'):
            cmd = [
                'ffmpeg',
                '-i', stream_url,
                '-c', 'copy',# 复制流,不重新编码
                '-bsf:a', 'aac_adtstoasc',# 音频流修复
                '-f', 'segment',# 使用分段格式
                '-segment_time', '1800',# 30分钟分段
                '-segment_format', 'mp4',# 分段格式为MP4
                '-reset_timestamps', '1',# 重置时间戳
                '-segment_list_flags', 'live',# 实时分段列表
                segment_template# 分段文件名模板
            ]
      else:
            cmd = [
                'ffmpeg',
                '-i', stream_url,
                '-c', 'copy',# 复制流,不重新编码
                '-f', 'segment',# 使用分段格式
                '-segment_time', '1800',# 30分钟分段
                '-segment_format', 'mp4',# 分段格式为MP4
                '-reset_timestamps', '1',# 重置时间戳
                '-segment_list_flags', 'live',# 实时分段列表
                segment_template# 分段文件名模板
            ]
         
      logger.info(f"FFmpeg 命令: {' '.join(cmd)}")
         
      # 执行录制
      process = subprocess.Popen(
            cmd,
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE,
            universal_newlines=True
      )
         
      # 更新录制会话状态
      with recording_lock:
            recording_sessions['process'] = process
         
      # 等待进程结束或手动停止
      stdout, stderr = process.communicate()
         
      # 更新最终状态
      with recording_lock:
            if session_id in recording_sessions:
                if process.returncode == 0:
                  recording_sessions['status'] = 'completed'
                  logger.info(f"录制会话 {session_id} 成功完成")
                else:
                  recording_sessions['status'] = 'failed'
                  recording_sessions['error'] = stderr
                  logger.error(f"录制会话 {session_id} 失败: {stderr}")
                recording_sessions['end_time'] = datetime.now()
            
    except Exception as e:
      logger.error(f"录制会话 {session_id} 异常: {str(e)}")
      with recording_lock:
            if session_id in recording_sessions:
                recording_sessions['status'] = 'failed'
                recording_sessions['error'] = str(e)
                recording_sessions['end_time'] = datetime.now()

@app.route('/api/record/start', methods=['POST'])
@handle_exceptions
def start_recording():
    """
    开始录制直播流
    """
    data = request.get_json()
    stream_url = data.get('stream_url')
    session_id = data.get('session_id') or f"recording_{int(time.time())}"
    anchor_name = data.get('anchor_name', 'unknown_anchor')# 新增:主播名字参数
   
    if not stream_url:
      return jsonify({
            'success': False,
            'message': '缺少直播流地址'
      })
   
    # 检查是否已在录制
    with recording_lock:
      if session_id in recording_sessions and recording_sessions['status'] == 'recording':
            return jsonify({
                'success': False,
                'message': '该会话已在录制中'
            })
   
    # 清理文件名中的非法字符
    safe_anchor_name = re.sub(r'[<>:"/\\|?*]', '_', anchor_name)
   
    # 生成输出文件路径(使用主播名字)
    timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
    output_path = f"recordings/{safe_anchor_name}_{timestamp}.mp4"
   
    # 启动录制线程
    thread = threading.Thread(
      target=record_stream,
      args=(stream_url, output_path, session_id),
      daemon=True
    )
    thread.start()
   
    return jsonify({
      'success': True,
      'session_id': session_id,
      'output_path': output_path,
      'message': '录制已开始'
    })

@app.route('/api/record/stop', methods=['POST'])
@handle_exceptions
def stop_recording():
    """
    停止录制
    """
    data = request.get_json()
    session_id = data.get('session_id')
   
    if not session_id:
      return jsonify({
            'success': False,
            'message': '缺少会话ID'
      })
   
    with recording_lock:
      if session_id not in recording_sessions:
            return jsonify({
                'success': False,
                'message': '找不到录制会话'
            })
         
      session = recording_sessions
      if session['status'] != 'recording':
            return jsonify({
                'success': False,
                'message': f'会话状态为 {session["status"]}, 无法停止'
            })
         
      # 终止 FFmpeg 进程
      try:
            session['process'].terminate()
            session['status'] = 'stopped'
            session['end_time'] = datetime.now()
            logger.info(f"已停止录制会话 {session_id}")
      except Exception as e:
            logger.error(f"停止录制会话 {session_id} 失败: {str(e)}")
            return jsonify({
                'success': False,
                'message': f'停止录制失败: {str(e)}'
            })
   
    return jsonify({
      'success': True,
      'message': '录制已停止'
    })

@app.route('/api/get_current_stream', methods=['GET'])
@handle_exceptions
def get_current_stream():
    """
    获取当前最新的直播流地址
    """
    import os
   
    stream_file = 'current_stream.txt'
   
    if os.path.exists(stream_file):
      try:
            with open(stream_file, 'r', encoding='utf-8') as f:
                stream_url = f.read().strip()
            if stream_url:
                logger.info(f"读取到当前直播流地址: {stream_url}")
                return jsonify({
                  'success': True,
                  'stream_url': stream_url,
                  'message': '成功获取直播流地址'
                })
            else:
                return jsonify({
                  'success': False,
                  'message': '直播流文件为空'
                })
      except Exception as e:
            logger.error(f"读取直播流文件失败: {str(e)}")
            return jsonify({
                'success': False,
                'message': f'读取文件失败: {str(e)}'
            })
    else:
      return jsonify({
            'success': False,
            'message': '直播流文件不存在'
      })

@app.route('/api/record/split', methods=['POST'])
@handle_exceptions
def split_recording():
    """
    手动分段录制
    """
    data = request.get_json()
    session_id = data.get('session_id')
   
    if not session_id:
      return jsonify({
            'success': False,
            'message': '缺少会话ID'
      })
   
    with recording_lock:
      if session_id not in recording_sessions:
            return jsonify({
                'success': False,
                'message': '找不到录制会话'
            })
         
      session = recording_sessions
      if session['status'] != 'recording':
            return jsonify({
                'success': False,
                'message': f'会话状态为 {session["status"]}, 无法分段'
            })
         
      # 向 FFmpeg 进程发送分割信号
      try:
            # FFmpeg 的 segment 功能会自动创建新分段,这里只需记录操作
            session['current_segment'] += 1
            logger.info(f"已为录制会话 {session_id} 创建新分段 {session['current_segment']}")
            
            return jsonify({
                'success': True,
                'message': f'已创建新分段 {session["current_segment"]}',
                'segment_number': session['current_segment']
            })
      except Exception as e:
            logger.error(f"分段录制会话 {session_id} 失败: {str(e)}")
            return jsonify({
                'success': False,
                'message': f'分段失败: {str(e)}'
            })

@app.route('/api/poll', methods=['POST'])
@handle_exceptions
def poll_live_stream():
    data = request.get_json()
    live_url = data.get('live_url')
    logger.info(f"收到轮询请求,直播间地址: {live_url}")
   
    # 检查URL是否有效
    if not live_url:
      logger.warning("轮询请求中URL为空")
      return jsonify({
            'success': False,
            'message': '直播间地址为空'
      })
   
    # 处理不同格式的输入
    processed_url = live_url.strip()
   
    # 1. 检查是否是纯数字(主播ID)
    if re.match(r'^\d+$', processed_url):
      logger.info(f"检测到主播ID格式: {processed_url}")
      room_id = processed_url
      full_url = f"https://live.douyin.com/{room_id}"
   
    # 2. 检查是否是完整的抖音直播URL
    elif "douyin.com" in processed_url:
      logger.info(f"检测到抖音URL格式: {processed_url}")
      # 提取房间号
      room_id_match = re.search(r'live\.douyin\.com\/([^/?]+)', processed_url)
      if room_id_match:
            room_id = room_id_match.group(1)
            full_url = f"https://live.douyin.com/{room_id}"
      else:
            # 尝试从URL路径中提取最后一部分
            url_parts = processed_url.split('/')
            room_id = url_parts[-1] or url_parts[-2]
            full_url = processed_url
   
    # 3. 其他格式(可能是短链接或其他标识符)
    else:
      logger.info(f"未识别的URL格式,尝试直接使用: {processed_url}")
      room_id = processed_url
      full_url = processed_url
   
    logger.info(f"处理后的房间ID: {room_id}, 完整URL: {full_url}")
   
    # 检查主播是否开播
    try:
      is_live = check_anchor_status(room_id)
         
      # 如果检测为未开播,但用户确认已开播,增加额外检查
      if not is_live:
            logger.warning(f"初步检测主播 {room_id} 未开播,进行二次验证")
            # 增加等待时间
            time.sleep(5)
            # 再次检查
            is_live = check_anchor_status(room_id)
         
      # 如果检测到开播,尝试解析直播流地址
      stream_url = None
      if is_live:
            logger.info(f"检测到主播 {room_id} 正在直播,开始解析直播流地址")
            try:
                stream_url = get_real_stream_url(full_url)
                if stream_url:
                  logger.info(f"成功解析直播流地址: {stream_url}")
                else:
                  logger.warning(f"无法解析直播流地址,但主播确实在直播")
            except Exception as parse_error:
                logger.error(f"解析直播流地址异常: {str(parse_error)}")
                # 解析失败不影响轮询结果,只是记录日志
            
      logger.info(f"最终轮询结果: 主播 {room_id} {'正在直播' if is_live else '未开播'}")
         
      # 按照API接口规范返回数据
      response_data = {
            'success': True,
            'message': '轮询请求已处理',
            'data': {
                'live_url': live_url,
                'is_live': is_live,
                'room_id': room_id,
                'full_url': full_url
            }
      }
         
      # 如果解析到了直播流地址,添加到返回数据中
      if stream_url:
            response_data['data']['stream_url'] = stream_url
            
      return jsonify(response_data)
    except Exception as e:
      logger.error(f"轮询处理异常: {str(e)}")
      return jsonify({
            'success': False,
            'message': f'轮询处理异常: {str(e)}',
            'live_url': live_url
      })

@app.route('/api/record/status', methods=['GET'])
@handle_exceptions
def get_recording_status():
    """
    获取录制状态
    """
    session_id = request.args.get('session_id')
   
    if session_id:
      with recording_lock:
            if session_id in recording_sessions:
                session = recording_sessions
                return jsonify({
                  'success': True,
                  'session_id': session_id,
                  'status': session['status'],
                  'output_path': session.get('output_path'),
                  'start_time': session.get('start_time'),
                  'end_time': session.get('end_time'),
                  'stream_url': session.get('stream_url')
                })
            else:
                return jsonify({
                  'success': False,
                  'message': '找不到录制会话'
                })
    else:
      # 返回所有录制会话状态
      with recording_lock:
            sessions = {
                sid: {
                  'status': session['status'],
                  'output_path': session.get('output_path'),
                  'start_time': session.get('start_time'),
                  'end_time': session.get('end_time'),
                  'stream_url': session.get('stream_url')
                }
                for sid, session in recording_sessions.items()
            }
      return jsonify({
            'success': True,
            'sessions': sessions
      })

@app.route('/api/multi-poll/add', methods=['POST'])
@handle_exceptions
def add_polling_room():
    """添加直播间到轮询列表"""
    data = request.get_json()
    room_url = data.get('room_url')
    room_id = data.get('room_id')
    check_interval = data.get('check_interval', 60)# 默认60秒检查一次
    auto_record = data.get('auto_record', False)# 是否自动录制
   
    if not room_url:
      return jsonify({
            'success': False,
            'message': '缺少直播间地址'
      })
   
    # 如果没有提供room_id,尝试从 URL解析
    if not room_id:
      # 处理不同格式的输入
      processed_url = room_url.strip()
      logger.info(f"尝试解析URL: {processed_url}")
         
      # 1. 检查是否是纯数字(主播ID)
      if re.match(r'^\d+$', processed_url):
            logger.info(f"检测到主播ID格式: {processed_url}")
            room_id = processed_url
         
      # 2. 检查是否是完整的抖音直播URL
      elif "douyin.com" in processed_url:
            logger.info(f"检测到抖音URL格式: {processed_url}")
            
            # 尝试多种URL格式的解析
            # 格式1: https://live.douyin.com/123456
            room_id_match = re.search(r'live\.douyin\.com/([^/?&#]+)', processed_url)
            if room_id_match:
                room_id = room_id_match.group(1)
                logger.info(f"从live.douyin.com URL提取房间ID: {room_id}")
            else:
                # 格式2: https://www.douyin.com/user/MS4wLjABAAAA...
                user_id_match = re.search(r'/user/([^/?&#]+)', processed_url)
                if user_id_match:
                  room_id = user_id_match.group(1)
                  logger.info(f"从用户主页URL提取用户ID: {room_id}")
                else:
                  # 格式3: 尝试从URL路径中提取数字部分
                  url_parts = processed_url.split('/')
                  for part in reversed(url_parts):
                        if part and part != '' and not part.startswith('?'):
                            # 移除可能的参数
                            clean_part = part.split('?').split('#')
                            if clean_part:
                              # 如果是纯数字,直接使用
                              if re.match(r'^\d+$', clean_part):
                                    room_id = clean_part
                                    logger.info(f"从URL路径提取房间ID: {room_id}")
                                    break
                              # 否则使用完整的部分
                              else:
                                    room_id = clean_part
                                    logger.info(f"从URL路径提取标识符: {room_id}")
                                    break
                     
                  if not room_id:
                        return jsonify({
                            'success': False,
                            'message': f'无法从 URL解析房间ID: {processed_url}'
                        })
         
      # 3. 其他格式(可能是短链接或其他标识符)
      else:
            logger.info(f"未识别的URL格式,尝试直接使用: {processed_url}")
            room_id = processed_url
   
    logger.info(f"最终解析得到的房间ID: {room_id}")
   
    success = multi_poller.add_room(room_id, room_url, check_interval, auto_record)
   
    if success:
      return jsonify({
            'success': True,
            'message': f'已添加直播间 {room_id} 到轮询列表',
            'room_id': room_id
      })
    else:
      return jsonify({
            'success': False,
            'message': f'直播间 {room_id} 已在轮询列表中'
      })

@app.route('/api/multi-poll/remove', methods=['POST'])
@handle_exceptions
def remove_polling_room():
    """从轮询列表移除直播间"""
    data = request.get_json()
    room_id = data.get('room_id')
   
    if not room_id:
      return jsonify({
            'success': False,
            'message': '缺少房间ID'
      })
   
    success = multi_poller.remove_room(room_id)
   
    if success:
      return jsonify({
            'success': True,
            'message': f'已移除直播间 {room_id}'
      })
    else:
      return jsonify({
            'success': False,
            'message': f'直播间 {room_id} 不在轮询列表中'
      })

@app.route('/api/multi-poll/status', methods=['GET'])
@handle_exceptions
def get_multi_polling_status():
    """获取多直播间轮询状态"""
    status = multi_poller.get_status()
   
    return jsonify({
      'success': True,
      'polling_rooms': status,
      'total_rooms': len(status)
    })

@app.route('/api/multi-poll/history', methods=['GET'])
@handle_exceptions
def get_polling_history():
    """获取轮询历史记录"""
    # 获取查询参数
    limit = request.args.get('limit', 50, type=int)
    room_id = request.args.get('room_id')
    action = request.args.get('action')
   
    # 限制limit的范围
    limit = min(max(1, limit), 200)# 限制在1-200之间
   
    history = multi_poller.get_history(limit=limit, room_id=room_id, action=action)
   
    return jsonify({
      'success': True,
      'history': history,
      'total_records': len(history),
      'filters': {
            'limit': limit,
            'room_id': room_id,
            'action': action
      }
    })

@app.route('/api/multi-poll/start-record', methods=['POST'])
@handle_exceptions
def start_manual_recording():
    """手动为指定直播间启动录制"""
    data = request.get_json()
    room_id = data.get('room_id')
   
    if not room_id:
      return jsonify({
            'success': False,
            'message': '缺少房间ID'
      })
   
    status = multi_poller.get_status()
    if room_id not in status:
      return jsonify({
            'success': False,
            'message': f'直播间 {room_id} 不在轮询列表中'
      })
   
    room_info = status
    if room_info['status'] != 'live' or not room_info['stream_url']:
      return jsonify({
            'success': False,
            'message': f'直播间 {room_id} 当前不在直播或无流地址'
      })
   
    if room_info['recording_session_id']:
      return jsonify({
            'success': False,
            'message': f'直播间 {room_id} 已在录制中'
      })
   
    # 启动录制
    multi_poller._start_recording(room_id, room_info['stream_url'])
   
    # 简化版:不记录手动录制
   
    return jsonify({
      'success': True,
      'message': f'已为直播间 {room_id} 启动录制'
    })

@app.route('/api/multi-poll/stop-record', methods=['POST'])
@handle_exceptions
def stop_manual_recording():
    """手动停止指定直播间的录制"""
    data = request.get_json()
    room_id = data.get('room_id')
   
    if not room_id:
      return jsonify({
            'success': False,
            'message': '缺少房间ID'
      })
   
    status = multi_poller.get_status()
    if room_id not in status:
      return jsonify({
            'success': False,
            'message': f'直播间 {room_id} 不在轮询列表中'
      })
   
    room_info = status
    if not room_info['recording_session_id']:
      return jsonify({
            'success': False,
            'message': f'直播间 {room_id} 当前未在录制'
      })
   
    # 停止录制
    multi_poller._stop_recording(room_id)
   
    # 简化版:不记录手动停止录制
   
    return jsonify({
      'success': True,
      'message': f'已停止直播间 {room_id} 的录制'
    })

@app.route('/api/multi-poll/pause', methods=['POST'])
@handle_exceptions
def pause_polling_room():
    """暂停指定直播间的轮询"""
    data = request.get_json()
    room_id = data.get('room_id')
   
    if not room_id:
      return jsonify({
            'success': False,
            'message': '缺少房间ID'
      })
   
    success = multi_poller.pause_room(room_id)
   
    if success:
      return jsonify({
            'success': True,
            'message': f'已暂停直播间 {room_id} 的轮询'
      })
    else:
      return jsonify({
            'success': False,
            'message': f'直播间 {room_id} 不在轮询列表中或已暂停'
      })

@app.route('/api/multi-poll/resume', methods=['POST'])
@handle_exceptions
def resume_polling_room():
    """恢复指定直播间的轮询"""
    data = request.get_json()
    room_id = data.get('room_id')
   
    if not room_id:
      return jsonify({
            'success': False,
            'message': '缺少房间ID'
      })
   
    success = multi_poller.resume_room(room_id)
   
    if success:
      return jsonify({
            'success': True,
            'message': f'已恢复直播间 {room_id} 的轮询'
      })
    else:
      return jsonify({
            'success': False,
            'message': f'直播间 {room_id} 不在轮询列表中或未暂停'
      })

if __name__ == '__main__':
    # 创建录制目录
    os.makedirs('recordings', exist_ok=True)
    # 监听所有接口,允许外部访问
    app.run(host='0.0.0.0', port=5000, debug=True)
复制代码前端:
<template>
<div class="multi-room-manager">
    <div class="header">
      <h3>多直播间管理</h3>
      <div class="header-actions">
      <button @click="showHistory = !showHistory" class="history-btn">
          {{ showHistory ? '隐藏历史' : '查看历史' }}
      </button>
      <button @click="showAddDialog = true" class="add-btn">添加直播间</button>
      </div>
    </div>

    <!-- 播放器区域 -->
    <div class="players-section">
      <h3>直播播放器</h3>
      <div class="players-container">
      <div
          v-for="(player, index) in players"
          :key="index"
          class="player-wrapper"
      >
          <div class="player-header">
            <span class="player-title">{{ player.title }}</span>
            <button @click="closePlayer(index)" class="close-player-btn">×</button>
          </div>
          <div class="player-controls">
            <button @click="toggleMute(index)" class="mute-btn">
            {{ player.muted ? ' 静音' : ' 取消静音' }}
            </button>
            <button @click="play(index)" class="play-btn">播放</button>
          </div>
          <video :ref="`videoPlayer${index}`" controls autoplay muted class="inline-video-player"></video>
          <div v-if="player.error" class="player-error">{{ player.error }}</div>
      </div>
      <div v-if="players.length === 0" class="no-players">
          暂无播放器,请点击直播间中的"播放"按钮添加播放器
      </div>
      </div>
    </div>

    <!-- 批量操作栏 -->
    <div v-if="selectedRooms.length > 0" class="bulk-action-bar">
      <div class="bulk-info">
      已选择 {{ selectedRooms.length }} 个直播间
      </div>
      <div class="bulk-actions">
      <button @click="bulkStartRecording" class="bulk-record-btn">批量录制</button>
      <button @click="bulkStopRecording" class="bulk-stop-btn">批量停止录制</button>
      <button @click="bulkPause" class="bulk-pause-btn">批量暂停</button>
      <button @click="bulkResume" class="bulk-resume-btn">批量恢复</button>
      <button @click="bulkRemove" class="bulk-remove-btn">批量移除</button>
      <button @click="clearSelection" class="bulk-clear-btn">取消选择</button>
      </div>
    </div>

    <!-- 添加直播间对话框 -->
    <div v-if="showAddDialog" class="dialog-overlay">
      <div class="dialog">
      <h4>添加直播间</h4>
      <div class="form-group">
          <label>直播间地址:</label>
          <input
            v-model="newRoom.url"
            placeholder="输入房间号或直播链接(如:123456 或 https://live.douyin.com/123456)"
            class="input-field"
          />
      </div>
      <div class="form-group">
          <label>检查间隔(秒):</label>
          <input
            v-model.number="newRoom.interval"
            type="number"
            placeholder="60"
            min="30"
            max="3600"
            class="input-field"
          />
      </div>
      <div class="form-group">
          <label>
            <input
            v-model="newRoom.autoRecord"
            type="checkbox"
            />
            开播时自动录制
          </label>
      </div>
      <div class="dialog-actions">
          <button @click="addRoom" class="confirm-btn">添加</button>
          <button @click="cancelAdd" class="cancel-btn">取消</button>
      </div>
      </div>
    </div>

    <!-- 直播间列表 -->
    <div class="room-list">
      <div
      v-for="(room, roomId) in sortedPollingRooms"
      :key="roomId"
      class="room-item"
      :class=""
      @click.ctrl.exact="toggleRoomSelection(roomId)"
      @click.shift.exact="selectRoomRange(roomId)"
      >
      <div class="room-selection">
          <input
            type="checkbox"
            :checked="selectedRooms.includes(roomId)"
            @click.stop="toggleRoomSelection(roomId)"
            class="room-checkbox"
          />
      </div>
      <div class="room-info">
          <div class="room-id">房间: {{ roomId }}
            <span v-if="room.anchor_name && room.anchor_name !== `anchor_${roomId}`" class="anchor-name">
            ({{ room.anchor_name }})
            </span>
          </div>
          <div class="room-status">
            状态: {{ getStatusText(room.status) }}
            <span v-if="room.status === 'live' && (room.online_count > 0 || room.viewer_count_text)" class="popularity">
            人气:{{ formatPopularity(room) }}
            </span>
            <span v-if="room.last_check" class="last-check">
            ({{ formatTime(room.last_check) }})
            </span>
          </div>
          <div class="room-url">{{ room.room_url }}</div>
          <div v-if="room.stream_url" class="stream-url">
            流地址: {{ room.stream_url.substring(0, 50) }}...
          </div>
      </div>
         
      <div class="room-actions">
          <!-- 播放按钮 -->
          <button
            v-if="room.status === 'live' && room.stream_url"
            @click.stop="playStream(room.stream_url)"
            class="play-btn"
          >
            播放
          </button>
         
          <!-- 录制控制 -->
          <button
            v-if="room.status === 'live' && !room.recording_session_id"
            @click.stop="startRecording(roomId)"
            class="record-btn"
          >
            开始录制
          </button>
          <button
            v-if="room.recording_session_id"
            @click.stop="stopRecording(roomId)"
            class="stop-record-btn"
          >
            停止录制
          </button>
         
          <!-- 暂停/恢复按钮 -->
          <button
            v-if="room.status !== 'paused'"
            @click.stop="pauseRoom(roomId)"
            class="pause-btn"
          >
            暂停
          </button>
          <button
            v-else
            @click.stop="resumeRoom(roomId)"
            class="resume-btn"
          >
            恢复
          </button>
         
          <!-- 删除直播间 -->
          <button
            @click.stop="removeRoom(roomId)"
            class="remove-btn"
          >
            移除
          </button>
      </div>
      </div>
    </div>

    <!-- 统计信息 -->
    <div class="stats">
      <div class="stat-item">
      <span class="stat-label">总房间数:</span>
      <span class="stat-value">{{ totalRooms }}</span>
      </div>
      <div class="stat-item">
      <span class="stat-label">在线房间:</span>
      <span class="stat-value">{{ liveRooms }}</span>
      </div>
      <div class="stat-item">
      <span class="stat-label">录制中:</span>
      <span class="stat-value">{{ recordingRooms }}</span>
      </div>
      <div class="stat-item">
      <span class="stat-label">已暂停:</span>
      <span class="stat-value">{{ pausedRooms }}</span>
      </div>
    </div>

    <!-- 错误信息 -->
    <div v-if="error" class="error-message">
      {{ error }}
    </div>

    <!-- 历史记录区域 -->
    <div v-if="showHistory" class="history-section">
      <div class="history-header">
      <h4>轮询历史记录</h4>
      <div class="history-filters">
          <button @click="refreshHistory" class="refresh-btn">刷新</button>
      </div>
      </div>
      
      <div class="history-list">
      <div v-if="historyLoading" class="loading">加载中...</div>
      <div v-else-if="historyRecords.length === 0" class="no-history">暂无历史记录</div>
      <div v-else>
          <div
            v-for="record in historyRecords"
            :key="record.id"
            class="history-item"
          >
            <div class="history-info">
            <div class="history-main">
                <span class="anchor-name">{{ record.anchor_name }}</span>
                <span class="room-url">{{ record.room_url }}</span>
            </div>
            <div class="history-time">{{ record.date }} {{ record.time }}</div>
            </div>
          </div>
         
          <!-- 加载更多按钮 -->
          <div v-if="historyRecords.length >= 50" class="load-more">
            <button @click="loadMoreHistory" class="load-more-btn">加载更多</button>
          </div>
      </div>
      </div>
    </div>
   

</div>
</template>

<script>
import flvjs from 'flv.js';

export default {
name: 'MultiRoomManager',
props: {
   
},
data() {
    return {
      pollingRooms: {},
      showAddDialog: false,
      showHistory: false,
      newRoom: {
      url: '',
      interval: 60,
      autoRecord: false
      },
      error: '',
      updateInterval: null,
      historyRecords: [],
      historyLoading: false,
      // 播放器列表,支持多个播放器
      players: [],
      selectedRooms: [],
      lastSelectedRoom: null,
      playerError: ''
    };
},
computed: {
    totalRooms() {
      return Object.keys(this.pollingRooms).length;
    },
    liveRooms() {
      return Object.values(this.pollingRooms).filter(room => room.status === 'live').length;
    },
    recordingRooms() {
      return Object.values(this.pollingRooms).filter(room => room.recording_session_id).length;
    },
    pausedRooms() {
      return Object.values(this.pollingRooms).filter(room => room.status === 'paused').length;
    },
    // 新增:排序后的直播间列表
    sortedPollingRooms() {
      // 将对象转换为数组并排序
      const roomsArray = Object.entries(this.pollingRooms);
      
      // 排序规则:
      // 1. 录制中的直播间在最上面
      // 2. 在线但未录制的直播间
      // 3. 暂停和直播结束的直播间在最下面
      roomsArray.sort((a, b) => {
      const = a;
      const = b;
         
      // 录制中的直播间优先级最高
      const isRecordingA = roomA.recording_session_id ? 1 : 0;
      const isRecordingB = roomB.recording_session_id ? 1 : 0;
         
      if (isRecordingA !== isRecordingB) {
          return isRecordingB - isRecordingA; // 录制中的在前面
      }
         
      // 在线状态的直播间优先级次之
      const isLiveA = roomA.status === 'live' ? 1 : 0;
      const isLiveB = roomB.status === 'live' ? 1 : 0;
         
      if (isLiveA !== isLiveB) {
          return isLiveB - isLiveA; // 在线的在前面
      }
         
      // 暂停和直播结束的直播间优先级最低
      const isPausedOrEndedA = (roomA.status === 'paused' || roomA.status === 'live_no_stream') ? 1 : 0;
      const isPausedOrEndedB = (roomB.status === 'paused' || roomB.status === 'live_no_stream') ? 1 : 0;
         
      if (isPausedOrEndedA !== isPausedOrEndedB) {
          return isPausedOrEndedA - isPausedOrEndedB; // 暂停和结束的在后面
      }
         
      // 如果优先级相同,按房间ID排序
      return roomIdA.localeCompare(roomIdB);
      });
      
      // 转换回对象格式
      const sortedRooms = {};
      roomsArray.forEach(() => {
      sortedRooms = room;
      });
      
      return sortedRooms;
    }
},
mounted() {
    this.loadStatus();
    this.loadHistory(); // 加载历史记录
    // 每5秒更新一次状态
    this.updateInterval = setInterval(this.loadStatus, 5000);
},
beforeDestroy() {
    if (this.updateInterval) {
      clearInterval(this.updateInterval);
    }
   
    // 销毁所有播放器
    this.players.forEach(playerObj => {
      if (playerObj.player) {
      playerObj.player.destroy();
      }
    });
},
methods: {
    async loadStatus() {
      try {
      const response = await fetch('http://127.0.0.1:5000/api/multi-poll/status');
      if (!response.ok) {
          throw new Error(`HTTP error! status: ${response.status}`);
      }
      const data = await response.json();
         
      if (data.success) {
          this.pollingRooms = data.polling_rooms;
          this.error = '';
      } else {
          this.error = data.message || '获取状态失败';
      }
      } catch (error) {
      console.error('获取状态失败:', error);
      this.error = '连接服务器失败';
      }
    },
   
    async addRoom() {
      if (!this.newRoom.url.trim()) {
      this.error = '请输入直播间地址';
      return;
      }
      
      try {
      const requestData = {
          room_url: this.newRoom.url.trim(),
          check_interval: this.newRoom.interval,
          auto_record: this.newRoom.autoRecord
      };
         
      console.log('发送添加直播间请求:', requestData);
         
      const response = await fetch('http://127.0.0.1:5000/api/multi-poll/add', {
          method: 'POST',
          headers: { 'Content-Type': 'application/json' },
          body: JSON.stringify(requestData)
      });
         
      if (!response.ok) {
          throw new Error(`HTTP error! status: ${response.status}`);
      }
         
      const data = await response.json();
      console.log('后端响应:', data);
         
      if (data.success) {
          this.showAddDialog = false;
          this.resetNewRoom();
          this.loadStatus(); // 刷新状态
          this.error = '';
          console.log('直播间添加成功:', data.room_id);
      } else {
          this.error = data.message || '添加失败';
          console.error('后端返回错误:', data.message);
      }
      } catch (error) {
      console.error('添加直播间失败:', error);
      this.error = '添加直播间失败: ' + error.message;
      }
    },
   
    async removeRoom(roomId) {
      if (!confirm(`确定要移除直播间 ${roomId} 吗?`)) {
      return;
      }
      
      try {
      const response = await fetch('http://127.0.0.1:5000/api/multi-poll/remove', {
          method: 'POST',
          headers: { 'Content-Type': 'application/json' },
          body: JSON.stringify({ room_id: roomId })
      });
         
      if (!response.ok) {
          throw new Error(`HTTP error! status: ${response.status}`);
      }
         
      const data = await response.json();
         
      if (data.success) {
          // 从选中列表中移除
          const index = this.selectedRooms.indexOf(roomId);
          if (index > -1) {
            this.selectedRooms.splice(index, 1);
          }
         
          this.loadStatus(); // 刷新状态
          this.error = '';
      } else {
          this.error = data.message || '移除失败';
      }
      } catch (error) {
      console.error('移除直播间失败:', error);
      this.error = '移除直播间失败: ' + error.message;
      }
    },
   
    async startRecording(roomId) {
      try {
      const response = await fetch('http://127.0.0.1:5000/api/multi-poll/start-record', {
          method: 'POST',
          headers: { 'Content-Type': 'application/json' },
          body: JSON.stringify({ room_id: roomId })
      });
         
      if (!response.ok) {
          throw new Error(`HTTP error! status: ${response.status}`);
      }
         
      const data = await response.json();
         
      if (data.success) {
          this.loadStatus(); // 刷新状态
          this.error = '';
      } else {
          this.error = data.message || '开始录制失败';
      }
      } catch (error) {
      console.error('开始录制失败:', error);
      this.error = '开始录制失败: ' + error.message;
      }
    },
   
    async stopRecording(roomId) {
      try {
      const response = await fetch('http://127.0.0.1:5000/api/multi-poll/stop-record', {
          method: 'POST',
          headers: { 'Content-Type': 'application/json' },
          body: JSON.stringify({ room_id: roomId })
      });
         
      if (!response.ok) {
          throw new Error(`HTTP error! status: ${response.status}`);
      }
         
      const data = await response.json();
         
      if (data.success) {
          this.loadStatus(); // 刷新状态
          this.error = '';
      } else {
          this.error = data.message || '停止录制失败';
      }
      } catch (error) {
      console.error('停止录制失败:', error);
      this.error = '停止录制失败: ' + error.message;
      }
    },
   
    // 新增:暂停直播间(停止轮询)
    async pauseRoom(roomId) {
      try {
      const response = await fetch('http://127.0.0.1:5000/api/multi-poll/pause', {
          method: 'POST',
          headers: { 'Content-Type': 'application/json' },
          body: JSON.stringify({ room_id: roomId })
      });
         
      if (!response.ok) {
          throw new Error(`HTTP error! status: ${response.status}`);
      }
         
      const data = await response.json();
         
      if (data.success) {
          this.loadStatus(); // 刷新状态
          this.error = '';
      } else {
          this.error = data.message || '暂停失败';
      }
      } catch (error) {
      console.error('暂停直播间失败:', error);
      this.error = '暂停直播间失败: ' + error.message;
      }
    },
   
    // 新增:恢复直播间(恢复轮询)
    async resumeRoom(roomId) {
      try {
      const response = await fetch('http://127.0.0.1:5000/api/multi-poll/resume', {
          method: 'POST',
          headers: { 'Content-Type': 'application/json' },
          body: JSON.stringify({ room_id: roomId })
      });
         
      if (!response.ok) {
          throw new Error(`HTTP error! status: ${response.status}`);
      }
         
      const data = await response.json();
         
      if (data.success) {
          this.loadStatus(); // 刷新状态
          this.error = '';
      } else {
          this.error = data.message || '恢复失败';
      }
      } catch (error) {
      console.error('恢复直播间失败:', error);
      this.error = '恢复直播间失败: ' + error.message;
      }
    },
   
    cancelAdd() {
      this.showAddDialog = false;
      this.resetNewRoom();
    },
   
    resetNewRoom() {
      this.newRoom = {
      url: '',
      interval: 60,
      autoRecord: false
      };
    },
   
    getStatusClass(status) {
      return {
      'status-live': status === 'live',
      'status-offline': status === 'offline' || status === 'live_no_stream',
      'status-checking': status === 'checking',
      'status-error': status === 'error',
      'status-waiting': status === 'waiting',
      'status-paused': status === 'paused'
      };
    },
   
    getStatusText(status) {
      const statusMap = {
      'waiting': '等待中',
      'checking': '检查中',
      'live': '在线',
      'offline': '离线',
      'error': '错误',
      'live_no_stream': '直播结束',
      'paused': '已暂停'
      };
      return statusMap || status;
    },
   
    formatTime(timeStr) {
      if (!timeStr) return '';
      const date = new Date(timeStr);
      return date.toLocaleTimeString();
    },
   
    formatPopularity(room) {
      // 优先使用原始文本(如"32人在线")
      if (room.viewer_count_text && room.viewer_count_text.trim()) {
      // 如果原始文本包含太多信息,尝试提取数字
      if (room.viewer_count_text.length > 20) {
          // 提取数字部分
          const match = room.viewer_count_text.match(/(在线观众[\s·]*([\d,]+)|观众[\s·]*([\d,]+)|([\d,]+)\s*人在线)/);
          if (match) {
            const count = (match || match || match || '0').replace(',', '');
            return `${count}人`;
          }
      } else {
          return room.viewer_count_text;
      }
      }
      
      // 否则格式化数字
      const count = room.online_count || 0;
      if (count >= 10000) {
      const wan = (count / 10000).toFixed(1);
      return `${wan}万人`;
      } else if (count > 0) {
      return `${count}人`;
      }
      
      return '0人';
    },
   
    async loadHistory() {
      this.historyLoading = true;
      try {
      const response = await fetch('http://127.0.0.1:5000/api/multi-poll/history?limit=50');
      if (!response.ok) {
          throw new Error(`HTTP error! status: ${response.status}`);
      }
         
      const data = await response.json();
         
      if (data.success) {
          this.historyRecords = data.history;
      } else {
          console.error('获取历史记录失败:', data.message);
      }
      } catch (error) {
      console.error('加载历史记录失败:', error);
      } finally {
      this.historyLoading = false;
      }
    },
   
    async loadMoreHistory() {
      // 加载更多历史记录(简单实现,可以扩展为真正的分页)
      this.loadHistory();
    },
   
    refreshHistory() {
      this.loadHistory();
    },
   
    // 新增:播放直播流
    playStream(streamUrl) {
      // 查找对应的直播间信息
      let roomInfo = null;
      let roomTitle = '未知直播间';
      
      // 遍历所有直播间查找匹配的流地址
      for (const of Object.entries(this.pollingRooms)) {
      if (room.stream_url === streamUrl && room.status === 'live') {
          roomInfo = room;
          // 使用主播名作为标题,如果没有则使用房间ID
          roomTitle = (room.anchor_name && room.anchor_name !== `anchor_${roomId}`) ? room.anchor_name : `房间 ${roomId}`;
          break;
      }
      }
      
      // 添加新的播放器到播放器列表
      const playerIndex = this.players.length;
      this.players.push({
      url: streamUrl,
      player: null,
      error: '',
      muted: true,// 默认静音
      title: roomTitle// 添加直播间标题
      });
      
      this.$nextTick(() => {
      this.initPlayer(playerIndex);
      });
    },
   
    // 初始化FLV播放器
    initPlayer(playerIndex) {
      // 销毁已存在的播放器
      if (this.players.player) {
      this.players.player.destroy();
      this.players.player = null;
      }
      
      this.players.error = '';
      
      try {
      if (flvjs.isSupported()) {
          const videoElement = this.$refs[`videoPlayer${playerIndex}`];
          this.players.player = flvjs.createPlayer({
            type: 'flv',
            url: this.players.url
          });
         
          this.players.player.attachMediaElement(videoElement);
          this.players.player.load();
         
          // 设置默认静音状态
          videoElement.muted = this.players.muted;
         
          this.players.player.play().catch(error => {
            console.error('播放失败:', error);
            this.players.error = '播放失败: ' + error.message;
          });
      } else {
          this.players.error = '当前浏览器不支持FLV播放';
          console.error('FLV.js is not supported');
      }
      } catch (error) {
      console.error('初始化播放器失败:', error);
      this.players.error = '初始化播放器失败: ' + error.message;
      }
    },
   
    // 新增:关闭播放器
    closePlayer(playerIndex) {
      // 销毁指定的播放器
      if (this.players.player) {
      this.players.player.destroy();
      this.players.player = null;
      }
      
      // 从播放器列表中移除
      this.players.splice(playerIndex, 1);
    },
   
    // 新增:切换静音状态
    toggleMute(playerIndex) {
      const playerObj = this.players;
      if (playerObj.player) {
      const videoElement = this.$refs[`videoPlayer${playerIndex}`];
      playerObj.muted = !playerObj.muted;
      videoElement.muted = playerObj.muted;
      }
    },
   
    // 新增:播放方法
    play(playerIndex) {
      const playerObj = this.players;
      if (playerObj.player) {
      // 取消静音并播放
      playerObj.muted = false;
      const videoElement = this.$refs[`videoPlayer${playerIndex}`];
      videoElement.muted = false;
      playerObj.player.play().catch(error => {
          console.error('播放失败:', error);
          playerObj.error = '播放失败: ' + error.message;
      });
      }
    },
   
    // 新增:切换直播间选择
    toggleRoomSelection(roomId) {
      const index = this.selectedRooms.indexOf(roomId);
      if (index > -1) {
      // 如果已选中,则取消选中
      this.selectedRooms.splice(index, 1);
      } else {
      // 如果未选中,则选中
      this.selectedRooms.push(roomId);
      }
      this.lastSelectedRoom = roomId;
    },
   
    // 新增:选择范围内的直播间(Shift键功能)
    selectRoomRange(roomId) {
      if (!this.lastSelectedRoom) {
      this.toggleRoomSelection(roomId);
      return;
      }
      
      const roomIds = Object.keys(this.pollingRooms);
      const lastIndex = roomIds.indexOf(this.lastSelectedRoom);
      const currentIndex = roomIds.indexOf(roomId);
      
      if (lastIndex === -1 || currentIndex === -1) {
      this.toggleRoomSelection(roomId);
      return;
      }
      
      // 确定范围
      const start = Math.min(lastIndex, currentIndex);
      const end = Math.max(lastIndex, currentIndex);
      
      // 选中范围内的所有直播间
      const newSelection = roomIds.slice(start, end + 1);
      
      // 合并选中项(避免重复)
      const uniqueSelection = [...new Set([...this.selectedRooms, ...newSelection])];
      this.selectedRooms = uniqueSelection;
      this.lastSelectedRoom = roomId;
    },
   
    // 新增:清除选择
    clearSelection() {
      this.selectedRooms = [];
      this.lastSelectedRoom = null;
    },
   
    // 新增:批量开始录制
    async bulkStartRecording() {
      if (this.selectedRooms.length === 0) {
      this.error = '请先选择直播间';
      return;
      }
      
      let successCount = 0;
      let failCount = 0;
      
      for (const roomId of this.selectedRooms) {
      try {
          // 检查直播间是否在线且未在录制
          const room = this.pollingRooms;
          if (room.status === 'live' && !room.recording_session_id) {
            await this.startRecording(roomId);
            successCount++;
          }
      } catch (error) {
          console.error(`批量开始录制失败 (房间 ${roomId}):`, error);
          failCount++;
      }
      }
      
      this.error = `批量开始录制完成: 成功 ${successCount} 个, 失败 ${failCount} 个`;
      // 重新加载状态以更新界面
      await this.loadStatus();
    },
   
    // 新增:批量停止录制
    async bulkStopRecording() {
      if (this.selectedRooms.length === 0) {
      this.error = '请先选择直播间';
      return;
      }
      
      let successCount = 0;
      let failCount = 0;
      
      for (const roomId of this.selectedRooms) {
      try {
          // 检查直播间是否正在录制
          const room = this.pollingRooms;
          if (room.recording_session_id) {
            await this.stopRecording(roomId);
            successCount++;
          }
      } catch (error) {
          console.error(`批量停止录制失败 (房间 ${roomId}):`, error);
          failCount++;
      }
      }
      
      this.error = `批量停止录制完成: 成功 ${successCount} 个, 失败 ${failCount} 个`;
      // 重新加载状态以更新界面
      await this.loadStatus();
    },
   
    // 新增:批量暂停(停止轮询)
    async bulkPause() {
      if (this.selectedRooms.length === 0) {
      this.error = '请先选择直播间';
      return;
      }
      
      let successCount = 0;
      let failCount = 0;
      
      for (const roomId of this.selectedRooms) {
      try {
          // 检查直播间是否未暂停
          const room = this.pollingRooms;
          if (room.status !== 'paused') {
            await this.pauseRoom(roomId);
            successCount++;
          }
      } catch (error) {
          console.error(`批量暂停失败 (房间 ${roomId}):`, error);
          failCount++;
      }
      }
      
      this.error = `批量暂停完成: 成功 ${successCount} 个, 失败 ${failCount} 个`;
      // 重新加载状态以更新界面
      await this.loadStatus();
    },
   
    // 新增:批量恢复(恢复轮询)
    async bulkResume() {
      if (this.selectedRooms.length === 0) {
      this.error = '请先选择直播间';
      return;
      }
      
      let successCount = 0;
      let failCount = 0;
      
      for (const roomId of this.selectedRooms) {
      try {
          // 检查直播间是否已暂停
          const room = this.pollingRooms;
          if (room.status === 'paused') {
            await this.resumeRoom(roomId);
            successCount++;
          }
      } catch (error) {
          console.error(`批量恢复失败 (房间 ${roomId}):`, error);
          failCount++;
      }
      }
      
      this.error = `批量恢复完成: 成功 ${successCount} 个, 失败 ${failCount} 个`;
      // 重新加载状态以更新界面
      await this.loadStatus();
    },
   
    // 新增:批量移除
    async bulkRemove() {
      if (this.selectedRooms.length === 0) {
      this.error = '请先选择直播间';
      return;
      }
      
      if (!confirm(`确定要移除选中的 ${this.selectedRooms.length} 个直播间吗?`)) {
      return;
      }
      
      let successCount = 0;
      let failCount = 0;
      
      // 创建选中房间的副本,因为在移除过程中会修改selectedRooms数组
      const roomsToRemove = [...this.selectedRooms];
      
      for (const roomId of roomsToRemove) {
      try {
          await this.removeRoom(roomId);
          successCount++;
      } catch (error) {
          console.error(`批量移除失败 (房间 ${roomId}):`, error);
          failCount++;
      }
      }
      
      // 清空选中列表
      this.selectedRooms = [];
      this.lastSelectedRoom = null;
      
      this.error = `批量移除完成: 成功 ${successCount} 个, 失败 ${failCount} 个`;
      // 重新加载状态以更新界面
      await this.loadStatus();
    }
}
};
</script>

<style scoped>
.multi-room-manager {
background-color: #1e2127;
border-radius: 8px;
padding: 20px;
color: white;
}

.header {
display: flex;
justify-content: space-between;
align-items: center;
margin-bottom: 20px;
border-bottom: 1px solid #61dafb;
padding-bottom: 10px;
}

.header-actions {
display: flex;
gap: 10px;
}

/* 新增:批量操作栏样式 */
.bulk-action-bar {
display: flex;
justify-content: space-between;
align-items: center;
background-color: #2d3748;
border-radius: 6px;
padding: 10px 15px;
margin-bottom: 15px;
border: 1px solid #4a5568;
}

.bulk-info {
font-weight: bold;
color: #61dafb;
}

.bulk-actions {
display: flex;
gap: 10px;
flex-wrap: wrap;
}

.bulk-record-btn, .bulk-stop-btn, .bulk-pause-btn, .bulk-resume-btn, .bulk-remove-btn, .bulk-clear-btn {
padding: 6px 12px;
border: none;
border-radius: 4px;
cursor: pointer;
font-size: 12px;
min-width: 80px;
}

.bulk-record-btn {
background-color: #4caf50;
color: white;
}

.bulk-stop-btn {
background-color: #ff9800;
color: white;
}

.bulk-pause-btn {
background-color: #ff5722;
color: white;
}

.bulk-resume-btn {
background-color: #2196f3;
color: white;
}

.bulk-remove-btn {
background-color: #f44336;
color: white;
}

.bulk-clear-btn {
background-color: #6c757d;
color: white;
}

.history-btn {
background-color: #2196f3;
color: white;
border: none;
padding: 8px 16px;
border-radius: 4px;
cursor: pointer;
font-weight: bold;
}

.history-btn:hover {
background-color: #1976d2;
}

.parser-btn {
background-color: #ff9800;
color: white;
border: none;
padding: 8px 16px;
border-radius: 4px;
cursor: pointer;
font-weight: bold;
}

.parser-btn:hover {
background-color: #f57c00;
}

.add-btn {
background-color: #61dafb;
color: #282c34;
border: none;
padding: 8px 16px;
border-radius: 4px;
cursor: pointer;
font-weight: bold;
}

.add-btn:hover {
background-color: #4fa8c5;
}

.dialog-overlay {
position: fixed;
top: 0;
left: 0;
right: 0;
bottom: 0;
background-color: rgba(0, 0, 0, 0.5);
display: flex;
justify-content: center;
align-items: center;
z-index: 1000;
}

.dialog {
background-color: #282c34;
border-radius: 8px;
padding: 20px;
width: 400px;
max-width: 90vw;
}

.form-group {
margin-bottom: 15px;
}

.form-group label {
display: block;
margin-bottom: 5px;
color: #61dafb;
}

.input-field {
width: 100%;
padding: 8px;
border: 1px solid #61dafb;
border-radius: 4px;
background-color: #1e2127;
color: white;
box-sizing: border-box;
}

.dialog-actions {
display: flex;
gap: 10px;
margin-top: 20px;
}

.confirm-btn {
background-color: #4caf50;
color: white;
border: none;
padding: 8px 16px;
border-radius: 4px;
cursor: pointer;
flex: 1;
}

.cancel-btn {
background-color: #f44336;
color: white;
border: none;
padding: 8px 16px;
border-radius: 4px;
cursor: pointer;
flex: 1;
}

.room-list {
max-height: 400px;
overflow-y: auto;
}

.room-item {
border: 1px solid #444;
border-radius: 6px;
padding: 15px;
margin-bottom: 10px;
display: flex;
justify-content: space-between;
align-items: flex-start;
cursor: pointer;
transition: background-color 0.2s;
}

.room-item:hover {
background-color: rgba(97, 218, 251, 0.05);
}

.room-item.selected {
border-color: #61dafb;
background-color: rgba(97, 218, 251, 0.15);
}

.status-live {
border-color: #4caf50;
background-color: rgba(76, 175, 80, 0.1);
}

.status-offline {
border-color: #666;
background-color: rgba(102, 102, 102, 0.1);
}

.status-checking {
border-color: #ff9800;
background-color: rgba(255, 152, 0, 0.1);
}

.status-error {
border-color: #f44336;
background-color: rgba(244, 67, 54, 0.1);
}

.status-waiting {
border-color: #2196f3;
background-color: rgba(33, 150, 243, 0.1);
}

.status-paused {
border-color: #ff5722;
background-color: rgba(255, 87, 34, 0.1);
}

.room-selection {
display: flex;
align-items: center;
margin-right: 10px;
}

.room-checkbox {
width: 18px;
height: 18px;
cursor: pointer;
}

.room-info {
flex: 1;
text-align: left;
}

.room-id {
font-weight: bold;
color: #61dafb;
margin-bottom: 5px;
}

.anchor-name {
color: #4caf50;
font-weight: normal;
font-size: 14px;
}

.room-status {
font-size: 14px;
margin-bottom: 5px;
}

.last-check {
color: #888;
font-size: 12px;
}

.popularity {
color: #ff6b6b;
font-weight: bold;
font-size: 13px;
margin-left: 8px;
padding: 2px 6px;
background-color: rgba(255, 107, 107, 0.1);
border-radius: 3px;
}

.room-url {
font-size: 12px;
color: #aaa;
margin-bottom: 5px;
word-break: break-all;
}

.stream-url {
font-size: 11px;
color: #888;
font-family: monospace;
}

.room-actions {
display: flex;
flex-direction: column;
gap: 8px;
}

.play-btn, .record-btn, .stop-record-btn, .pause-btn, .resume-btn, .remove-btn {
padding: 6px 12px;
border: none;
border-radius: 4px;
cursor: pointer;
font-size: 12px;
min-width: 80px;
}

.play-btn {
background-color: #2196f3;
color: white;
}

.record-btn {
background-color: #4caf50;
color: white;
}

.stop-record-btn {
background-color: #ff9800;
color: white;
}

.pause-btn {
background-color: #ff5722;
color: white;
}

.resume-btn {
background-color: #2196f3;
color: white;
}

.remove-btn {
background-color: #f44336;
color: white;
}

.stats {
display: flex;
justify-content: space-around;
margin-top: 20px;
padding-top: 15px;
border-top: 1px solid #444;
}

.stat-item {
text-align: center;
}

.stat-label {
display: block;
font-size: 12px;
color: #aaa;
margin-bottom: 5px;
}

.stat-value {
font-size: 18px;
font-weight: bold;
color: #61dafb;
}

.error-message {
background-color: #f44336;
color: white;
padding: 10px;
border-radius: 4px;
margin-top: 15px;
text-align: center;
}

/* 历史记录样式 */
.history-section {
margin-top: 20px;
border-top: 2px solid #61dafb;
padding-top: 20px;
}

.history-header {
display: flex;
justify-content: space-between;
align-items: center;
margin-bottom: 15px;
}

.history-header h4 {
color: #61dafb;
margin: 0;
}

.history-filters {
display: flex;
gap: 10px;
align-items: center;
}

.refresh-btn {
background-color: #4caf50;
color: white;
border: none;
padding: 5px 10px;
border-radius: 4px;
cursor: pointer;
font-size: 12px;
}

.refresh-btn:hover {
background-color: #45a049;
}

.history-list {
max-height: 400px;
overflow-y: auto;
border: 1px solid #444;
border-radius: 4px;
padding: 10px;
}

.loading, .no-history {
text-align: center;
color: #aaa;
padding: 20px;
}

.history-item {
padding: 10px;
margin-bottom: 8px;
border-radius: 4px;
border-left: 4px solid #61dafb;
background-color: rgba(97, 218, 251, 0.1);
}

.history-info {
text-align: left;
}

.history-main {
display: flex;
align-items: center;
gap: 8px;
margin-bottom: 5px;
}

.anchor-name {
font-weight: bold;
color: #61dafb;
}

.room-url {
color: #aaa;
font-size: 12px;
word-break: break-all;
}

.history-time {
color: #888;
font-size: 11px;
}

.load-more {
text-align: center;
margin-top: 15px;
}

.load-more-btn {
background-color: #61dafb;
color: #282c34;
border: none;
padding: 8px 16px;
border-radius: 4px;
cursor: pointer;
font-weight: bold;
}

.load-more-btn:hover {
background-color: #4fa8c5;
}

/* 新增:播放器模态框样式 */
.player-modal {
position: fixed;
top: 0;
left: 0;
width: 100%;
height: 100%;
background-color: rgba(0, 0, 0, 0.8);
display: flex;
justify-content: center;
align-items: center;
z-index: 2000;
}

.player-content {
background-color: #282c34;
border-radius: 8px;
padding: 20px;
width: 80%;
max-width: 800px;
max-height: 80vh;
}

.player-header {
display: flex;
justify-content: space-between;
align-items: center;
margin-bottom: 15px;
}

.player-header h3 {
margin: 0;
color: #61dafb;
}

.close-btn {
background-color: #f44336;
color: white;
border: none;
padding: 5px 10px;
border-radius: 4px;
cursor: pointer;
}

.modal-video-player {
width: 100%;
height: auto;
max-height: 60vh;
background-color: #000;
border-radius: 4px;
}

.players-section {
margin-top: 20px;
border-top: 2px solid #61dafb;
padding-top: 20px;
}

.players-section h3 {
color: #61dafb;
margin-bottom: 15px;
}

.players-container {
display: flex;
flex-wrap: wrap;
gap: 20px;
}

.player-wrapper {
flex: 1;
min-width: 300px;
background-color: #2d3748;
border-radius: 8px;
padding: 15px;
box-sizing: border-box;
}

.player-header {
display: flex;
justify-content: space-between;
align-items: center;
margin-bottom: 10px;
}

.player-title {
font-weight: bold;
color: #61dafb;
}

.close-player-btn {
background-color: #f44336;
color: white;
border: none;
width: 24px;
height: 24px;
border-radius: 50%;
cursor: pointer;
font-size: 16px;
display: flex;
align-items: center;
justify-content: center;
}

.player-controls {
display: flex;
gap: 10px;
margin-bottom: 10px;
}

.mute-btn, .play-btn {
padding: 5px 10px;
border: none;
border-radius: 4px;
cursor: pointer;
font-size: 12px;
}

.mute-btn {
background-color: #ff9800;
color: white;
}

.play-btn {
background-color: #4caf50;
color: white;
}

.inline-video-player {
width: 100%;
height: 200px;
background-color: #000;
border-radius: 4px;
}

.player-error {
color: #f44336;
text-align: center;
padding: 10px;
margin-top: 10px;
border: 1px solid #f44336;
border-radius: 4px;
background-color: rgba(244, 67, 54, 0.1);
}

.no-players {
color: #888;
font-style: italic;
text-align: center;
padding: 20px;
width: 100%;
}
</style>
复制代码
页: [1]
查看完整版本: 抖音直播无人值守全天候轮询录制工具2.0