wehub-resource-sync
59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
清空数据库中的所有项目数据
|
|
"""
|
|
import sys
|
|
import os
|
|
from pathlib import Path
|
|
|
|
# 添加项目根目录到Python路径
|
|
project_root = Path(__file__).parent
|
|
sys.path.insert(0, str(project_root))
|
|
|
|
from backend.core.database import get_db
|
|
from backend.models.project import Project
|
|
from backend.models.clip import Clip
|
|
from backend.models.collection import Collection
|
|
from backend.models.task import Task
|
|
from sqlalchemy.orm import Session
|
|
|
|
def clean_database():
|
|
"""清空数据库中的所有项目相关数据"""
|
|
print("🧹 开始清理数据库...")
|
|
|
|
# 获取数据库会话
|
|
db = next(get_db())
|
|
|
|
try:
|
|
# 删除所有数据（按依赖关系顺序）
|
|
print("删除任务数据...")
|
|
deleted_tasks = db.query(Task).delete()
|
|
print(f"✅ 删除了 {deleted_tasks} 个任务")
|
|
|
|
print("删除合集数据...")
|
|
deleted_collections = db.query(Collection).delete()
|
|
print(f"✅ 删除了 {deleted_collections} 个合集")
|
|
|
|
print("删除切片数据...")
|
|
deleted_clips = db.query(Clip).delete()
|
|
print(f"✅ 删除了 {deleted_clips} 个切片")
|
|
|
|
print("删除项目数据...")
|
|
deleted_projects = db.query(Project).delete()
|
|
print(f"✅ 删除了 {deleted_projects} 个项目")
|
|
|
|
# 提交事务
|
|
db.commit()
|
|
|
|
print("\n🎉 数据库清理完成!")
|
|
print("现在数据库是干净的，没有任何项目数据")
|
|
|
|
except Exception as e:
|
|
print(f"❌ 清理数据库时发生错误: {e}")
|
|
db.rollback()
|
|
finally:
|
|
db.close()
|
|
|
|
if __name__ == "__main__":
|
|
clean_database()
|