utils.py

import logging
import os
from pathlib import Path

import pandas as pd


def setup_logging(level: str):
    logging.basicConfig(
        level=level,
        format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
        datefmt="%H:%M:%S",
    )

    logging.getLogger("prophet").setLevel(logging.WARNING)
    logging.getLogger("cmdstanpy").setLevel(logging.WARNING)


def save_excel(df: pd.DataFrame, filename: str, output_dir: str):
    Path(output_dir).mkdir(exist_ok=True, parents=True)
    path = os.path.join(output_dir, filename)
    df.to_excel(path, index=False)
    logging.info(f"Results saved to: {path}")