66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
#!.trans/bin/python
|
|
'''
|
|
Created on 11 mai 2025
|
|
## pip install argostranslate polib
|
|
@author: denis
|
|
'''
|
|
import argparse
|
|
import os
|
|
import polib
|
|
import argostranslate.package
|
|
import argostranslate.translate
|
|
|
|
DJANGO_PROJECT = "test_tube_scanner"
|
|
|
|
def install_translation_model(from_lang="en", to_lang="fr"):
|
|
print("Cheching translation model...")
|
|
argostranslate.package.update_package_index()
|
|
available_packages = argostranslate.package.get_available_packages()
|
|
|
|
try:
|
|
package_to_install = next(
|
|
p for p in available_packages if p.from_code == from_lang and p.to_code == to_lang
|
|
)
|
|
print(f"Loading model {from_lang} >> {to_lang}...")
|
|
argostranslate.package.install_from_path(package_to_install.download())
|
|
print("Installation model ")
|
|
except StopIteration:
|
|
print("No models found for this language pair.")
|
|
exit(1)
|
|
|
|
|
|
def translate_po_file(input_path, output_path, from_lang="en", to_lang="fr"):
|
|
print(f"Loading file {input_path}...")
|
|
po = polib.pofile(input_path)
|
|
|
|
print("Start traduction...")
|
|
for entry in po:
|
|
if entry.msgstr.strip() == "" and entry.msgid.strip():
|
|
try:
|
|
translation = argostranslate.translate.translate(entry.msgid, from_lang, to_lang)
|
|
entry.msgstr = translation
|
|
except Exception as e:
|
|
print(f"Translation error: {entry.msgid}\n{e}")
|
|
continue
|
|
|
|
po.save(output_path)
|
|
print(f"End of translation. Output file in: {output_path}")
|
|
|
|
|
|
def main(app, from_lang, to_lang):
|
|
INPUT_FILE = f'{app}/locale/{from_lang}/LC_MESSAGES/django.po'
|
|
OUTPUT_FILE = f'{app}/locale/{to_lang}/LC_MESSAGES/django.po'
|
|
install_translation_model(from_lang, to_lang)
|
|
translate_po_file(INPUT_FILE, OUTPUT_FILE, from_lang, to_lang)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Translation for Django")
|
|
parser.add_argument("--app", type=str, default='home', help="Django application name")
|
|
parser.add_argument("--from_lang", type=str, default="fr", help="From language")
|
|
parser.add_argument("--to_lang", type=str, default='en', help="To language", required=True)
|
|
args = parser.parse_args()
|
|
|
|
main(f'{DJANGO_PROJECT}/{args.app}', args.from_lang, args.to_lang)
|
|
exit(0)
|