import logging

from fansifter.callbacks.slack import on_dag_failure_slack_callback
from fansifter.hooks.collection import CollectionHook


def on_enrichment_failure_callback(context):
    conf = context["dag_run"].conf

    profile = conf["profile"]
    workspace_schema = conf.get("workspace_schema")
    alliance_schema = conf.get("alliance_schema")
    user_id = conf["user_id"]
    collection_id = conf["collection_id"]

    collection_hook = CollectionHook(
        collection_id=collection_id,
        user_id=user_id,
        profile=profile,
        workspace_schema=workspace_schema,
        alliance_schema=alliance_schema,
    )

    try:
        collection_hook.change_status("failed")
    except Exception as e:
        logging.error(e)

    on_dag_failure_slack_callback(**context)