add updater

Author: sshane

selfdrive/ui/layouts/home.py

@@ -9,6 +9,7 @@
 from openpilot.selfdrive.ui.widgets.setup import SetupWidget
 from openpilot.system.ui.lib.text_measure import measure_text_cached
 from openpilot.system.ui.lib.application import gui_app, FontWeight, MousePos
+from openpilot.system.ui.lib.multilang import tr
 from openpilot.system.ui.widgets.label import gui_label
 from openpilot.system.ui.widgets import Widget
 
@@ -151,7 +152,7 @@ def _render_header(self):
       highlight_color = rl.Color(75, 95, 255, 255) if self.current_state == HomeLayoutState.UPDATE else rl.Color(54, 77, 239, 255)
       rl.draw_rectangle_rounded(self.update_notif_rect, 0.3, 10, highlight_color)
 
-      text = "UPDATE"
+      text = tr("UPDATE")
       text_size = measure_text_cached(font, text, HEAD_BUTTON_FONT_SIZE)
       text_x = self.update_notif_rect.x + (self.update_notif_rect.width - text_size.x) // 2
       text_y = self.update_notif_rect.y + (self.update_notif_rect.height - text_size.y) // 2

selfdrive/ui/update_translations_raylib.py

@@ -0,0 +1,47 @@
+#!/usr/bin/env python3
+import argparse
+import json
+import os
+
+from openpilot.common.basedir import BASEDIR
+
+UI_DIR = os.path.join(BASEDIR, "selfdrive", "ui")
+TRANSLATIONS_DIR = os.path.join(UI_DIR, "translations")
+LANGUAGES_FILE = os.path.join(TRANSLATIONS_DIR, "languages.json")
+
+
+def update_translations():
+  files = []
+  for root, _, filenames in os.walk(os.path.join(UI_DIR, "widgets")):
+    for filename in filenames:
+      if filename.endswith(".py"):
+        files.append(os.path.join(root, filename))
+
+  # Create main translation file
+  print(files)
+  cmd = ("xgettext -L Python --keyword=tr --keyword=trn:1,2 --keyword=pgettext:1c,2 --from-code=UTF-8 " +
+         "--flag=tr:1:python-brace-format --flag=trn:1:python-brace-format --flag=trn:2:python-brace-format " +
+         "-o translations/app.pot {}").format(" ".join(files))
+  print(cmd)
+
+  ret = os.system(cmd)
+  assert ret == 0
+
+  # Generate/update translation files for each language
+  with open(LANGUAGES_FILE) as f:
+    translation_files = json.load(f).values()
+
+  for file in translation_files:
+    name = file.replace("main_", "")
+    if os.path.exists(os.path.join(TRANSLATIONS_DIR, f"app_{name}.po")):
+      cmd = "msgmerge --update --backup=none --sort-output translations/app.pot translations/app_{}.po".format(name)
+      ret = os.system(cmd)
+      assert ret == 0
+    else:
+      cmd = "msginit -l es --no-translator --input translations/app.pot --output-file translations/app_{}.po".format(name)
+      ret = os.system(cmd)
+      assert ret == 0
+
+
+if __name__ == "__main__":
+  update_translations()

selfdrive/ui/update_translations_raylib.sh

@@ -0,0 +1,23 @@
+#!/usr/bin/env bash
+
+BASEDIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+cd "$BASEDIR"
+pwd
+
+PY_FILES=$(git ls-files 'layouts/*.py' 'widgets/*.py')
+
+xgettext -L Python \
+  --keyword=tr \
+  --keyword=trn:1,2 \
+  --keyword=pgettext:1c,2 \
+  --from-code=UTF-8 \
+  --flag=tr:1:python-brace-format \
+  --flag=trn:1:python-brace-format --flag=trn:2:python-brace-format \
+  -o translations/app.pot \
+  $PY_FILES
+
+msginit \
+  -l es \
+  --no-translator \
+  --input translations/app.pot \
+  --output-file translations/app.po

