Christian Krakau-Louis
75 lines
2.8 KiB
Python
75 lines
2.8 KiB
Python
import os
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
# Define base directory
|
|
BASE_DIR = Path(__file__).parent.parent
|
|
|
|
def update_pot_file():
|
|
"""Extract translatable strings from templates and create a POT file"""
|
|
print("Extracting strings from templates...")
|
|
|
|
pot_file = BASE_DIR / "app" / "i18n" / "messages.pot"
|
|
template_dir = BASE_DIR / "app" / "templates"
|
|
|
|
try:
|
|
subprocess.run([
|
|
"pybabel", "extract",
|
|
"-F", str(BASE_DIR / "babel.cfg"),
|
|
"-o", str(pot_file),
|
|
str(template_dir)
|
|
], check=True)
|
|
print("Successfully created template file")
|
|
except subprocess.CalledProcessError as e:
|
|
print(f"Error creating template file: {e}")
|
|
|
|
def update_po_files():
|
|
"""Update all .po files with the strings from the POT file"""
|
|
print("Updating .po files...")
|
|
|
|
pot_file = BASE_DIR / "app" / "i18n" / "messages.pot"
|
|
|
|
for lang_dir in (BASE_DIR / "app" / "i18n" / "locales").iterdir():
|
|
if lang_dir.is_dir():
|
|
lang_code = lang_dir.name
|
|
po_file = lang_dir / "LC_MESSAGES" / "messages.po"
|
|
|
|
if po_file.exists():
|
|
print(f"Updating {lang_code} translations...")
|
|
try:
|
|
subprocess.run([
|
|
"pybabel", "update",
|
|
"-i", str(pot_file),
|
|
"-o", str(po_file),
|
|
"-l", lang_code
|
|
], check=True)
|
|
print(f"Successfully updated {lang_code} translations")
|
|
except subprocess.CalledProcessError as e:
|
|
print(f"Error updating {lang_code} translations: {e}")
|
|
else:
|
|
print(f"Initializing {lang_code} translations...")
|
|
# Create LC_MESSAGES directory if it doesn't exist
|
|
os.makedirs(lang_dir / "LC_MESSAGES", exist_ok=True)
|
|
try:
|
|
subprocess.run([
|
|
"pybabel", "init",
|
|
"-i", str(pot_file),
|
|
"-o", str(po_file),
|
|
"-l", lang_code
|
|
], check=True)
|
|
print(f"Successfully initialized {lang_code} translations")
|
|
except subprocess.CalledProcessError as e:
|
|
print(f"Error initializing {lang_code} translations: {e}")
|
|
|
|
if __name__ == "__main__":
|
|
# Create babel.cfg if it doesn't exist
|
|
babel_cfg = BASE_DIR / "babel.cfg"
|
|
if not babel_cfg.exists():
|
|
with open(babel_cfg, "w") as f:
|
|
f.write("[python: **.py]\n")
|
|
f.write("[jinja2: **/templates/**.html]\n")
|
|
f.write("extensions=jinja2.ext.autoescape,jinja2.ext.with_\n")
|
|
|
|
update_pot_file()
|
|
update_po_files()
|