#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
检查图片文件的内容
"""

import os
import re
from PIL import Image


def check_image_content():
    """
    检查图片文件的内容
    """
    # 获取图片文件
    img_dir = "我们需要知识分子吗_思考_images"
    img_files = []
    for filename in os.listdir(img_dir):
        if filename.endswith('.png') and filename.startswith('vertical_subtitle_'):
            match = re.match(r'vertical_subtitle_(\d+)\.png', filename)
            if match:
                idx = int(match.group(1))
                img_files.append((idx, filename))
    
    img_files.sort(key=lambda x: x[0])
    
    print(f"找到 {len(img_files)} 个图片文件")
    print("\n前10个图片文件:")
    for idx, filename in img_files[:10]:
        print(f"idx: {idx}, filename: {filename}")
        
        # 尝试读取图片
        try:
            img_path = os.path.join(img_dir, filename)
            img = Image.open(img_path)
            print(f"  尺寸: {img.size}")
        except Exception as e:
            print(f"  错误: {e}")


if __name__ == "__main__":
    check_image_content()