82 lines
2.7 KiB
Python
82 lines
2.7 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""Merge all translation sources into a single JSON file."""
|
||
|
|
import glob
|
||
|
|
import os
|
||
|
|
import re
|
||
|
|
|
||
|
|
from homeassistant.util import json as json_util
|
||
|
|
|
||
|
|
FILENAME_FORMAT = re.compile(r'strings\.(?P<suffix>\w+)\.json')
|
||
|
|
|
||
|
|
|
||
|
|
def get_language(path):
|
||
|
|
"""Get the language code for the given file path."""
|
||
|
|
return os.path.splitext(os.path.basename(path))[0]
|
||
|
|
|
||
|
|
|
||
|
|
def get_component_path(lang, component):
|
||
|
|
"""Get the component translation path."""
|
||
|
|
if os.path.isdir(os.path.join("homeassistant", "components", component)):
|
||
|
|
return os.path.join(
|
||
|
|
"homeassistant", "components", component, ".translations",
|
||
|
|
"{}.json".format(lang))
|
||
|
|
else:
|
||
|
|
return os.path.join(
|
||
|
|
"homeassistant", "components", ".translations",
|
||
|
|
"{}.{}.json".format(component, lang))
|
||
|
|
|
||
|
|
|
||
|
|
def get_platform_path(lang, component, platform):
|
||
|
|
"""Get the platform translation path."""
|
||
|
|
if os.path.isdir(os.path.join(
|
||
|
|
"homeassistant", "components", component, platform)):
|
||
|
|
return os.path.join(
|
||
|
|
"homeassistant", "components", component, platform,
|
||
|
|
".translations", "{}.json".format(lang))
|
||
|
|
else:
|
||
|
|
return os.path.join(
|
||
|
|
"homeassistant", "components", component, ".translations",
|
||
|
|
"{}.{}.json".format(platform, lang))
|
||
|
|
|
||
|
|
|
||
|
|
def get_component_translations(translations):
|
||
|
|
"""Get the component level translations."""
|
||
|
|
translations = translations.copy()
|
||
|
|
translations.pop('platform', None)
|
||
|
|
|
||
|
|
return translations
|
||
|
|
|
||
|
|
|
||
|
|
def save_language_translations(lang, translations):
|
||
|
|
"""Distribute the translations for this language."""
|
||
|
|
components = translations.get('component', {})
|
||
|
|
for component, component_translations in components.items():
|
||
|
|
base_translations = get_component_translations(component_translations)
|
||
|
|
if base_translations:
|
||
|
|
path = get_component_path(lang, component)
|
||
|
|
os.makedirs(os.path.dirname(path), exist_ok=True)
|
||
|
|
json_util.save_json(path, base_translations)
|
||
|
|
|
||
|
|
for platform, platform_translations in component_translations.get(
|
||
|
|
'platform', {}).items():
|
||
|
|
path = get_platform_path(lang, component, platform)
|
||
|
|
os.makedirs(os.path.dirname(path), exist_ok=True)
|
||
|
|
json_util.save_json(path, platform_translations)
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
"""Main section of the script."""
|
||
|
|
if not os.path.isfile("requirements_all.txt"):
|
||
|
|
print("Run this from HA root dir")
|
||
|
|
return
|
||
|
|
|
||
|
|
paths = glob.iglob("build/translations-download/*.json")
|
||
|
|
for path in paths:
|
||
|
|
lang = get_language(path)
|
||
|
|
translations = json_util.load_json(path)
|
||
|
|
save_language_translations(lang, translations)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|