Add a script to clean the frontend translations (#34309)

script/translations/clean.py

@@ -1,8 +1,20 @@
 """Find translation keys that are in Lokalise but no longer defined in source."""
+import argparse
 import json
 
-from .const import INTEGRATIONS_DIR
+from .const import CORE_PROJECT_ID, FRONTEND_DIR, FRONTEND_PROJECT_ID, INTEGRATIONS_DIR
+from .error import ExitApp
 from .lokalise import get_api
+from .util import get_base_arg_parser
+
+
+def get_arguments() -> argparse.Namespace:
+    """Get parsed passed in arguments."""
+    parser = get_base_arg_parser()
+    parser.add_argument(
+        "--target", type=str, default="core", choices=["core", "frontend"],
+    )
+    return parser.parse_args()
 
 
 def find_extra(base, translations, path_prefix, missing_keys):
@@ -19,8 +31,8 @@ def find_extra(base, translations, path_prefix, missing_keys):
             missing_keys.append(cur_path)
 
 
-def find():
+def find_core():
-    """Find all missing keys."""
+    """Find all missing keys in core."""
     missing_keys = []
 
     for int_dir in INTEGRATIONS_DIR.iterdir():
@@ -41,16 +53,38 @@ def find():
     return missing_keys
 
 
+def find_frontend():
+    """Find all missing keys in frontend."""
+    if not FRONTEND_DIR.is_dir():
+        raise ExitApp(f"Unable to find frontend at {FRONTEND_DIR}")
+
+    source = FRONTEND_DIR / "src/translations/en.json"
+    translated = FRONTEND_DIR / "translations/en.json"
+
+    missing_keys = []
+    find_extra(
+        json.loads(source.read_text()),
+        json.loads(translated.read_text()),
+        "",
+        missing_keys,
+    )
+    return missing_keys
+
+
 def run():
     """Clean translations."""
-    missing_keys = find()
+    args = get_arguments()
+    if args.target == "frontend":
+        missing_keys = find_frontend()
+        lokalise = get_api(FRONTEND_PROJECT_ID)
+    else:
+        missing_keys = find_core()
+        lokalise = get_api(CORE_PROJECT_ID)
 
     if not missing_keys:
         print("No missing translations!")
         return 0
 
-    lokalise = get_api()
-
     key_data = lokalise.keys_list(
         {"filter_keys": ",".join(missing_keys), "limit": 1000}
     )

script/translations/const.py

@@ -5,3 +5,4 @@ CORE_PROJECT_ID = "130246255a974bd3b5e8a1.51616605"
 FRONTEND_PROJECT_ID = "3420425759f6d6d241f598.13594006"
 DOCKER_IMAGE = "b8329d20280263cad04f65b843e54b9e8e6909a348a678eac959550b5ef5c75f"
 INTEGRATIONS_DIR = pathlib.Path("homeassistant/components")
+FRONTEND_DIR = pathlib.Path("../frontend")

script/translations/util.py

@@ -7,7 +7,7 @@ import subprocess
 from .error import ExitApp
 
 
-def get_base_arg_parser():
+def get_base_arg_parser() -> argparse.ArgumentParser:
     """Get a base argument parser."""
     parser = argparse.ArgumentParser(description="Home Assistant Translations")
     parser.add_argument(