Update scripts/backup_to_dataset.py

scripts/backup_to_dataset.py

@@ -7,6 +7,10 @@ import sqlite3
 import tarfile
 from huggingface_hub import HfApi, login
 
+# 强制 stdout/stderr 实时输出（对 cron 日志很重要）
+sys.stdout.reconfigure(line_buffering=True)
+sys.stderr.reconfigure(line_buffering=True)
+
 # ===== 配置 =====
 DB_PATH = "/app/server/data/freeapi.db"          # 数据库路径
 DATASET_REPO = "lydgs/freellm-backup"           # 数据集名称
@@ -46,6 +50,12 @@ def export_db_to_json(db_path, temp_dir):
     return exported_files
 
 def backup_database():
+    # 打印调试信息：检查环境变量
+    print(f"[DEBUG] HF_TOKEN present: {bool(os.getenv('HF_TOKEN'))}")
+    print(f"[DEBUG] BAILIAN_API_KEY present: {bool(os.getenv('BAILIAN_API_KEY'))}")
+    print(f"[DEBUG] Database file exists: {os.path.exists(DB_PATH)}")
+    print(f"[DEBUG] Database file size: {os.path.getsize(DB_PATH) if os.path.exists(DB_PATH) else 'N/A'}")
+
     if not os.path.exists(DB_PATH):
         print(f"❌ 数据库文件不存在: {DB_PATH}")
         return False
@@ -59,7 +69,13 @@ def backup_database():
     backup_name = f"{BACKUP_PREFIX}_{timestamp}.db"
     config_tar_name = f"{CONFIG_PREFIX}_{timestamp}.tar.gz"
 
-    login(token=token)
+    try:
+        login(token=token)
+        print("[DEBUG] Hugging Face login successful")
+    except Exception as e:
+        print(f"❌ 登录 Hugging Face 失败: {e}")
+        return False
+
     api = HfApi()
 
     # 1. 上传二进制备份
@@ -134,7 +150,9 @@ def backup_database():
 
 if __name__ == "__main__":
     print(f"[{datetime.datetime.now()}] 开始备份及配置导出...")
-    if backup_database():
+    success = backup_database()
+    print(f"[DEBUG] backup_database returned: {success}")
+    if success:
         print("备份任务完成")
         sys.exit(0)
     else: