liaozhaorun
53225b9443
- 新增 api_daily_basic.py 封装 Tushare 每日指标接口 - 因子引擎移除 lookback_days，支持 daily_basic 表字段路由 - 将每日指标纳入自动同步流程 - 删除废弃的 training/main.py
46 lines
1.0 KiB
Python
46 lines
1.0 KiB
Python
"""数据规格和执行计划定义。
|
|
|
|
定义因子计算所需的数据规格和执行计划结构。
|
|
"""
|
|
|
|
from dataclasses import dataclass, field
|
|
from typing import Any, Dict, List, Optional, Set, Union
|
|
|
|
import polars as pl
|
|
|
|
|
|
@dataclass
|
|
class DataSpec:
|
|
"""数据规格定义。
|
|
|
|
描述因子计算所需的数据表和字段。
|
|
|
|
Attributes:
|
|
table: 数据表名称
|
|
columns: 需要的字段列表
|
|
"""
|
|
|
|
table: str
|
|
columns: List[str]
|
|
|
|
|
|
@dataclass
|
|
class ExecutionPlan:
|
|
"""执行计划。
|
|
|
|
包含完整的执行所需信息：数据源、转换逻辑、输出格式。
|
|
|
|
Attributes:
|
|
data_specs: 数据规格列表
|
|
polars_expr: Polars 表达式
|
|
dependencies: 依赖的原始字段
|
|
output_name: 输出因子名称
|
|
factor_dependencies: 依赖的其他因子名称（用于分步执行）
|
|
"""
|
|
|
|
data_specs: List[DataSpec]
|
|
polars_expr: pl.Expr
|
|
dependencies: Set[str]
|
|
output_name: str
|
|
factor_dependencies: Set[str] = field(default_factory=set)
|