Update sync_data.sh

sync_data.sh

@@ -1,8 +1,8 @@
-#!/bin/bash
+#!/bin/sh
 
 # 检查环境变量
-if [[ -z "$WEBDAV_URL" ]] || [[ -z "$WEBDAV_USERNAME" ]] || [[ -z "$WEBDAV_PASSWORD" ]]; then
-    echo "Starting without backup functionality - missing WEBDAV_URL, WEBDAV_USERNAME, or WEBDAV_PASSWORD"
+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
 
@@ -13,87 +13,87 @@ if [ -n "$WEBDAV_BACKUP_PATH" ]; then
     FULL_WEBDAV_URL="${WEBDAV_URL}/${WEBDAV_BACKUP_PATH}"
 fi
 
-# 激活虚拟环境
-source $HOME/venv/bin/activate
-
 # 下载最新备份并恢复
 restore_backup() {
-    echo "开始从 WebDAV 下载最新备份..."
+    echo "开始从 网页DAV 下载最新备份..."
     python3 -c "
 import sys
 import os
 import tarfile
 import requests
 import shutil
+import subprocess
 from webdav3.client import Client
+# 设置 WebDAV 客户端
 options = {
-    'webdav_hostname': '$FULL_WEBDAV_URL',
-    'webdav_login': '$WEBDAV_USERNAME',
-    'webdav_password': '$WEBDAV_PASSWORD'
+    '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('alist_backup_')]
+# 列出备份文件
+try:
+    backups = [file for file in client.list() if file.endswith('.tar.gz') and file.startswith('qexo_backup_')]
+except Exception as e:
+    print(f'连接 WebDAV 服务器出错: {e}')
+    sys.exit(1)
 if not backups:
-    print('没有找到备份文件')
-    sys.exit()
+    print('没有找到备份文件，跳过恢复步骤。')
+    sys.exit(0)
+# 获取最新备份文件
 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}'):
-            # 如果目录已存在，先删除它
-            if os.path.exists('$HOME/data'):
-                shutil.rmtree('$HOME/data')
-            os.makedirs('$HOME/data', exist_ok=True)
-            
+# 下载最新备份
+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/db/db.sqlite3']).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('$HOME/data')
-            
+                tar.extractall('/app/')
             print(f'成功从 {latest_backup} 恢复备份')
         else:
-            print('下载的备份文件不存在')
-    else:
-        print(f'下载备份失败：{r.status_code}')
+            print(f'下载备份失败：{r.status_code}')
+except Exception as e:
+    print(f'恢复备份过程中出错: {e}')
 "
 }
 
 # 首次启动时下载最新备份
-echo "Downloading latest backup from WebDAV..."
+echo "Downloading latest backup from 网页DAV..."
 restore_backup
 
-# 等待30秒后启动程序
-sleep 30
-
-# 启动程序
-./app server &
-
 # 同步函数
 sync_data() {
     while true; do
         echo "Starting sync process at $(date)"
 
-        if [ ! -d $HOME/data ]; then
-            mkdir -p $HOME/data
-            echo "Data directory created."
-        fi
-
+        # 直接备份数据库文件
         timestamp=$(date +%Y%m%d_%H%M%S)
-        backup_file="alist_backup_${timestamp}.tar.gz"
+        backup_file="qexo_backup_${timestamp}.tar.gz"
 
-        # 压缩数据目录
-        tar -czf "/tmp/${backup_file}" -C $HOME/data .
+        # 备份数据库和配置文件
+        cd /app
+        tar -czf "/tmp/${backup_file}" db/db.sqlite3 configs.py
 
-        # 上传新备份到WebDAV
+        # 上传新备份到 网页DAV
         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"
+            echo "Successfully uploaded ${backup_file} to 网页DAV"
         else
-            echo "Failed to upload ${backup_file} to WebDAV"
+            echo "Failed to upload ${backup_file} to 网页DAV"
         fi
 
         # 清理旧备份文件
@@ -101,12 +101,12 @@ sync_data() {
 import sys
 from webdav3.client import Client
 options = {
-    'webdav_hostname': '$FULL_WEBDAV_URL',
-    'webdav_login': '$WEBDAV_USERNAME',
-    'webdav_password': '$WEBDAV_PASSWORD'
+    '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('alist_backup_')]
+backups = [file for file in client.list() if file.endswith('.tar.gz') and file.startswith('qexo_backup_')]
 backups.sort()
 if len(backups) > 5:
     to_delete = len(backups) - 5
@@ -118,7 +118,7 @@ else:
 " 2>&1
 
         rm -f "/tmp/${backup_file}"
-        
+
         SYNC_INTERVAL=${SYNC_INTERVAL:-600}
         echo "Next sync in ${SYNC_INTERVAL} seconds..."
         sleep $SYNC_INTERVAL