[1812] Adapt Localized String Search

EDCD · Apr 24, 2024 · 4a1a107 · 4a1a107
1 parent 26c8a8b
commit 4a1a107
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/scripts/find_localised_strings.py b/scripts/find_localised_strings.py
@@ -38,8 +38,9 @@ def find_calls_in_stmt(statement: ast.AST) -> list[ast.Call]:
     out = []
     for n in ast.iter_child_nodes(statement):
         out.extend(find_calls_in_stmt(n))
-    if isinstance(statement, ast.Call) and get_func_name(statement.func) == '_':
-        out.append(statement)
+    if isinstance(statement, ast.Call) and get_func_name(statement.func) in ('tr', 'translations'):
+        if ast.unparse(statement).find('.tl') != -1 or ast.unparse(statement).find('translate') != -1:
+            out.append(statement)
     return out