Fix config generation script

BlueGlassBlock · BlueGlassBlock · commit c259ee1c2ba0 · 2023-02-28T14:11:44.000+08:00
diff --git a/.github/scripts/generate_tx_config.py b/.github/scripts/generate_tx_config.py
@@ -1,4 +1,5 @@
 """Please note that this script requires a Transifex API token to run."""
+import glob
 import subprocess
 from functools import partial
 from pathlib import Path
@@ -32,20 +33,33 @@ def add_files(version: str):
 def name_replacer(match: re.Match[str]):
     prefix, resource = match.group("prefix", "resource")
     override_prefix = prefix.replace("file_filter", "trans.zh_CN")
-    override_resource = (
+    pattern = (
         resource.replace("trans/<lang>/", "")
-        .replace("--", "/")
         .replace("glossary_", "glossary")
-        .replace("_", ".")
+        .replace("--", "/")
+        .replace("_", "?")
     )
-    return f"{prefix}{resource}\n{override_prefix}{override_resource}"
-
-
-def patch_config():
+    matches = list(glob.glob(pattern.replace(".po", ".rst")))
+    if not matches:
+        print("missing", pattern)
+        return f"{prefix}{resource}\n{override_prefix}{pattern}"
+    elif len(matches) == 1:
+        filename = matches[0].replace(".rst", ".po").replace("\\", "/")
+    else:
+        raise ValueError("multi match", resource, pattern, matches)
+    return f"{prefix}{resource}\n{override_prefix}{filename}"
+
+
+def patch_config(path: str):
     tx_config_path = Path(".tx", "config")
 
     config_content = tx_config_path.read_text("utf-8")
+
+    cwd = os.getcwd()
+    os.chdir(path)
     config_content = FILTER_PATTERN.sub(name_replacer, config_content)
+    os.chdir(cwd)
+
     tx_config_path.write_text(config_content, "utf-8")
 
 
@@ -55,13 +69,14 @@ def patch_config():
 
     parser = ArgumentParser()
 
-    parser.add_argument("--token")
-    parser.add_argument("--version")
+    parser.add_argument("--token", required=True)
+    parser.add_argument("--version", required=True)
+    parser.add_argument("--doc-path", required=True)
 
     params = parser.parse_args()
 
     os.environ["TX_TOKEN"] = params.token
 
     init_project()
     add_files(params.version)
-    patch_config()
+    patch_config(params.doc_path)
diff --git a/.github/scripts/update.sh b/.github/scripts/update.sh
@@ -2,4 +2,4 @@
 
 tx=$(realpath ./tx)
 cd docs || exit 1
-$tx pull --languages "$LOCALE"
+$tx pull --languages "$LOCALE" -t --use-git-timestamps
