mirror of
https://github.com/YFGaia/dify-plus.git
synced 2026-06-04 10:14:00 +08:00
npc0-hue
106 lines
3.1 KiB
Python
106 lines
3.1 KiB
Python
"""
|
|
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)
|