from typing import List, Type
import pg_request as pg
import models as m
import echarts_visualization as ev


def main(target: str,
         target_name: str,
         exog_columns: List[str],
         models: List[Type] = [
             m.VAR_Forecasting, m.ARIMA_Forecasting, m.SARIMA_Forecasting,
             m.RF_Forecasting
         ]):
    pg.run()
    m.run(forecast_target=target, exog_columns=exog_columns, models=models)
    ev.run(target=target, target_name=target_name, models=models)


if __name__ == '__main__':
    main('liugan_index', '流感指数',
         ['infection_number.1', 'infection_number.2', 'jijin_data', 'shoupan'])