#!/usr/bin/env python3
"""
统计 generated_icons 目录下各个子目录中的 PNG 文件数量
"""

import os
from pathlib import Path

def count_png_files():
    # 设置 generated_icons 目录路径
    base_dir = Path(__file__).parent / "generated_icons"
    
    if not base_dir.exists():
        print(f"错误: 目录 {base_dir} 不存在")
        return
    
    # 存储每个子目录的统计结果
    subdirs_stats = {}
    total_png_count = 0
    
    # 遍历所有子目录
    for subdir in sorted(base_dir.iterdir()):
        if subdir.is_dir():
            # 统计当前子目录下的 PNG 文件数量
            png_files = list(subdir.glob("*.png"))
            png_count = len(png_files)
            
            subdirs_stats[subdir.name] = png_count
            total_png_count += png_count
    
    # 打印结果
    print("=" * 50)
    print(f"{'子目录':<20} {'PNG 文件数量':>15}")
    print("=" * 50)
    
    for subdir_name, count in subdirs_stats.items():
        print(f"{subdir_name:<20} {count:>15}")
    
    print("=" * 50)
    print(f"{'总计':<20} {total_png_count:>15}")
    print("=" * 50)
    
    # 额外统计信息
    print(f"\n子目录总数: {len(subdirs_stats)}")
    print(f"PNG 文件总数: {total_png_count}")
    
    if subdirs_stats:
        avg_png_per_dir = total_png_count / len(subdirs_stats)
        print(f"平均每个子目录的 PNG 数量: {avg_png_per_dir:.2f}")

if __name__ == "__main__":
    count_png_files()