frea: admin初始化修改

2026-06-14 20:41:21 +08:00 · 2026-03-30 16:24:02 +08:00
parent 8341905b21
commit 1cc7f4bc7b
5 changed files with 130 additions and 10 deletions
@@ -0,0 +1,105 @@
+"""
+Trigger gin-vue-admin InitDB after Dify console setup.
+
+Uses the same DB_* settings as the API container and the admin password from the setup form.
+Does not read or write local config files; outbound call only if ADMIN_INITDB_ENABLED is true.
+"""
+
+from __future__ import annotations
+
+import logging
+from typing import Any, Literal
+
+import httpx
+
+from configs import dify_config
+
+logger = logging.getLogger(__name__)
+
+_ADMIN_ALREADY_INIT_MSG = "已存在数据库配置"
+_DEFAULT_TIMEOUT = httpx.Timeout(120.0, connect=10.0)
+
+
+def _admin_db_type(db_type: str) -> Literal["pgsql", "mysql"]:
+    if db_type == "postgresql":
+        return "pgsql"
+    return "mysql"
+
+
+def _build_payload(admin_password: str) -> dict[str, Any]:
+    db_type = _admin_db_type(dify_config.DB_TYPE)
+    return {
+        "adminPassword": admin_password,
+        "dbType": db_type,
+        "host": dify_config.DB_HOST,
+        "port": str(dify_config.DB_PORT),
+        "userName": dify_config.DB_USERNAME,
+        "password": dify_config.DB_PASSWORD,
+        "dbName": dify_config.DB_DATABASE,
+        "dbPath": "",
+    }
+
+
+def _is_acceptable_admin_response(body: dict[str, Any]) -> bool:
+    code = body.get("code")
+    msg = body.get("msg") or ""
+    if code == 0:
+        return True
+    if msg == _ADMIN_ALREADY_INIT_MSG:
+        return True
+    return False
+
+
+def trigger_admin_initdb_if_configured(*, admin_password: str) -> None:
+    """
+    Best-effort POST to Admin InitDB. Logs warnings on failure; never raises.
+    """
+    if not dify_config.ADMIN_INITDB_ENABLED:
+        return
+    if dify_config.DB_TYPE not in ("postgresql", "mysql", "oceanbase", "seekdb"):
+        logger.warning(
+            "skip admin initdb: unsupported DB_TYPE for admin bridge: %s",
+            dify_config.DB_TYPE,
+        )
+        return
+
+    url = (dify_config.ADMIN_INITDB_URL or "").strip()
+    if not url:
+        logger.warning("ADMIN_INITDB_ENABLED is true but ADMIN_INITDB_URL is empty, skip admin initdb")
+        return
+
+    payload = _build_payload(admin_password)
+    try:
+        with httpx.Client(timeout=_DEFAULT_TIMEOUT, follow_redirects=True) as client:
+            response = client.post(
+                url,
+                json=payload,
+                headers={"Content-Type": "application/json", "Accept": "application/json"},
+            )
+    except httpx.RequestError as exc:
+        logger.warning("admin initdb request failed: %s", exc)
+        return
+
+    if response.status_code != 200:
+        logger.warning(
+            "admin initdb HTTP %s: %s",
+            response.status_code,
+            (response.text or "")[:500],
+        )
+        return
+
+    try:
+        body = response.json()
+    except ValueError:
+        logger.warning("admin initdb returned non-JSON body: %s", (response.text or "")[:500])
+        return
+
+    if not isinstance(body, dict):
+        logger.warning("admin initdb returned unexpected JSON: %s", body)
+        return
+
+    if _is_acceptable_admin_response(body):
+        logger.info("admin initdb finished: %s", body.get("msg"))
+        return
+
+    logger.warning("admin initdb rejected: %s", body)