yuqei
47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
from pathlib import Path
|
|
import json
|
|
import os
|
|
import sys
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
if str(ROOT) not in sys.path:
|
|
sys.path.insert(0, str(ROOT))
|
|
|
|
from sqlalchemy import text
|
|
from sqlalchemy.exc import SQLAlchemyError
|
|
|
|
from src.db.engine import engine
|
|
|
|
|
|
def main(env=None) -> int:
|
|
env = os.environ if env is None else env
|
|
output_json = _get_bool(env, "CHECK_DB_OUTPUT_JSON", False)
|
|
try:
|
|
with engine.connect() as connection:
|
|
result = connection.execute(text("select 1"))
|
|
value = result.scalar_one()
|
|
except SQLAlchemyError as exc:
|
|
payload = {"passed": False, "available": False, "error_type": exc.__class__.__name__, "error": str(exc)}
|
|
if output_json:
|
|
print(json.dumps(payload, default=str, ensure_ascii=False, sort_keys=True))
|
|
else:
|
|
print(f"database connection failed: {exc.__class__.__name__}")
|
|
return 2
|
|
payload = {"passed": True, "available": True, "value": value}
|
|
if output_json:
|
|
print(json.dumps(payload, default=str, ensure_ascii=False, sort_keys=True))
|
|
else:
|
|
print(f"database connection ok: {value}")
|
|
return 0
|
|
|
|
|
|
def _get_bool(env, key: str, default: bool) -> bool:
|
|
value = env.get(key)
|
|
if value is None or value == "":
|
|
return default
|
|
return str(value).strip().lower() in {"1", "true", "yes", "on"}
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|