"""
models包功能：
该包主要实现了4种主要的时间序列预测模型：向量自回归(VAR)、自回归移动平均(ARIMA)、
季节性自回归移动平均 (SARIMA) 以及随机森林（Random Forest）。

modules:
    - utils.py: 包含用于数据读取和数据保存的通用函数。
    - VAR_model.py: 实现向量自回归 (VAR)预测模型。
    - ARIMA_model.py: 实现自回归移动平均 (ARIMA)预测模型。
    - SARIMA_model.py: 实现季节性自回归移动平均 (SARIMA)预测模型。
    - RF_model.py: 实现随机森林（Random Forest）预测模型。

函数：
    - run(): 是整个预测过程的主函数。该函数首先读取数据，然后调用所有模型进行预测，
    最后把预测结果保存到文件中。

使用示例:
    ```
    from models import run
    run_result = run(forecast_target=target_column,
                     exog_columns=exog_columns_list,
                     models=[VAR_Forecasting])
    ```
"""
from typing import List, Type
from . import utils, VAR_Forecasting, ARIMA_Forecasting, SARIMA_Forecasting, RF_Forecasting

__all__ = [
    'utils', 'VAR_Forecasting', 'ARIMA_Forecasting', 'SARIMA_Forecasting',
    'RF_Forecasting'
]


def run(
    forecast_target: str,
    exog_columns: List[str],
    steps: int = 20,
    file_path: str = 'data/normalized_df.csv',
    models: List[Type] = [
        VAR_Forecasting, ARIMA_Forecasting, SARIMA_Forecasting, RF_Forecasting
    ],
) -> None:
    """
    执行数据读取、预处理、模型训练、预测并保存预测结果等一系列步骤的主函数。

    Args:
        forecast_target (str): 需要被预测的目标变量的列名。
        exog_columns (List[str]): 用于预测的特征变量的列名列表。
        steps (int, default=20): 需要进行预测的步数。
        file_path (str, default='data/normalized_df.csv'): 数据文件的路径。
        models (List[Type]) : 需要运行的预测模型列表，默认包括VAR、ARIMA、SARIMA和Random Forest模型。

    Returns:
        None
    """
    # 载入数据
    input_df = utils.read_csv(file_path)

    # 使用每个模型进行预测并保存结果
    for model in models:
        try:
            model_name = model.__name__.split('.')[-1]
            print(f"正在执行 {model_name} 模型进行预测...")

            # 调用模型进行预测
            model_df = model.run(input_df, forecast_target, exog_columns,
                                 steps)

            # 保存预测结果
            utils.save_csv(model_df, f'data/{model_name}_df.csv')

            print(f"{model_name} 模型的预测结果已保存至 data/{model_name}_df.csv")

        except Exception as e:
            print(f"{model_name} 模型预测过程出现错误: {e}")
    print("所有模型预测都已完成。")