Initial commit: add AutoDeploy project

deploy_tool/service_console.py

@@ -0,0 +1,70 @@
+from __future__ import annotations
+
+import argparse
+import ctypes
+import locale
+import os
+import subprocess
+import sys
+from pathlib import Path
+
+
+def _set_console_title(title: str) -> None:
+    if os.name == "nt":
+        ctypes.windll.kernel32.SetConsoleTitleW(title)
+
+
+def main() -> int:
+    parser = argparse.ArgumentParser(description="Run a service in a visible console and tee its output to a log file.")
+    parser.add_argument("--title", default="后台服务")
+    parser.add_argument("--log", required=True)
+    parser.add_argument("--pid-file", required=True)
+    parser.add_argument("command", nargs=argparse.REMAINDER)
+    args = parser.parse_args()
+
+    command = list(args.command)
+    if command and command[0] == "--":
+        command.pop(0)
+    if not command:
+        parser.error("missing service command after --")
+
+    _set_console_title(args.title)
+    log_path = Path(args.log)
+    pid_path = Path(args.pid_file)
+    log_path.parent.mkdir(parents=True, exist_ok=True)
+    pid_path.parent.mkdir(parents=True, exist_ok=True)
+
+    encoding = locale.getpreferredencoding(False) or "utf-8"
+    with log_path.open("a", encoding="utf-8") as log_file:
+        process = subprocess.Popen(
+            command,
+            stdin=subprocess.DEVNULL,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.STDOUT,
+            text=True,
+            encoding=encoding,
+            errors="replace",
+            bufsize=1,
+            shell=False,
+        )
+        pid_path.write_text(str(process.pid), encoding="ascii")
+        assert process.stdout is not None
+        try:
+            for line in process.stdout:
+                print(line, end="", flush=True)
+                log_file.write(line)
+                log_file.flush()
+            return process.wait()
+        except KeyboardInterrupt:
+            process.terminate()
+            return process.wait()
+        finally:
+            process.stdout.close()
+            try:
+                pid_path.unlink()
+            except FileNotFoundError:
+                pass
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())