|
|
# -*- coding: utf-8 -*-
|
|
|
"""
|
|
|
===================================
|
|
|
历史记录接口
|
|
|
===================================
|
|
|
|
|
|
职责：
|
|
|
1. 提供 GET /api/v1/history 历史列表查询接口
|
|
|
2. 提供 GET /api/v1/history/{query_id} 历史详情查询接口
|
|
|
"""
|
|
|
|
|
|
import logging
|
|
|
from typing import Optional
|
|
|
|
|
|
from fastapi import APIRouter, HTTPException, Query, Depends
|
|
|
|
|
|
from api.deps import get_database_manager
|
|
|
from api.v1.schemas.history import (
|
|
|
HistoryListResponse,
|
|
|
HistoryItem,
|
|
|
NewsIntelItem,
|
|
|
NewsIntelResponse,
|
|
|
AnalysisReport,
|
|
|
ReportMeta,
|
|
|
ReportSummary,
|
|
|
ReportStrategy,
|
|
|
ReportDetails,
|
|
|
)
|
|
|
from api.v1.schemas.common import ErrorResponse
|
|
|
from src.storage import DatabaseManager
|
|
|
from src.services.history_service import HistoryService
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
|
|
|
@router.get(
|
|
|
"",
|
|
|
response_model=HistoryListResponse,
|
|
|
responses={
|
|
|
200: {"description": "历史记录列表"},
|
|
|
500: {"description": "服务器错误", "model": ErrorResponse},
|
|
|
},
|
|
|
summary="获取历史分析列表",
|
|
|
description="分页获取历史分析记录摘要，支持按股票代码和日期范围筛选"
|
|
|
)
|
|
|
def get_history_list(
|
|
|
stock_code: Optional[str] = Query(None, description="股票代码筛选"),
|
|
|
start_date: Optional[str] = Query(None, description="开始日期 (YYYY-MM-DD)"),
|
|
|
end_date: Optional[str] = Query(None, description="结束日期 (YYYY-MM-DD)"),
|
|
|
page: int = Query(1, ge=1, description="页码（从 1 开始）"),
|
|
|
limit: int = Query(20, ge=1, le=100, description="每页数量"),
|
|
|
db_manager: DatabaseManager = Depends(get_database_manager)
|
|
|
) -> HistoryListResponse:
|
|
|
"""
|
|
|
获取历史分析列表
|
|
|
|
|
|
分页获取历史分析记录摘要，支持按股票代码和日期范围筛选
|
|
|
|
|
|
Args:
|
|
|
stock_code: 股票代码筛选
|
|
|
start_date: 开始日期
|
|
|
end_date: 结束日期
|
|
|
page: 页码
|
|
|
limit: 每页数量
|
|
|
db_manager: 数据库管理器依赖
|
|
|
|
|
|
Returns:
|
|
|
HistoryListResponse: 历史记录列表
|
|
|
"""
|
|
|
try:
|
|
|
service = HistoryService(db_manager)
|
|
|
|
|
|
# 使用 def 而非 async def，FastAPI 自动在线程池中执行
|
|
|
result = service.get_history_list(
|
|
|
stock_code=stock_code,
|
|
|
start_date=start_date,
|
|
|
end_date=end_date,
|
|
|
page=page,
|
|
|
limit=limit
|
|
|
)
|
|
|
|
|
|
# 转换为响应模型
|
|
|
items = [
|
|
|
HistoryItem(
|
|
|
query_id=item.get("query_id", ""),
|
|
|
stock_code=item.get("stock_code", ""),
|
|
|
stock_name=item.get("stock_name"),
|
|
|
report_type=item.get("report_type"),
|
|
|
sentiment_score=item.get("sentiment_score"),
|
|
|
operation_advice=item.get("operation_advice"),
|
|
|
created_at=item.get("created_at")
|
|
|
)
|
|
|
for item in result.get("items", [])
|
|
|
]
|
|
|
|
|
|
return HistoryListResponse(
|
|
|
total=result.get("total", 0),
|
|
|
page=page,
|
|
|
limit=limit,
|
|
|
items=items
|
|
|
)
|
|
|
|
|
|
except Exception as e:
|
|
|
logger.error(f"查询历史列表失败: {e}", exc_info=True)
|
|
|
raise HTTPException(
|
|
|
status_code=500,
|
|
|
detail={
|
|
|
"error": "internal_error",
|
|
|
"message": f"查询历史列表失败: {str(e)}"
|
|
|
}
|
|
|
)
|
|
|
|
|
|
|
|
|
@router.get(
|
|
|
"/{query_id}",
|
|
|
response_model=AnalysisReport,
|
|
|
responses={
|
|
|
200: {"description": "报告详情"},
|
|
|
404: {"description": "报告不存在", "model": ErrorResponse},
|
|
|
500: {"description": "服务器错误", "model": ErrorResponse},
|
|
|
},
|
|
|
summary="获取历史报告详情",
|
|
|
description="根据 query_id 获取完整的历史分析报告"
|
|
|
)
|
|
|
def get_history_detail(
|
|
|
query_id: str,
|
|
|
db_manager: DatabaseManager = Depends(get_database_manager)
|
|
|
) -> AnalysisReport:
|
|
|
"""
|
|
|
获取历史报告详情
|
|
|
|
|
|
根据 query_id 获取完整的历史分析报告
|
|
|
|
|
|
Args:
|
|
|
query_id: 分析记录唯一标识
|
|
|
db_manager: 数据库管理器依赖
|
|
|
|
|
|
Returns:
|
|
|
AnalysisReport: 完整分析报告
|
|
|
|
|
|
Raises:
|
|
|
HTTPException: 404 - 报告不存在
|
|
|
"""
|
|
|
try:
|
|
|
service = HistoryService(db_manager)
|
|
|
|
|
|
# 使用 def 而非 async def，FastAPI 自动在线程池中执行
|
|
|
result = service.get_history_detail(query_id)
|
|
|
|
|
|
if result is None:
|
|
|
raise HTTPException(
|
|
|
status_code=404,
|
|
|
detail={
|
|
|
"error": "not_found",
|
|
|
"message": f"未找到 query_id={query_id} 的分析记录"
|
|
|
}
|
|
|
)
|
|
|
|
|
|
# 从 context_snapshot 中提取价格信息
|
|
|
current_price = None
|
|
|
change_pct = None
|
|
|
context_snapshot = result.get("context_snapshot")
|
|
|
if context_snapshot and isinstance(context_snapshot, dict):
|
|
|
# 尝试从 enhanced_context.realtime 获取
|
|
|
enhanced_context = context_snapshot.get("enhanced_context") or {}
|
|
|
realtime = enhanced_context.get("realtime") or {}
|
|
|
current_price = realtime.get("price")
|
|
|
change_pct = realtime.get("change_pct") or realtime.get("change_60d")
|
|
|
|
|
|
# 也尝试从 realtime_quote_raw 获取
|
|
|
if current_price is None:
|
|
|
realtime_quote_raw = context_snapshot.get("realtime_quote_raw") or {}
|
|
|
current_price = realtime_quote_raw.get("price")
|
|
|
change_pct = change_pct or realtime_quote_raw.get("change_pct") or realtime_quote_raw.get("pct_chg")
|
|
|
|
|
|
# 构建响应模型
|
|
|
meta = ReportMeta(
|
|
|
query_id=result.get("query_id", query_id),
|
|
|
stock_code=result.get("stock_code", ""),
|
|
|
stock_name=result.get("stock_name"),
|
|
|
report_type=result.get("report_type"),
|
|
|
created_at=result.get("created_at"),
|
|
|
current_price=current_price,
|
|
|
change_pct=change_pct
|
|
|
)
|
|
|
|
|
|
summary = ReportSummary(
|
|
|
analysis_summary=result.get("analysis_summary"),
|
|
|
operation_advice=result.get("operation_advice"),
|
|
|
trend_prediction=result.get("trend_prediction"),
|
|
|
sentiment_score=result.get("sentiment_score"),
|
|
|
sentiment_label=result.get("sentiment_label")
|
|
|
)
|
|
|
|
|
|
strategy = ReportStrategy(
|
|
|
ideal_buy=result.get("ideal_buy"),
|
|
|
secondary_buy=result.get("secondary_buy"),
|
|
|
stop_loss=result.get("stop_loss"),
|
|
|
take_profit=result.get("take_profit")
|
|
|
)
|
|
|
|
|
|
details = ReportDetails(
|
|
|
news_content=result.get("news_content"),
|
|
|
raw_result=result.get("raw_result"),
|
|
|
context_snapshot=result.get("context_snapshot")
|
|
|
)
|
|
|
|
|
|
return AnalysisReport(
|
|
|
meta=meta,
|
|
|
summary=summary,
|
|
|
strategy=strategy,
|
|
|
details=details
|
|
|
)
|
|
|
|
|
|
except HTTPException:
|
|
|
raise
|
|
|
except Exception as e:
|
|
|
logger.error(f"查询历史详情失败: {e}", exc_info=True)
|
|
|
raise HTTPException(
|
|
|
status_code=500,
|
|
|
detail={
|
|
|
"error": "internal_error",
|
|
|
"message": f"查询历史详情失败: {str(e)}"
|
|
|
}
|
|
|
)
|
|
|
|
|
|
|
|
|
@router.get(
|
|
|
"/{query_id}/news",
|
|
|
response_model=NewsIntelResponse,
|
|
|
responses={
|
|
|
200: {"description": "新闻情报列表"},
|
|
|
500: {"description": "服务器错误", "model": ErrorResponse},
|
|
|
},
|
|
|
summary="获取历史报告关联新闻",
|
|
|
description="根据 query_id 获取关联的新闻情报列表（为空也返回 200）"
|
|
|
)
|
|
|
def get_history_news(
|
|
|
query_id: str,
|
|
|
limit: int = Query(20, ge=1, le=100, description="返回数量限制"),
|
|
|
db_manager: DatabaseManager = Depends(get_database_manager)
|
|
|
) -> NewsIntelResponse:
|
|
|
"""
|
|
|
获取历史报告关联新闻
|
|
|
|
|
|
Args:
|
|
|
query_id: 分析记录唯一标识
|
|
|
limit: 返回数量限制
|
|
|
db_manager: 数据库管理器依赖
|
|
|
|
|
|
Returns:
|
|
|
NewsIntelResponse: 新闻情报列表
|
|
|
"""
|
|
|
try:
|
|
|
service = HistoryService(db_manager)
|
|
|
items = service.get_news_intel(query_id=query_id, limit=limit)
|
|
|
|
|
|
response_items = [
|
|
|
NewsIntelItem(
|
|
|
title=item.get("title", ""),
|
|
|
snippet=item.get("snippet"),
|
|
|
url=item.get("url", "")
|
|
|
)
|
|
|
for item in items
|
|
|
]
|
|
|
|
|
|
return NewsIntelResponse(
|
|
|
total=len(response_items),
|
|
|
items=response_items
|
|
|
)
|
|
|
|
|
|
except Exception as e:
|
|
|
logger.error(f"查询新闻情报失败: {e}", exc_info=True)
|
|
|
raise HTTPException(
|
|
|
status_code=500,
|
|
|
detail={
|
|
|
"error": "internal_error",
|
|
|
"message": f"查询新闻情报失败: {str(e)}"
|
|
|
}
|
|
|
)
|
