Update sync_data.sh

sync_data.sh

@@ -1,7 +1,13 @@
 #!/bin/sh
 
+# 设置默认的同步间隔为 600 秒 (10 分钟)，可以作为环境变量传入
+SYNC_INTERVAL=${SYNC_INTERVAL:-600}
+
+# 设置备份保留数量，默认为 5，可以作为环境变量传入
+BACKUP_RETENTION=${BACKUP_RETENTION:-5}
+
 # 检查环境变量
-if [ -z "$HF_TOKEN" ] || [ -z "$DATASET_ID" ]; then
+if [[ -z "$HF_TOKEN" ]] || [[ -z "$DATASET_ID" ]]; then
     echo "Starting without backup functionality - missing HF_TOKEN or DATASET_ID"
     exec ./cloudreve
     exit 0
@@ -62,7 +68,7 @@ try:
     backup_files = [f for f in files if f.startswith('cloudreve_backup_') and f.endswith('.tar.gz')]
 
     if not backup_files:
-        print('No backup files found on HuggingFace Dataset')
+        print('在 HuggingFace Dataset 上没有找到备份文件')
         return
 
     latest_backup = sorted(backup_files)[-1]
@@ -76,59 +82,58 @@ try:
         )
 
         if filepath and os.path.exists(filepath):
-            print(f'Found latest backup: {latest_backup}, attempting to restore...')
+            print(f'找到最新的备份：{latest_backup}，尝试恢复...')
             # 删除现有的数据目录
             if os.path.isdir('/opt/cloudreve/data'):
-                echo "Deleting existing data directory: /opt/cloudreve/data"
-                rm -rf /opt/cloudreve/data
+                echo "删除现有的数据目录：/opt/cloudreve/data"
+                shutil.rmtree('/opt/cloudreve/data')
             # 创建数据目录（如果被删除）
             os.makedirs('/opt/cloudreve/data', exist_ok=True)
 
             with tarfile.open(filepath, 'r:gz') as tar:
                 tar.extractall('/opt/cloudreve/data')
-            echo f'Successfully restored backup from {latest_backup}'
+            echo f'成功从 {latest_backup} 恢复备份'
 
 except Exception as e:
-    echo f'Error downloading backup from HuggingFace Dataset: {e}'
+    echo f'从 HuggingFace Dataset 下载备份时出错：{e}'
 "
 }
 
 # 首次启动时下载最新备份和后续每次启动时都尝试恢复
-echo "Attempting to download and restore latest backup from HuggingFace Dataset..."
+echo "尝试从 HuggingFace Dataset 下载并恢复最新备份..."
 download_latest_backup
 
 # 同步函数
 sync_data() {
     while true; do
-        echo "Starting sync process at $(date)"
+        echo "开始同步进程于 $(date)"
 
         if [ -d /opt/cloudreve/data ]; then
             timestamp=$(date +%Y%m%d_%H%M%S)
             backup_file="cloudreve_backup_${timestamp}.tar.gz"
             backup_file_path="/tmp/${backup_file}"
 
-            echo "Creating backup archive: ${backup_file_path}"
-            echo "Listing content of /opt/cloudreve/data before backup:"
+            echo "创建备份归档：${backup_file_path}"
+            echo "在备份之前列出 /opt/cloudreve/data 的内容："
             ls -al /opt/cloudreve/data
 
             # 尝试直接打包 /opt/cloudreve/data 目录，并显示详细信息
             tar -czvf "${backup_file_path}" /opt/cloudreve/data
 
             if [ -f "${backup_file_path}" ]; then
-                echo "Uploading backup to HuggingFace Dataset..."
+                echo "正在上传备份到 HuggingFace Dataset..."
                 upload_backup "${backup_file_path}" "${backup_file}"
                 rm -f "${backup_file_path}"
             else:
-                echo "Failed to create backup archive."
+                echo "创建备份归档失败。"
             fi
-
         else:
-            echo "Data directory does not exist yet, waiting for next sync..."
+            echo "数据目录尚不存在，等待下一次同步..."
         fi
 
         SYNC_INTERVAL=${SYNC_INTERVAL:-7200}
-        echo "Next sync in ${SYNC_INTERVAL} seconds..."
-        sleep $SYNC_INTERVAL
+        echo "下一次同步将在 ${SYNC_INTERVAL} 秒后进行..."
+        sleep "$SYNC_INTERVAL"
     done
 }