"""数据规格和执行计划定义。

定义因子计算所需的数据规格和执行计划结构。
"""

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)