#!/usr/bin/env python3
"""
生成所有字幕图片
"""
from PIL import Image, ImageDraw, ImageFont
import sqlite3
import os

# 数据库路径
db_path = "\\\\ll\\D\\xtrssvjj\\dpd.db"

# 输出目录
output_dir = "shitu_frames"
if not os.path.exists(output_dir):
    os.makedirs(output_dir)

# 创建黑色背景图片 (720x1280)
width, height = 720, 1280

# 尝试使用系统中文字体
font_paths = [
    "C:/Windows/Fonts/msyh.ttc",      # 微软雅黑
    "C:/Windows/Fonts/simhei.ttf",    # 黑体
    "C:/Windows/Fonts/simsun.ttc",    # 宋体
]

title_font = None
subtitle_font = None
cn_font = None
en_font = None

for font_path in font_paths:
    if os.path.exists(font_path):
        try:
            title_font = ImageFont.truetype(font_path, 36)
            subtitle_font = ImageFont.truetype(font_path, 24)
            cn_font = ImageFont.truetype(font_path, 32)
            en_font = ImageFont.truetype(font_path, 24)
            print(f"使用字体: {font_path}")
            break
        except:
            continue

if title_font is None:
    print("未找到中文字体，使用默认字体")
    title_font = ImageFont.load_default()
    subtitle_font = ImageFont.load_default()
    cn_font = ImageFont.load_default()
    en_font = ImageFont.load_default()

# 固定文字内容
title = "师徒一场_张雪峰走好_上"
subtitle = "源自：@梁老师的故事汇"

# 查询所有字幕
try:
    conn = sqlite3.connect(db_path)
    cursor = conn.cursor()
    
    cursor.execute("SELECT idx, cn, en FROM srts WHERE adid = 8281 ORDER BY idx ASC")
    results = cursor.fetchall()
    
    print(f"找到 {len(results)} 条字幕")
    
    for idx, cn, en in results:
        # 创建新图片
        img = Image.new('RGB', (width, height), color='black')
        draw = ImageDraw.Draw(img)
        
        # 标题位置
        title_bbox = draw.textbbox((0, 0), title, font=title_font)
        title_width = title_bbox[2] - title_bbox[0]
        title_x = (width - title_width) // 2
        title_y = 150
        
        # 副标题位置
        subtitle_bbox = draw.textbbox((0, 0), subtitle, font=subtitle_font)
        subtitle_width = subtitle_bbox[2] - subtitle_bbox[0]
        subtitle_x = (width - subtitle_width) // 2
        subtitle_y = 220
        
        # 中文字幕位置
        cn_bbox = draw.textbbox((0, 0), cn, font=cn_font)
        cn_width = cn_bbox[2] - cn_bbox[0]
        cn_x = (width - cn_width) // 2
        cn_y = height // 2 - 40
        
        # 英文字幕位置
        en_bbox = draw.textbbox((0, 0), en, font=en_font)
        en_width = en_bbox[2] - en_bbox[0]
        en_x = (width - en_width) // 2
        en_y = height // 2 + 40
        
        # 绘制文字
        draw.text((title_x, title_y), title, font=title_font, fill='white')
        draw.text((subtitle_x, subtitle_y), subtitle, font=subtitle_font, fill='gray')
        draw.text((cn_x, cn_y), cn, font=cn_font, fill='yellow')
        draw.text((en_x, en_y), en, font=en_font, fill='white')
        
        # 保存图片
        output_path = os.path.join(output_dir, f"frame_{idx:03d}.jpg")
        img.save(output_path, quality=95)
        
        if (idx + 1) % 50 == 0:
            print(f"已生成 {idx + 1} 张图片")
    
    conn.close()
    
    print(f"完成！共生成 {len(results)} 张图片")
    print(f"图片保存在: {output_dir}")
    
except Exception as e:
    print(f"错误: {e}")