63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
|
"""每日 02:00 数据库备份 job —— 包装 backend/scripts/pg_backup.sh。
|
|||
|
|
|
|||
|
|
独立脚本可手动/CI 直跑；系统调度经此包装，保证与脚本同一份备份逻辑与轮转策略。
|
|||
|
|
Windows 下须显式定位 Git Bash（PATH 里的 bash 可能是 WSL shim，mount/PATH 均不匹配），
|
|||
|
|
脚本路径统一转 MSYS /d/... 格式传给子进程。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import asyncio
|
|||
|
|
import logging
|
|||
|
|
import shutil
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
# backend/app/utils/backup_job.py → parents[2] = backend/
|
|||
|
|
_SCRIPT = Path(__file__).resolve().parents[2] / "scripts" / "pg_backup.sh"
|
|||
|
|
|
|||
|
|
_GIT_BASH_CANDIDATES = (
|
|||
|
|
Path(r"C:\Program Files\Git\bin\bash.exe"),
|
|||
|
|
Path(r"C:\Program Files\Git\usr\bin\bash.exe"),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _find_git_bash() -> str | None:
|
|||
|
|
for cand in _GIT_BASH_CANDIDATES:
|
|||
|
|
if cand.exists():
|
|||
|
|
return str(cand)
|
|||
|
|
return shutil.which("bash")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _msys_path(p: Path) -> str:
|
|||
|
|
"""D:/dpb/... → /d/dpb/...（Git Bash 挂载约定）。"""
|
|||
|
|
s = p.as_posix()
|
|||
|
|
if len(s) >= 2 and s[1] == ":":
|
|||
|
|
return f"/{s[0].lower()}{s[2:]}"
|
|||
|
|
return s
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def run_daily_backup() -> bool:
|
|||
|
|
"""每日备份：调用 pg_backup.sh（独立脚本，pg_dump 自定义格式 + N 天轮转）。"""
|
|||
|
|
script = _SCRIPT
|
|||
|
|
if not script.exists():
|
|||
|
|
alt = Path.cwd() / "scripts" / "pg_backup.sh"
|
|||
|
|
script = alt if alt.exists() else script
|
|||
|
|
if not script.exists():
|
|||
|
|
logger.error(f"备份脚本不存在: {script}")
|
|||
|
|
return False
|
|||
|
|
git_bash = _find_git_bash()
|
|||
|
|
if not git_bash:
|
|||
|
|
logger.error("未找到 Git Bash（C:/Program Files/Git/bin/bash.exe）")
|
|||
|
|
return False
|
|||
|
|
proc = await asyncio.create_subprocess_exec(
|
|||
|
|
git_bash, _msys_path(script),
|
|||
|
|
stdout=asyncio.subprocess.PIPE,
|
|||
|
|
stderr=asyncio.subprocess.PIPE,
|
|||
|
|
)
|
|||
|
|
stdout, stderr = await proc.communicate()
|
|||
|
|
if proc.returncode == 0:
|
|||
|
|
logger.info(f"每日备份完成: {stdout.decode().strip()}")
|
|||
|
|
return True
|
|||
|
|
logger.error(f"每日备份失败(rc={proc.returncode}): {stderr.decode()[:500]}")
|
|||
|
|
return False
|