Change [scripts]: The logic of the scripts is completely separated from the main code

scripts/pybabel.py

@@ -0,0 +1,128 @@
+# /// script
+# requires-python = ">=3.13"
+# dependencies = [
+#     "argparse",
+#     "pathlib",
+# ]
+# ///
+
+import argparse
+import subprocess
+from pathlib import Path
+
+# Constant paths for project.
+LOCALES_DIR = Path("src/locales")
+LOCALES_DOMAIN = "messages"
+LOCALES_POT = LOCALES_DIR / f"{LOCALES_DOMAIN}.pot"
+LOCALES_WIDTH = 80
+
+
+def run_commands(cmds: list[list[str]]) -> None:
+    """
+    Run terminal commands from Python.
+    Used for pybabel commands.
+
+    Args:
+        cmds (list[list[str]]): list of commands split by space.
+        Example: [["echo", "Hello World"], ["uv", "run", "main.py"]]
+    """
+    try:
+        # Try to run command and print output.
+        # Print only stderr output (pybabel use stderr).
+        for cmd in cmds:
+            # DEBUG: User input is completely safe,
+            # as it only accepts specific values.
+            result = subprocess.run(  # noqa: S603
+                cmd,
+                capture_output=True,
+                text=True,
+                check=True,
+                shell=False,
+            )
+            print(result.stderr)
+    except Exception as error:
+        print(f"Error: {error}")
+
+
+def main() -> None:
+    """
+    Main logic of script.
+    Parse args, try to run commands.
+    """
+    parser = argparse.ArgumentParser(
+        description="Wrapper for pybabel operations."
+    )
+
+    parser.add_argument(
+        "operation",
+        choices=["extract", "update", "compile", "init"],
+        help="Type of pybabel operation.",
+    )
+
+    parser.add_argument(
+        "language",
+        nargs="?",
+        default="all",
+        choices=["ru", "en", "zh", "es", "ar", "fr", "de", "all"],
+        help=""
+        + "Language code. "
+        + "If omitted, all locales are processed. "
+        + "Only for update & compile operations.",
+    )
+
+    args = parser.parse_args()
+
+    if args.operation == "init" and args.language == "all":
+        print("Error: You need to specify new language.")
+        return
+
+    cmd = list()
+    cmd.append("pybabel")
+    cmd.append(args.operation)
+
+    # Extract operation require only extract command.
+    if args.operation == "extract":
+        cmd.append(".")
+        cmd.append("-o")
+        cmd.append(str(LOCALES_POT))
+
+        run_commands([cmd])
+        return
+
+    # Other operations (init, update, compile)
+    # may require multiple commands.
+
+    # Set base flags for these operations.
+    cmd.append("-D")
+    cmd.append(LOCALES_DOMAIN)
+
+    cmd.append("-d")
+    cmd.append(str(LOCALES_DIR))
+
+    # Specific flags
+    if args.operation in ("update", "init"):
+        cmd.append("-i")
+        cmd.append(str(LOCALES_POT))
+
+        if args.operation == "update":
+            cmd.append("-w")
+            cmd.append(str(LOCALES_WIDTH))
+
+    # Add all langs or specific lang.
+    if args.language == "all":
+        langs = [
+            str(path_name.name)
+            for path_name in LOCALES_DIR.iterdir()
+            if path_name.is_dir()
+        ]
+    else:
+        langs = [args.language]
+
+    cmds = [cmd + ["-l"] + [lang] for lang in langs]
+
+    # Run commands
+    run_commands(cmds)
+
+
+if __name__ == "__main__":
+    main()