import os
import subprocess

# 配置变量
POT_FILE = 'messages.pot'
LOCALES_DIR = 'locales'
SRC_DIR = 'src'

def run_command(command):
    """运行系统命令"""
    try:
        subprocess.check_call(command, shell=True)
    except subprocess.CalledProcessError as e:
        print(f"执行时出错 {' '.join(command)}: {e}")

def get_babel_cfg():

    command = f"pybabel extract -F babel.cfg -o messages.pot ."
def extract_pot_file():
    """提取可翻译的文本生成.pot文件"""
    command = f"pybabel extract -F babel.cfg -o {POT_FILE} {SRC_DIR}"
    run_command(command)

def init_po_file(lang_code):
    """初始化翻译文件生成.po文件"""
    po_file = os.path.join(LOCALES_DIR, lang_code, 'LC_MESSAGES', 'messages.po')
    command = f"pybabel init -i {POT_FILE} -d {LOCALES_DIR} -l {lang_code}"
    run_command(command)

def update_po_files():
    """更新翻译文件"""
    command = f"pybabel update -i {POT_FILE} -d {LOCALES_DIR}"
    run_command(command)

def compile_po_files():
    """编译.po文件生成.mo文件"""
    command = f"pybabel compile -d {LOCALES_DIR}"
    run_command(command)

def main():
    # 提取.pot文件
    extract_pot_file()
    #
    # # 初始化.po文件，这里以中文为例
    # init_po_file('zh_CN')
    #
    # 更新.po文件
    update_po_files()

    # 编译.po文件
    compile_po_files()

if __name__ == '__main__':
    main()

root:.
│   babel.cfg
│   babel_use.py
│             
├──src
   │   main.py
   │   setup.py
   │
   ├───assets
   │   └───dark_ui
   │
   ├───functions
   └───uis

# 提取.pot文件
extract_pot_file()
#
# 初始化.po文件，这里以中文为例
init_po_file('zh_CN')

# 编译.po文件
compile_po_files()

# 提取.pot文件
extract_pot_file()

# 更新.po文件
update_po_files()

root:.
│   babel.cfg
│   babel_use.py
│   messages.pot
│   
│         
├───locales
│   └───zh_CN
│       └───LC_MESSAGES
│               messages.mo
│               messages.po
│             
├───src
│   │   main.py
│   │   setup.py
│   │
│   ├───assets
│   │   └───dark_ui
│   │
│   ├───functions
│   └───uis

└───__pycache__