146 lines
4.9 KiB
Python
146 lines
4.9 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""将 PO 文件编译为 MO 文件，支持模块化 PO 文件合并
|
||
|
||
使用方法:
|
||
python tools/i18n/build_mo.py
|
||
|
||
功能:
|
||
1. 扫描 src/i18n/locales 下的语言目录
|
||
2. 如果存在 modules/ 目录，将所有 .po 文件合并
|
||
3. 生成/更新 LC_MESSAGES/messages.po (作为合并后的完整文件)
|
||
4. 编译生成 LC_MESSAGES/messages.mo
|
||
5. 同时处理其他独立的 .po 文件 (如 game_configs.po)
|
||
"""
|
||
|
||
import sys
|
||
import shutil
|
||
from pathlib import Path
|
||
|
||
def compile_modules(lang_dir: Path) -> bool:
|
||
"""
|
||
编译该语言目录下的模块
|
||
|
||
Args:
|
||
lang_dir: 语言目录，如 src/i18n/locales/zh_CN
|
||
|
||
Returns:
|
||
bool: 是否成功
|
||
"""
|
||
try:
|
||
import polib
|
||
except ImportError:
|
||
print("[ERROR] polib 库未安装。请运行 pip install polib 安装。")
|
||
return False
|
||
|
||
lc_messages_dir = lang_dir / "LC_MESSAGES"
|
||
lc_messages_dir.mkdir(parents=True, exist_ok=True)
|
||
|
||
modules_dir = lang_dir / "modules"
|
||
|
||
# 1. 处理 messages.po (可能来自 modules 拆分，也可能就是单个文件)
|
||
if modules_dir.exists() and list(modules_dir.glob("*.po")):
|
||
print(f" [合并] 发现模块化文件: {lang_dir.name}/modules/ -> messages.po")
|
||
|
||
# 创建一个新的 PO 对象
|
||
merged_po = polib.POFile()
|
||
|
||
# 收集所有模块文件
|
||
module_files = sorted(list(modules_dir.glob("*.po")))
|
||
|
||
# 为了保留 metadata，尝试先读取 common.po 或 misc.po，或者直接从第一个文件读
|
||
# 最好是如果你有专门的 header.po，这里我们直接用第一个文件的 metadata
|
||
if module_files:
|
||
try:
|
||
first_po = polib.pofile(str(module_files[0]))
|
||
merged_po.metadata = first_po.metadata
|
||
except:
|
||
pass
|
||
|
||
# 遍历合并
|
||
count = 0
|
||
for po_file in module_files:
|
||
try:
|
||
po = polib.pofile(str(po_file))
|
||
for entry in po:
|
||
# 检查是否重复 (简单去重)
|
||
# 注意：如果 key 重复，polib 默认可能不会报错，但这里我们 append
|
||
# 真实场景可能需要 merge 逻辑，这里假设 modules 之间无交集
|
||
merged_po.append(entry)
|
||
count += 1
|
||
except Exception as e:
|
||
print(f" [警告] 读取 {po_file.name} 失败: {e}")
|
||
|
||
print(f" - 合并了 {count} 个文件，共 {len(merged_po)} 条目")
|
||
|
||
# 保存合并后的 messages.po (方便查看和作为中间文件)
|
||
merged_po_path = lc_messages_dir / "messages.po"
|
||
merged_po.save(str(merged_po_path))
|
||
|
||
# 编译为 MO
|
||
merged_mo_path = lc_messages_dir / "messages.mo"
|
||
merged_po.save_as_mofile(str(merged_mo_path))
|
||
print(f" - 生成: {merged_mo_path.name}")
|
||
|
||
else:
|
||
# 没有 modules，尝试直接编译现有的 messages.po
|
||
legacy_po = lc_messages_dir / "messages.po"
|
||
if legacy_po.exists():
|
||
print(f" [编译] {lang_dir.name}/messages.po")
|
||
try:
|
||
po = polib.pofile(str(legacy_po))
|
||
po.save_as_mofile(str(legacy_po.with_suffix('.mo')))
|
||
except Exception as e:
|
||
print(f" [错误] {e}")
|
||
return False
|
||
|
||
# 2. 处理其他独立的 PO 文件 (如 game_configs.po)
|
||
# 只要不是 messages.po (上面已经处理过)，都单独编译
|
||
for po_file in lc_messages_dir.glob("*.po"):
|
||
if po_file.name == "messages.po":
|
||
continue
|
||
|
||
print(f" [编译] {po_file.name}")
|
||
try:
|
||
po = polib.pofile(str(po_file))
|
||
po.save_as_mofile(str(po_file.with_suffix('.mo')))
|
||
except Exception as e:
|
||
print(f" [错误] {e}")
|
||
|
||
return True
|
||
|
||
def main():
|
||
print("="*60)
|
||
print("构建 i18n 文件 (Module -> PO -> MO)")
|
||
print("="*60)
|
||
|
||
# 查找项目根目录
|
||
script_dir = Path(__file__).parent
|
||
project_root = script_dir.parent.parent
|
||
i18n_dir = project_root / "static" / "locales"
|
||
|
||
if not i18n_dir.exists():
|
||
print(f"[ERROR] 找不到 i18n 目录: {i18n_dir}")
|
||
sys.exit(1)
|
||
|
||
# 遍历语言目录
|
||
success = True
|
||
for lang_dir in i18n_dir.iterdir():
|
||
if not lang_dir.is_dir() or lang_dir.name == "templates":
|
||
continue
|
||
|
||
print(f"\n处理语言: {lang_dir.name}")
|
||
if not compile_modules(lang_dir):
|
||
success = False
|
||
|
||
print("-" * 60)
|
||
if success:
|
||
print("\n[OK] 所有语言包构建完成")
|
||
return 0
|
||
else:
|
||
print("\n[FAIL] 构建过程中出现错误")
|
||
return 1
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|