Update scripts/sync_bailian_models.py

scripts/sync_bailian_models.py

@@ -4,6 +4,7 @@ import sys
 import sqlite3
 import json
 import urllib.request
+import subprocess          # 新增：用于调用备份脚本
 from datetime import datetime
 
 DB_PATH = "/app/server/data/freeapi.db"
@@ -37,12 +38,10 @@ def ensure_provider_table(conn):
 
 def get_or_create_bailian_provider(conn, provider_table):
     cursor = conn.cursor()
-    # 尝试查找已有百炼记录
     cursor.execute(f"SELECT id FROM {provider_table} WHERE base_url LIKE '%dashscope.aliyuncs.com%' LIMIT 1")
     row = cursor.fetchone()
     if row:
         return row[0]
-    # 插入新记录
     cursor.execute(f"""
         INSERT INTO {provider_table} (name, base_url, api_key, created_at)
         VALUES (?, ?, ?, ?)
@@ -74,15 +73,10 @@ def main():
     provider_table = ensure_provider_table(conn)
     provider_id = get_or_create_bailian_provider(conn, provider_table)
 
-    # 获取 models 表的列
     model_columns = get_table_columns(conn, "models")
     print(f"models table columns: {model_columns}")
 
-    # 准备插入语句（根据实际列名动态构建）
-    insert_cols = []
-    for col in ['id', 'provider_id', 'enabled', 'created_at', 'name']:
-        if col in model_columns:
-            insert_cols.append(col)
+    insert_cols = [col for col in ['id', 'provider_id', 'enabled', 'created_at', 'name'] if col in model_columns]
     if not insert_cols:
         print("❌ models table missing expected columns")
         return 1
@@ -115,11 +109,21 @@ def main():
     conn.commit()
     conn.close()
     print(f"✅ Sync completed. New models added: {added}")
+
+    # 如果有新增模型，立即触发一次备份，确保重启后不丢失
+    if added > 0:
+        print("Triggering immediate backup to persist new models...")
+        backup_script = "/app/scripts/backup_to_dataset.py"
+        if os.path.exists(backup_script):
+            result = subprocess.run(["python3", backup_script])
+            if result.returncode == 0:
+                print("Backup completed successfully.")
+            else:
+                print("Backup failed, but models were added.")
+        else:
+            print("Backup script not found, skipping immediate backup.")
+
     return 0
 
 if __name__ == "__main__":
-    sys.exit(main())
-# 在 main() 函数最后，return 之前
-if added > 0:
-    print("Triggering immediate backup to persist new models...")
-    subprocess.run(["python3", "/app/scripts/backup_to_dataset.py"], check=False)
+    sys.exit(main())