selfdrive/ui/widgets/prime.py

@@ -2,6 +2,7 @@
 
 from openpilot.selfdrive.ui.ui_state import ui_state
 from openpilot.system.ui.lib.application import gui_app, FontWeight
+from openpilot.system.ui.lib.multilang import tr
 from openpilot.system.ui.lib.text_measure import measure_text_cached
 from openpilot.system.ui.lib.wrap_text import wrap_text
 from openpilot.system.ui.widgets import Widget
@@ -29,21 +30,21 @@ def _render_for_non_prime_users(self, rect: rl.Rectangle):
     w = rect.width - 160
 
     # Title
-    gui_label(rl.Rectangle(x, y, w, 90), "Upgrade Now", 75, font_weight=FontWeight.BOLD)
+    gui_label(rl.Rectangle(x, y, w, 90), tr("Upgrade Now"), 75, font_weight=FontWeight.BOLD)
 
     # Description with wrapping
     desc_y = y + 140
     font = gui_app.font(FontWeight.LIGHT)
-    wrapped_text = "\n".join(wrap_text(font, "Become a comma prime member at connect.comma.ai", 56, int(w)))
+    wrapped_text = "\n".join(wrap_text(font, tr("Become a comma prime member at connect.comma.ai"), 56, int(w)))
     text_size = measure_text_cached(font, wrapped_text, 56)
     rl.draw_text_ex(font, wrapped_text, rl.Vector2(x, desc_y), 56, 0, rl.WHITE)
 
     # Features section
     features_y = desc_y + text_size.y + 50
-    gui_label(rl.Rectangle(x, features_y, w, 50), "PRIME FEATURES:", 41, font_weight=FontWeight.BOLD)
+    gui_label(rl.Rectangle(x, features_y, w, 50), tr("PRIME FEATURES:"), 41, font_weight=FontWeight.BOLD)
 
     # Feature list
-    features = ["Remote access", "24/7 LTE connectivity", "1 year of drive storage", "Remote snapshots"]
+    features = [tr("Remote access"), tr("24/7 LTE connectivity"), tr("1 year of drive storage"), tr("Remote snapshots")]
     for i, feature in enumerate(features):
       item_y = features_y + 80 + i * 65
       gui_label(rl.Rectangle(x, item_y, 100, 60), "✓", 50, color=rl.Color(70, 91, 234, 255))
@@ -58,5 +59,5 @@ def _render_for_prime_user(self, rect: rl.Rectangle):
     y = rect.y + 40
 
     font = gui_app.font(FontWeight.BOLD)
-    rl.draw_text_ex(font, "✓ SUBSCRIBED", rl.Vector2(x, y), 41, 0, rl.Color(134, 255, 78, 255))
-    rl.draw_text_ex(font, "comma prime", rl.Vector2(x, y + 61), 75, 0, rl.WHITE)
+    rl.draw_text_ex(font, tr("✓ SUBSCRIBED"), rl.Vector2(x, y), 41, 0, rl.Color(134, 255, 78, 255))
+    rl.draw_text_ex(font, tr("comma prime"), rl.Vector2(x, y + 61), 75, 0, rl.WHITE)

system/ui/lib/multilang.py

@@ -13,6 +13,11 @@ def __init__(self):
     self._load_languages()
     self._hook_draw_text()
 
+  def translate(self, text: str) -> str:
+    if self._language not in self._translations:
+      return text
+    return self._translations[self._language].get(text, text)
+
   def _load_languages(self):
     self._language = Params().get("LanguageSetting")
 
@@ -44,3 +49,7 @@ def draw_text_ex_wrapper(font: Union[rl.Font, list, tuple], text: str, position:
 
     rl.draw_text = draw_text_wrapper
     rl.draw_text_ex = draw_text_ex_wrapper
+
+
+multilang = Multilang()
+tr = multilang.translate