simon
366e60e8a9
- 新增 report_db 包: MySQL 连接/建表/幂等写入 (news_report/news_event, myquant 库) - 新增 report_import 包: 历史 178 份日报 HTML 解析入库, 表头驱动列映射 - reporter.py 完全切换: generate_report 结构化入库, 不再生成/上传 HTML - CLI: 新增 report-import 子命令 - 依赖: uv add pymysql; 配置: NEWS_DB_* / REPORT_HISTORY_DIR - 文档: docs/report_db_design.md(实现逻辑), docs/db_schema.md(表结构供 API/前端) - 测试: 24 个单测通过 (parser/builder/models/importer)
35 lines
1021 B
Python
35 lines
1021 B
Python
"""日报结构化入库：数据模型。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import date, datetime
|
|
from typing import Any
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class EventRow(BaseModel):
|
|
"""一条事件记录，对应 news_event 一行。"""
|
|
|
|
section: str # xwlb | news | cninfo | intl
|
|
rank: int # 板块内序号（从 1 开始）
|
|
importance: int | None = None
|
|
event_type: str | None = None
|
|
title: str
|
|
summary: str | None = None
|
|
sentiment: str | None = None # positive | negative | neutral | ''
|
|
source: str | None = None
|
|
url: str | None = None
|
|
|
|
|
|
class ReportData(BaseModel):
|
|
"""一份完整日报：news_report 一行 + news_event 多行。"""
|
|
|
|
report_date: date
|
|
report_type: str # finance | intl
|
|
file_name: str = "" # 源文件名（新生成日报可为空）
|
|
generated_at: datetime
|
|
ai_summary: str | None = None
|
|
stats: dict[str, Any] = Field(default_factory=dict)
|
|
events: list[EventRow] = Field(default_factory=list)
|