Update sync_data.sh

sync_data.sh

@@ -95,22 +95,50 @@ EOL
 # 首次启动时的数据恢复策略
 echo "开始初始化数据恢复..."
 
-# 首先尝试从 WebDAV 恢复指定文件
-echo "正在尝试从 WebDAV 下载指定数据库文件..."
-curl -L --fail --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/openwebui/webui_03-05.db" -o "./data/webui.db" && {
-    echo "成功从 WebDAV 下载指定数据库文件"
-    # 上传到 HuggingFace
-    timestamp=$(date +%Y%m%d_%H%M%S)
-    backup_file="webui_backup_${timestamp}.db"
-    cp ./data/webui.db "/tmp/${backup_file}"
-    echo "正在上传初始备份到 HuggingFace..."
-    python3 /tmp/hf_sync.py upload "${HF_TOKEN}" "${DATASET_ID}" "/tmp/${backup_file}" "${backup_file}"
-    rm -f "/tmp/${backup_file}"
-} || {
-    echo "从 WebDAV 下载指定文件失败，尝试下载 WebDAV 的 webui.db..."
+# 首先尝试从 WebDAV 恢复最新文件
+echo "正在尝试从 WebDAV 获取文件列表..."
+
+# 获取 WebDAV 目录中的文件列表并找出最新的备份文件
+webdav_files=$(curl -s -X PROPFIND --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" -H "Depth: 1" "$WEBDAV_URL/openwebui/" | grep -o '<d:href>[^<]*webui_[0-9][0-9]-[0-9][0-9]\.db</d:href>' | sed 's/<d:href>//g' | sed 's/<\/d:href>//g')
+
+if [ -n "$webdav_files" ]; then
+    # 找出最新的文件 (按文件名排序，取最后一个)
+    latest_file=$(echo "$webdav_files" | sort | tail -n 1)
+    echo "找到最新的 WebDAV 备份文件: $latest_file"
     
-    curl -L --fail --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/webui.db" -o "./data/webui.db" && {
-        echo "成功从 WebDAV 下载最新数据库文件"
+    # 下载最新文件
+    curl -L --fail --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL$latest_file" -o "./data/webui.db" && {
+        echo "成功从 WebDAV 下载最新数据库文件: $latest_file"
+        # 上传到 HuggingFace
+        timestamp=$(date +%Y%m%d_%H%M%S)
+        backup_file="webui_backup_${timestamp}.db"
+        cp ./data/webui.db "/tmp/${backup_file}"
+        echo "正在上传初始备份到 HuggingFace..."
+        python3 /tmp/hf_sync.py upload "${HF_TOKEN}" "${DATASET_ID}" "/tmp/${backup_file}" "${backup_file}"
+        rm -f "/tmp/${backup_file}"
+    } || {
+        echo "从 WebDAV 下载最新文件失败，尝试下载 WebDAV 的 webui.db..."
+        
+        curl -L --fail --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/openwebui/webui.db" -o "./data/webui.db" && {
+            echo "成功从 WebDAV 下载主数据库文件"
+            # 上传到 HuggingFace
+            timestamp=$(date +%Y%m%d_%H%M%S)
+            backup_file="webui_backup_${timestamp}.db"
+            cp ./data/webui.db "/tmp/${backup_file}"
+            echo "正在上传初始备份到 HuggingFace..."
+            python3 /tmp/hf_sync.py upload "${HF_TOKEN}" "${DATASET_ID}" "/tmp/${backup_file}" "${backup_file}"
+            rm -f "/tmp/${backup_file}"
+        } || {
+            echo "从 WebDAV 下载失败，尝试从 HuggingFace 恢复..."
+            python3 /tmp/hf_sync.py download "${HF_TOKEN}" "${DATASET_ID}" || {
+                echo "所有恢复方式均失败，将使用空数据库开始"
+            }
+        }
+    }
+else
+    echo "WebDAV 中没有找到符合格式的备份文件，尝试下载 webui.db..."
+    curl -L --fail --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/openwebui/webui.db" -o "./data/webui.db" && {
+        echo "成功从 WebDAV 下载主数据库文件"
         # 上传到 HuggingFace
         timestamp=$(date +%Y%m%d_%H%M%S)
         backup_file="webui_backup_${timestamp}.db"
@@ -124,7 +152,7 @@ curl -L --fail --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/openwebui
             echo "所有恢复方式均失败，将使用空数据库开始"
         }
     }
-}
+fi
 
 # 定义 WebDAV 同步函数
 webdav_sync() {
@@ -143,11 +171,11 @@ webdav_sync() {
             echo "同步到 WebDAV..."
 
             # 上传以日期命名的数据库文件
-            curl -L -T "./data/webui.db" --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/$FILENAME" && {
+            curl -L -T "./data/webui.db" --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/openwebui/$FILENAME" && {
                 echo "WebDAV 上传成功: $FILENAME"
                 
                 # 覆盖Webdav目录下默认的webui.db文件
-                curl -L -T "./data/webui.db" --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/webui.db" && {
+                curl -L -T "./data/webui.db" --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/openwebui/webui.db" && {
                     echo "WebDAV 更新主文件成功"
                 } || {
                     echo "WebDAV 更新主文件失败"
@@ -155,7 +183,7 @@ webdav_sync() {
             } || {
                 echo "WebDAV 上传失败，等待重试..."
                 sleep 10
-                curl -L -T "./data/webui.db" --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/$FILENAME" || {
+                curl -L -T "./data/webui.db" --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/openwebui/$FILENAME" || {
                     echo "重试失败，放弃上传。"
                 }
             }
@@ -201,3 +229,4 @@ hf_sync() {
 # 后台启动同步进程
 webdav_sync &
 hf_sync &
+