openwebuidb / sync_data.sh
damxin's picture
sss
6a2dd29 verified
Raw History Blame Contribute Delete
4.82 kB
#!/bin/bash
source ~/.env
# 检查环境变量
# if [[ -z "$DATABASE_URL" ]]; then
# echo "未配置数据库(DATABASE_URL为空),退出流程。"
# exit 0
# fi
echo "当前 DATABASE_URL: $DATABASE_URL"
echo "已配置数据库,开始执行后续操作..."
# if [[ -z "$WEBDAV_URL" ]] || [[ -z "$WEBDAV_USERNAME" ]] || [[ -z "$WEBDAV_PASSWORD" ]]; then
# echo "缺少 WEBDAV_URL、WEBDAV_USERNAME 或 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 下载最新备份..."
# python3 -c "
# import sys
# import os
# import tarfile
# import requests
# from webdav3.client import Client
# import shutil
# 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('webui_backup_')]
# if not backups:
# print('没有找到备份文件')
# sys.exit()
# latest_backup = sorted(backups)[-1]
# print(f'最新备份文件:{latest_backup}')
# 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}')
# if os.path.exists(f'/tmp/{latest_backup}'):
# # 解压备份文件到临时目录
# temp_dir = '/tmp/restore'
# os.makedirs(temp_dir, exist_ok=True)
# tar = tarfile.open(f'/tmp/{latest_backup}', 'r:gz')
# tar.extractall(temp_dir)
# tar.close()
# # 查找并移动 webui.db 文件
# for root, dirs, files in os.walk(temp_dir):
# if 'webui.db' in files:
# db_path = os.path.join(root, 'webui.db')
# os.makedirs('./data', exist_ok=True)
# os.replace(db_path, './data/webui.db')
# print(f'成功从 {latest_backup} 恢复备份')
# break
# else:
# print('备份文件中未找到 webui.db')
# # 删除临时目录
# try:
# shutil.rmtree(temp_dir)
# except Exception as e:
# print(f'删除临时目录时出错:{e}')
# os.remove(f'/tmp/{latest_backup}')
# else:
# print('下载的备份文件不存在')
# else:
# print(f'下载备份失败:{r.status_code}')
# "
# }
# 首次启动时下载最新备份
# echo "正在从 WebDAV 下载最新备份..."
# restore_backup
# 同步函数
# sync_data() {
# while true; do
# echo "在 $(date) 开始同步进程"
# if [ -f "./data/webui.db" ]; then
# timestamp=$(date +%Y%m%d_%H%M%S)
# backup_file="webui_backup_${timestamp}.tar.gz"
# 打包数据库文件
# tar -czf "/tmp/${backup_file}" ./data/webui.db
# 上传新备份到WebDAV
# curl -u "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" -T "/tmp/${backup_file}" "$FULL_WEBDAV_URL/${backup_file}"
# if [ $? -eq 0 ]; then
# echo "成功将 ${backup_file} 上传至 WebDAV"
# else
# echo "上传 ${backup_file} 至 WebDAV 失败"
# fi
#
# 清理旧备份文件
# python3 -c "
# import sys
# 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('webui_backup_')]
# backups.sort()
# if len(backups) > 5:
# to_delete = len(backups) - 5
# for file in backups[:to_delete]:
# client.clean(file)
# print(f'成功删除 {file}。')
# else:
# print('仅找到 {} 个备份,无需清理。'.format(len(backups)))
# " 2>&1
# rm -f "/tmp/${backup_file}"
# else
# echo "数据库文件尚不存在,等待下次同步..."
# fi
# SYNC_INTERVAL=${SYNC_INTERVAL:-600}
# echo "下次同步将在 ${SYNC_INTERVAL} 秒后进行..."
# sleep $SYNC_INTERVAL
# done
# }
# 后台启动同步进程
# sync_data &