50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
"""
|
||
清理所有分析内容：删除提取和分析结果，重置报告状态
|
||
用于重新手动分析
|
||
"""
|
||
import asyncio
|
||
import sys
|
||
sys.path.insert(0, ".")
|
||
|
||
from app.database import AsyncSessionLocal, init_db
|
||
from app.models import Report, ExtractedContent, AnalysisResult, AnalysisStatus
|
||
from sqlalchemy import select, delete
|
||
|
||
async def cleanup_analysis():
|
||
print("=" * 60)
|
||
print("清理所有分析内容")
|
||
print("=" * 60)
|
||
|
||
await init_db()
|
||
|
||
async with AsyncSessionLocal() as db:
|
||
# 1. 删除所有分析结果
|
||
result = await db.execute(delete(AnalysisResult))
|
||
analysis_count = result.rowcount
|
||
print(f"✓ 删除了 {analysis_count} 条分析结果")
|
||
|
||
# 2. 删除所有提取内容
|
||
result = await db.execute(delete(ExtractedContent))
|
||
extracted_count = result.rowcount
|
||
print(f"✓ 删除了 {extracted_count} 条提取内容")
|
||
|
||
# 3. 重置报告分析状态（但保留is_extracted，防止定时任务重新触发）
|
||
stmt = select(Report)
|
||
result = await db.execute(stmt)
|
||
reports = result.scalars().all()
|
||
|
||
for report in reports:
|
||
# 保持 is_extracted = True，这样定时任务不会重新处理
|
||
report.is_analyzed = False
|
||
report.analysis_status = AnalysisStatus.PENDING.value
|
||
|
||
await db.commit()
|
||
print(f"✓ 重置了 {len(reports)} 份报告的分析状态（保留提取状态）")
|
||
|
||
print("\n" + "=" * 60)
|
||
print("清理完成！所有报告现在都可以手动点击分析")
|
||
print("=" * 60)
|
||
|
||
if __name__ == "__main__":
|
||
asyncio.run(cleanup_analysis())
|