#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
详细检查字幕和音频的对应关系
"""

import os
import sqlite3
import re


def detailed_check():
    """
    详细检查字幕和音频的对应关系
    """
    # 获取字幕数据
    db_path = r"\\ll\\d\\xtrssvjj\\dpd.db"
    conn = sqlite3.connect(db_path)
    cur = conn.cursor()
    
    cur.execute("SELECT idx, cn, en, start, end FROM srts WHERE adid=? ORDER BY idx LIMIT 10", (8407,))
    srtsdata = cur.fetchall()
    conn.close()
    
    print("前10条字幕详细信息:")
    for idx, cn, en, start, end in srtsdata:
        print(f"\nidx: {idx}")
        print(f"  cn: {cn}")
        print(f"  en: {en}")
        print(f"  start: {start}ms, end: {end}ms")
        print(f"  duration: {end-start}ms")
        
        # 检查对应的中文音频文件
        seg_dir = "我们需要知识分子吗_思考_segments"
        seg_files = []
        for filename in os.listdir(seg_dir):
            if filename.endswith('.mp3') and filename.startswith(f'seg{idx:03d}-'):
                seg_files.append(filename)
        
        if seg_files:
            print(f"  中文音频文件: {seg_files[0]}")
        else:
            print(f"  中文音频文件: 不存在")
        
        # 检查对应的TTS音频文件
        tts_dir = "我们需要知识分子吗_思考_tts"
        tts_file = os.path.join(tts_dir, f"tts_{idx}.mp3")
        if os.path.exists(tts_file):
            print(f"  TTS音频文件: 存在")
        else:
            print(f"  TTS音频文件: 不存在")
        
        # 检查对应的图片文件
        img_dir = "我们需要知识分子吗_思考_images"
        img_file = os.path.join(img_dir, f"vertical_subtitle_{idx:03d}.png")
        if os.path.exists(img_file):
            print(f"  图片文件: 存在")
        else:
            print(f"  图片文件: 不存在")


if __name__ == "__main__":
    detailed_check()