"""Task to create contract snapshots in snowflake."""

from hooks.royalty_snowflake_hook import RoyaltySnowflakeHook
from lib import config
from tasks.accounting_run_calculate_nr import helpers
from templates.accounting_run_calculate.snowflake_snapshot_contracts_nr import \
    insert_snapshot_contract_nr_template


def insert_snapshot_contracts_nr_task(dag_run: dict, **kwargs) -> None:
    """Query royalty_accounting replica and insert into snowflake snapshot_contract_nr.

    Args:
        dag_run (dict): config of the DAG this task belongs to
        kwargs (dict): any other optional arguments
    """
    event = helpers.get_event_from_params(dag_run, **kwargs)
    accounting_run_id = event.target_id

    hook = RoyaltySnowflakeHook(snowflake_conn_id=config.SNOWFLAKE_CONN_NAME)

    insert_statement = insert_snapshot_contract_nr_template().render(
        accounting_run_id=accounting_run_id,
        env='prod' if config.OWS_ENV.lower() == 'prod' else 'qa',
        schema=config.OWS_ENV,
    )
    hook.run(insert_statement, autocommit=True)