Update scripts/restore_from_dataset.py

scripts/restore_from_dataset.py

@@ -4,10 +4,8 @@ import sys
 import shutil
 from huggingface_hub import HfApi, login, hf_hub_download
 
-# ===== 配置 =====
-DB_TARGET = "/data/freeapi.db"                     # 数据库最终存放位置
-DATASET_REPO = "你的用户名/freellm-backup"          # 替换为你的数据集名称
-# ================
+DB_TARGET = "/data/freeapi.db"
+DATASET_REPO = "lydgs/freellm-backup"   # 已改为你的用户名
 
 def restore_latest_backup():
     token = os.getenv("HF_TOKEN")
@@ -19,19 +17,16 @@ def restore_latest_backup():
     api = HfApi()
 
     try:
-        # 列出数据集中的所有文件
         files = api.list_repo_files(repo_id=DATASET_REPO, repo_type="dataset")
         db_files = [f for f in files if f.endswith(".db") and f.startswith("freeapi_backup_")]
         if not db_files:
             print("ℹ️ 数据集中没有找到备份文件，跳过恢复")
             return
 
-        # 按文件名排序（时间戳升序），取最新的
         db_files.sort()
         latest = db_files[-1]
         print(f"🔄 发现最新备份: {latest}，开始下载...")
 
-        # 下载到临时目录
         downloaded = hf_hub_download(
             repo_id=DATASET_REPO,
             filename=latest,
@@ -39,14 +34,9 @@ def restore_latest_backup():
             local_dir="/tmp"
         )
 
-        # 确保目标目录存在
         os.makedirs(os.path.dirname(DB_TARGET), exist_ok=True)
-
-        # 复制到目标位置
         shutil.copy(downloaded, DB_TARGET)
         print(f"✅ 数据库恢复成功: {DB_TARGET} (来自备份 {latest})")
-
-        # 可选：删除临时下载文件
         os.remove(downloaded)
     except Exception as e:
         print(f"❌ 数据库恢复失败: {e}")