Move TEXT_TO_REPLACE outside function, move parse_args first

Author: rffontenelle

scripts/generate_txconfig.py

@@ -11,6 +11,51 @@
 import sys
 from pathlib import Path
 
+# Replaces required to fix the default values set by 'tx add remote' command.
+# Add or remove 
+TEXT_TO_REPLACE = {
+    "2_": "2.",
+    "3_": "3.",
+    "glossary_": "glossary",
+    "collections_": "collections.",
+    "compression_": "compression.",
+    "concurrent_": "concurrent.",
+    "curses_": "curses.",
+    "email_": "email.",
+    "html_": "html.",
+    "http_": "http.",
+    "importlib_resources_": "importlib.resources.",
+    "importlib_": "importlib.",
+    "logging_": "logging.",
+    "multiprocessing_": "multiprocessing.",
+    "os_": "os.",
+    "string_": "string.",
+    "sys_monitoring": "sys.monitoring",
+    "tkinter_": "tkinter.",
+    "unittest_": "unittest.",
+    "urllib_": "urllib.",
+    "xml_dom_": "xml.dom.",
+    "xml_etree_": "xml.etree.",
+    "xmlrpc_": "xmlrpc.",
+    "xml_sax_": "xml.sax.",
+    "xml_": "xml."
+}
+
+def parse_args():
+    parser = argparse.ArgumentParser(description=__doc__)
+    parser.add_argument(
+        "--root-path",
+        "-p",
+        default=Path("."),
+        help="Path to the translation files, and also the .tx/config (defaults to current directory)"
+    )
+    parser.add_argument(
+        "tx_project",
+        help="Slug of the Transifex project to query resources from"
+    )
+    return parser.parse_args()
+
+
 def reset_tx_config(txconfig: Path):
     """Create or reset the .tx/config file with basic header."""
     txconfig.parent.mkdir(exist_ok=True)
@@ -37,34 +82,6 @@ def patch_config(txconfig: Path):
     content = txconfig.read_text(encoding="utf-8").splitlines()
     new_lines = []
 
-    text_to_replace = {
-        "2_": "2.",
-        "3_": "3.",
-        "glossary_": "glossary",
-        "collections_": "collections.",
-        "compression_": "compression.",
-        "concurrent_": "concurrent.",
-        "curses_": "curses.",
-        "email_": "email.",
-        "html_": "html.",
-        "http_": "http.",
-        "importlib_resources_": "importlib.resources.",
-        "importlib_": "importlib.",
-        "logging_": "logging.",
-        "multiprocessing_": "multiprocessing.",
-        "os_": "os.",
-        "string_": "string.",
-        "sys_monitoring": "sys.monitoring",
-        "tkinter_": "tkinter.",
-        "unittest_": "unittest.",
-        "urllib_": "urllib.",
-        "xml_dom_": "xml.dom.",
-        "xml_etree_": "xml.etree.",
-        "xmlrpc_": "xmlrpc.",
-        "xml_sax_": "xml.sax.",
-        "xml_": "xml."
-    }
-
     for line in content:
         if line.startswith(("source_file", "source_lang")):
             continue
@@ -73,7 +90,7 @@ def patch_config(txconfig: Path):
             line = line.replace("<lang>/", "")
             line = line.replace("--", "/")
 
-            for pattern, replacement in text_to_replace.items():
+            for pattern, replacement in TEXT_TO_REPLACE.items():
                if pattern in line:
                   line = line.replace(pattern, replacement)
                   break
@@ -85,30 +102,14 @@ def patch_config(txconfig: Path):
     txconfig.write_text(text + "\n", encoding="utf-8")
     print("Updated .tx/config with character substitutions")
 
-
-def parse_args():
-    parser = argparse.ArgumentParser(description=__doc__)
-    parser.add_argument(
-        "--root-path",
-        "-p",
-        default=Path("."),
-        help="Path to the translation files, and also the .tx/config"
-    )
-    parser.add_argument(
-        "tx_project",
-        help="Name of the Transifex project to query resources from (e.g. python-newest or python-313)"
-    )
-    return parser.parse_args()
-
-
 def main():
     args = parse_args()
-    TX_CONFIG = Path(".tx/config")
+    config_path = Path(".tx/config")
     if args.root_path:
-        TX_CONFIG = args.root_path / TX_CONFIG
-    reset_tx_config(TX_CONFIG)
-    populate_resources_from_remote(TX_CONFIG, args.tx_project)
-    patch_config(TX_CONFIG)
+        config_path = args.root_path / config_path
+    reset_tx_config(config_path)
+    populate_resources_from_remote(config_path, args.tx_project)
+    patch_config(config_path)
 
 
 if __name__ == "__main__":