"""Query templates for snowflake_copy_contract_transactions task."""

from jinja2 import Template


def copy_contract_transaction_distro_template():
    """Copy results from staging table to distro transaction table."""
    return Template("""
        INSERT INTO ROYALTY_ACCOUNTING.{{environment}}.CONTRACT_TRANSACTION_DISTRO
        (
            CONTRACT_TXN_ID,
            TXN_ID,
            ACCOUNTING_RUN_ID,
            CONTRACT_ID,
            ACCOUNT_ID,
            PAYEE_CURRENCY_ID,
            CONTRACT_TERM_ID,
            CONTRACT_TERM_CONDITION_ID
        )
        SELECT
            CONTRACT_TXN_ID,
            TXN_ID,
            ACCOUNTING_RUN_ID,
            CONTRACT_ID,
            ACCOUNT_ID,
            PAYEE_CURRENCY_ID,
            CONTRACT_TERM_ID,
            CONTRACT_TERM_CONDITION_ID
        FROM ROYALTY_ACCOUNTING.{{environment}}.CONTRACT_TRANSACTION_STAGING AS source
        WHERE source.ACCOUNTING_RUN_ID = {{accounting_run_id}}
    """)


def copy_contract_transaction_nr_template():
    """Copy results from staging table to NR transaction table."""
    return Template("""
        INSERT INTO ROYALTY_ACCOUNTING.{{environment}}.CONTRACT_TRANSACTION_NR
        (
            CONTRACT_TXN_ID,
            TXN_ID,
            ACCOUNTING_RUN_ID,
            CONTRACT_ID,
            ACCOUNT_ID,
            PAYEE_CURRENCY_ID,
            CONTRACT_TERM_ID,
            CONTRACT_TERM_CONDITION_ID,
            SCHEDULE_ID,
            TERM_RATE
        )
        SELECT
            CONTRACT_TXN_ID,
            TXN_ID,
            ACCOUNTING_RUN_ID,
            CONTRACT_ID,
            ACCOUNT_ID,
            PAYEE_CURRENCY_ID,
            CONTRACT_TERM_ID,
            CONTRACT_TERM_CONDITION_ID,
            SCHEDULE_ID,
            TERM_RATE
        FROM ROYALTY_ACCOUNTING.{{environment}}.CONTRACT_TRANSACTION_NR_STAGING AS source
        WHERE source.ACCOUNTING_RUN_ID = {{accounting_run_id}}
    """)