if [ -z "$WEBDAV_URL" ] || [ -z "$WEBDAV_USERNAME" ] || [ -z "$WEBDAV_PASSWORD" ]; then echo "Starting without sync functionality - missing WEBDAV_URL, WEBDAV_USERNAME, or WEBDAV_PASSWORD" exit 0 fi
WEBDAV_BACKUP_PATH=${WEBDAV_BACKUP_PATH:-""} FULL_WEBDAV_URL="${WEBDAV_URL}" if [ -n "$WEBDAV_BACKUP_PATH" ]; then FULL_WEBDAV_URL="${WEBDAV_URL}/${WEBDAV_BACKUP_PATH}" fi
restore_backup() { echo "开始从 WebDAV 下载最新备份..." $VENV_PYTHON -c " import sys import os import tarfile import requests import shutil import subprocess from webdav3.client import Client options = { 'webdav_hostname': '${FULL_WEBDAV_URL}', 'webdav_login': '${WEBDAV_USERNAME}', 'webdav_password': '${WEBDAV_PASSWORD}' } client = Client(options) try: backups = [file for file in client.list() if file.endswith('.tar.gz') and file.startswith('komari_backup_')] except Exception as e: print(f'连接 WebDAV 服务器出错: {e}') sys.exit(1) if not backups: print('没有找到备份文件,跳过恢复步骤。') sys.exit(0) latest_backup = sorted(backups)[-1] print(f'最新备份文件:{latest_backup}') try: with requests.get(f'${FULL_WEBDAV_URL}/{latest_backup}', auth=('${WEBDAV_USERNAME}', '${WEBDAV_PASSWORD}'), stream=True) as r: if r.status_code == 200: with open(f'/tmp/{latest_backup}', 'wb') as f: for chunk in r.iter_content(chunk_size=8192): f.write(chunk) print(f'成功下载备份文件到 /tmp/{latest_backup}') # 检查数据库文件是否被占用 try: # 检查整个目录下的文件是否被占用(仅检测主要数据库文件,有需要可扩展) pids = subprocess.check_output(['lsof', '-t', '/app/data/komari.db']).decode().strip() if pids: print('数据库文件被占用,无法恢复备份。') sys.exit(1) except subprocess.CalledProcessError: pass # 解压整个目录 with tarfile.open(f'/tmp/{latest_backup}', 'r:gz') as tar: tar.extractall('/app/') print(f'成功从 {latest_backup} 恢复备份到 /app/data') else: print(f'下载备份失败:{r.status_code}') except Exception as e: print(f'恢复备份过程中出错: {e}') " }
echo "Downloading latest backup from WebDAV..." restore_backup
sync_data() { while true; do echo "Starting sync process at $(date)" timestamp=$(date +%Y%m%d_%H%M%S) backup_file="komari_backup_${timestamp}.tar.gz"
# 备份整个/app/data目录 cd /app tar -czf "/tmp/${backup_file}" data
curl -u "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" -T "/tmp/${backup_file}" "$FULL_WEBDAV_URL/${backup_file}" if [ $? -eq 0 ]; then echo "Successfully uploaded ${backup_file} to WebDAV" else echo "Failed to upload ${backup_file} to WebDAV" fi
$VENV_PYTHON -c " from webdav3.client import Client options = { 'webdav_hostname': '${FULL_WEBDAV_URL}', 'webdav_login': '${WEBDAV_USERNAME}', 'webdav_password': '${WEBDAV_PASSWORD}' } client = Client(options) backups = [file for file in client.list() if file.endswith('.tar.gz') and file.startswith('komari_backup_')] backups.sort() if len(backups) > 5: to_delete = len(backups) - 5 for file in backups[:to_delete]: client.clean(file) print(f'Successfully deleted {file}.') else: print('Only {} backups found, no need to clean.'.format(len(backups))) " 2>&1