Create scripts/restore_from_dataset.py

scripts/restore_from_dataset.py

@@ -0,0 +1,55 @@
+#!/usr/bin/env python3
+import os
+import sys
+import shutil
+from huggingface_hub import HfApi, login, hf_hub_download
+
+# ===== 配置 =====
+DB_TARGET = "/data/freeapi.db"                     # 数据库最终存放位置
+DATASET_REPO = "你的用户名/freellm-backup"          # 替换为你的数据集名称
+# ================
+
+def restore_latest_backup():
+    token = os.getenv("HF_TOKEN")
+    if not token:
+        print("⚠️ 未设置 HF_TOKEN 环境变量，跳过数据库恢复")
+        return
+
+    login(token=token)
+    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,
+            repo_type="dataset",
+            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}")
+
+if __name__ == "__main__":
+    restore_latest_backup()