first commit

inventory_sync/postgres.py

@@ -0,0 +1,51 @@
+from __future__ import annotations
+
+import csv
+import io
+import os
+import subprocess
+from typing import Any
+
+from .settings import AppConfig
+from .sql import POSTGRES_QUERY
+
+
+def query_postgres(config: AppConfig) -> dict[str, Any]:
+    env = os.environ.copy()
+    env["PGPASSWORD"] = config.analytics_db_password
+    command = [
+        "/usr/local/bin/psql",
+        "-h",
+        config.analytics_db_host,
+        "-p",
+        config.analytics_db_port,
+        "-U",
+        config.analytics_db_user,
+        "-d",
+        config.analytics_db_name,
+        "-v",
+        "ON_ERROR_STOP=1",
+        "-c",
+        "SET TIME ZONE 'Asia/Shanghai';",
+        "-c",
+        f"COPY ({POSTGRES_QUERY}) TO STDOUT WITH CSV HEADER",
+    ]
+
+    try:
+        completed = subprocess.run(
+            command,
+            check=True,
+            capture_output=True,
+            text=True,
+            env=env,
+        )
+    except subprocess.CalledProcessError as exc:
+        detail = exc.stderr.strip() or exc.stdout.strip()
+        raise RuntimeError(f"PostgreSQL 查询失败: {detail}") from exc
+
+    cleaned_output = "\n".join(
+        line for line in completed.stdout.splitlines() if line.strip() and line.strip() != "SET"
+    )
+    rows = list(csv.DictReader(io.StringIO(cleaned_output)))
+    return {"row_count": len(rows), "rows": rows}
+