112 lines
3.5 KiB
Python
112 lines
3.5 KiB
Python
import os
|
||
import json
|
||
|
||
# 支持的图片扩展名（不区分大小写）
|
||
SUPPORTED_EXTENSIONS = {'.png', '.jpg', '.jpeg', '.gif', '.webp', '.bmp'}
|
||
|
||
def is_image_file(filename):
|
||
"""检查文件是否是支持的图片格式"""
|
||
return os.path.splitext(filename)[1].lower() in SUPPORTED_EXTENSIONS
|
||
|
||
def generate_index_json(image_files):
|
||
"""生成index.json文件内容"""
|
||
return {"0": sorted(image_files)}
|
||
|
||
def generate_info_json(image_files, category_name, icon_file):
|
||
"""生成info.json文件内容"""
|
||
return {
|
||
"name": category_name,
|
||
"icon": icon_file,
|
||
"items": sorted(image_files)
|
||
}
|
||
|
||
def process_directory(category_path, overwrite=False):
|
||
"""处理单个目录"""
|
||
# 获取目录中所有支持的图片文件
|
||
image_files = [
|
||
f for f in os.listdir(category_path)
|
||
if is_image_file(f) and os.path.isfile(os.path.join(category_path, f))
|
||
]
|
||
|
||
if not image_files:
|
||
print(f'警告: 目录 {category_path} 中没有支持的图片文件')
|
||
return False
|
||
|
||
# 文件路径
|
||
index_path = os.path.join(category_path, 'index.json')
|
||
info_path = os.path.join(category_path, 'info.json')
|
||
|
||
# 如果不覆盖且文件已存在，则跳过
|
||
if not overwrite and (os.path.exists(index_path) or os.path.exists(info_path)):
|
||
print(f'已跳过目录(文件已存在): {category_path}')
|
||
return False
|
||
|
||
# 选择第一个存在的图片文件作为icon
|
||
icon_file = next(
|
||
(f for f in image_files if os.path.exists(os.path.join(category_path, f))),
|
||
image_files[0]
|
||
)
|
||
|
||
try:
|
||
# 生成index.json（保持原始字符）
|
||
with open(index_path, 'w', encoding='utf-8') as f:
|
||
json.dump(generate_index_json(image_files), f, ensure_ascii=False, indent=2)
|
||
|
||
# 生成info.json（强制Unicode转义）
|
||
with open(info_path, 'w', encoding='ascii') as f:
|
||
json.dump(
|
||
generate_info_json(image_files, os.path.basename(category_path), icon_file),
|
||
f,
|
||
ensure_ascii=True,
|
||
indent=2
|
||
)
|
||
|
||
print(f'成功处理目录: {category_path}')
|
||
print(f' 发现 {len(image_files)} 个图片文件')
|
||
print(f' 使用图标: {icon_file}')
|
||
return True
|
||
|
||
except Exception as e:
|
||
print(f'处理目录 {category_path} 时出错: {str(e)}')
|
||
return False
|
||
|
||
def main():
|
||
print("=== Waline 表情包JSON生成器 ===")
|
||
print("将遍历emoji目录并生成index.json和info.json")
|
||
|
||
# 询问是否覆盖已存在文件
|
||
overwrite = False
|
||
while True:
|
||
choice = input("是否重新生成所有已存在的JSON文件？(y/n): ").lower()
|
||
if choice == 'y':
|
||
overwrite = True
|
||
break
|
||
elif choice == 'n':
|
||
break
|
||
print("请输入 y 或 n")
|
||
|
||
root_dir = 'emoji'
|
||
if not os.path.exists(root_dir):
|
||
print(f"错误: 目录 {root_dir} 不存在")
|
||
return
|
||
|
||
processed = 0
|
||
skipped = 0
|
||
errors = 0
|
||
|
||
print("\n开始处理目录...")
|
||
for category in os.listdir(root_dir):
|
||
category_path = os.path.join(root_dir, category)
|
||
if os.path.isdir(category_path):
|
||
if process_directory(category_path, overwrite):
|
||
processed += 1
|
||
else:
|
||
skipped += 1
|
||
|
||
print("\n=== 处理完成 ===")
|
||
print(f"已处理目录: {processed}")
|
||
print(f"已跳过目录: {skipped}")
|
||
print(f"错误目录: {errors}")
|
||
|
||
if __name__ == '__main__':
|
||
main() |