mirror of
https://github.com/home-assistant/core.git
synced 2026-04-22 16:05:45 +00:00
99 lines
3.1 KiB
Python
99 lines
3.1 KiB
Python
"""Compile the current translation strings files for testing."""
|
|
|
|
import argparse
|
|
from pathlib import Path
|
|
import sys
|
|
|
|
import orjson
|
|
|
|
from . import download, upload
|
|
from .const import INTEGRATIONS_DIR
|
|
from .util import flatten_translations, get_base_arg_parser, substitute_references
|
|
|
|
|
|
def valid_integration(integration):
|
|
"""Test if it's a valid integration."""
|
|
if not (INTEGRATIONS_DIR / integration).is_dir():
|
|
raise argparse.ArgumentTypeError(
|
|
f"The integration {integration} does not exist."
|
|
)
|
|
|
|
return integration
|
|
|
|
|
|
def get_arguments() -> argparse.Namespace:
|
|
"""Get parsed passed in arguments."""
|
|
parser = get_base_arg_parser()
|
|
parser.add_argument(
|
|
"--integration", type=valid_integration, help="Integration to process."
|
|
)
|
|
parser.add_argument("--all", action="store_true", help="Process all integrations.")
|
|
return parser.parse_args()
|
|
|
|
|
|
def prepare_download_dir():
|
|
"""Ensure the download directory exists and is empty."""
|
|
if download.DOWNLOAD_DIR.is_dir():
|
|
for lang_file in download.DOWNLOAD_DIR.glob("*.json"):
|
|
lang_file.unlink()
|
|
else:
|
|
download.DOWNLOAD_DIR.mkdir(parents=True)
|
|
|
|
|
|
def run_translation(component_translations, flattened_translations):
|
|
"""Run the translation process for the given components."""
|
|
for integration in component_translations:
|
|
print(f"Generating translations for {integration}")
|
|
component_translations[integration] = substitute_references(
|
|
component_translations[integration],
|
|
flattened_translations,
|
|
fail_on_missing=True,
|
|
)
|
|
|
|
(download.DOWNLOAD_DIR / "en.json").write_bytes(
|
|
orjson.dumps({"component": component_translations})
|
|
)
|
|
download.save_integrations_translations()
|
|
|
|
|
|
def run_single(translations, flattened_translations, integration):
|
|
"""Run the script for a single integration."""
|
|
component_translations = translations["component"]
|
|
if integration not in component_translations:
|
|
print("Integration has no strings.json")
|
|
sys.exit(1)
|
|
|
|
prepare_download_dir()
|
|
run_translation(
|
|
{integration: component_translations[integration]}, flattened_translations
|
|
)
|
|
|
|
|
|
def run():
|
|
"""Run the script."""
|
|
args = get_arguments()
|
|
translations = upload.generate_upload_data()
|
|
flattened_translations = flatten_translations(translations)
|
|
|
|
if args.all:
|
|
prepare_download_dir()
|
|
run_translation(translations["component"], flattened_translations)
|
|
print("🌎 Generated translation files for all integrations")
|
|
return 0
|
|
|
|
if args.integration:
|
|
integration = args.integration
|
|
else:
|
|
integration = None
|
|
while (
|
|
integration is None
|
|
or not Path(f"homeassistant/components/{integration}").is_dir()
|
|
):
|
|
if integration is not None:
|
|
print(f"Integration {integration} doesn't exist!")
|
|
print()
|
|
integration = input("Integration to process: ")
|
|
|
|
run_single(translations, flattened_translations, integration)
|
|
return 0
|