64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
"""
|
||
处理已下载的报告：提取内容 + AI分析
|
||
"""
|
||
import asyncio
|
||
import sys
|
||
sys.path.insert(0, ".")
|
||
|
||
from app.database import AsyncSessionLocal, init_db
|
||
from app.models import Report
|
||
from app.services.pdf_extractor import pdf_extractor
|
||
from app.services.ai_analyzer import ai_analyzer
|
||
from sqlalchemy import select
|
||
from sqlalchemy.orm import selectinload
|
||
|
||
async def process_reports():
|
||
print("=" * 60)
|
||
print("处理已下载的报告：提取内容 + AI分析")
|
||
print("=" * 60)
|
||
|
||
await init_db()
|
||
|
||
async with AsyncSessionLocal() as db:
|
||
# 查找已下载但未提取的报告
|
||
stmt = select(Report).where(
|
||
Report.is_downloaded == True,
|
||
Report.is_extracted == False
|
||
).options(selectinload(Report.company))
|
||
|
||
result = await db.execute(stmt)
|
||
reports = result.scalars().all()
|
||
|
||
print(f"\n找到 {len(reports)} 份待处理的报告")
|
||
|
||
for i, report in enumerate(reports, 1):
|
||
company_name = report.company.short_name if report.company else "未知"
|
||
print(f"\n[{i}/{len(reports)}] 处理: {company_name} - {report.title[:30]}...")
|
||
|
||
try:
|
||
# 1. 提取PDF内容
|
||
print(" → 提取PDF内容...")
|
||
contents = await pdf_extractor.extract_and_save(db, report)
|
||
print(f" ✓ 提取了 {len(contents)} 个章节")
|
||
|
||
# 2. AI分析（如果有提取内容）
|
||
if contents:
|
||
print(" → 执行AI分析...")
|
||
success = await ai_analyzer.analyze_report(db, report)
|
||
if success:
|
||
print(" ✓ AI分析完成")
|
||
else:
|
||
print(" ✗ AI分析失败")
|
||
else:
|
||
print(" ⚠ 无提取内容，跳过AI分析")
|
||
|
||
except Exception as e:
|
||
print(f" ✗ 处理失败: {e}")
|
||
|
||
print("\n" + "=" * 60)
|
||
print("处理完成！")
|
||
print("=" * 60)
|
||
|
||
if __name__ == "__main__":
|
||
asyncio.run(process_reports())
|