import logging

from fansifter.constants import APPSYNC_PROFILE
from fansifter.hooks.lambda_function import FanSifterLambdaHook


class SaveRuntimeEstimationModelsHook:
    def save_models(self, df):
        payload = {"endpoint": "airflow/progress/save-runtime-models", "df": df.to_dict()}

        lambda_hook = FanSifterLambdaHook(function_name=f"frontend-api-{APPSYNC_PROFILE}-post-enrichment-requests")

        try:
            return lambda_hook.invoke_lambda(payload)
        except Exception as e:
            logging.error(e)
            raise e