"""Query templates for snowflake_copy_mech_deductions task."""

from jinja2 import Template


def copy_mech_deductions_template():
    """Copy results from staging table to distro mech deductions table."""
    return Template("""
        INSERT INTO ROYALTY_ACCOUNTING.{{environment}}.ACCOUNTING_RUN_RESULTS_DISTRO_MECH
        (
            CONTRACT_TXN_ID,
            MECHANICAL_TRANSACTION_ID,
            CONTRACT_MECHANICAL_DEDUCTION_ID,
            TERRITORY,
            MECHANICAL_DEDUCTION_ADMIN_TYPE,
            MECHANICAL_DEDUCTION_RATE,
            PUBLISHER_ADMIN_FEE_RATE,
            MECHANICAL_ROYALTY_AMOUNT_SALE_CURRENCY,
            GROSS_REVENUE_AFTER_WITHHOLDING_TAX_SALE_CURRENCY,
            NET_REVENUE_SALE_CURRENCY,
            MECHANICAL_DEDUCTION_AMOUNT_SALE_CURRENCY,
            PUBLISHER_ADMIN_FEE_SALE_CURRENCY,
            NET_REVENUE_AFTER_MECHANICAL_SALE_CURRENCY,
            MECHANICAL_ROYALTY_AMOUNT_PAYEE_CURRENCY,
            GROSS_REVENUE_AFTER_WITHHOLDING_TAX_PAYEE_CURRENCY,
            NET_REVENUE_PAYEE_CURRENCY,
            MECHANICAL_DEDUCTION_AMOUNT_PAYEE_CURRENCY,
            PUBLISHER_ADMIN_FEE_PAYEE_CURRENCY,
            NET_REVENUE_AFTER_MECHANICAL_PAYEE_CURRENCY
        )
        SELECT
            source.CONTRACT_TXN_ID,
            MECHANICAL_TRANSACTION_ID,
            CONTRACT_MECHANICAL_DEDUCTION_ID,
            TERRITORY,
            MECHANICAL_DEDUCTION_ADMIN_TYPE,
            MECHANICAL_DEDUCTION_RATE,
            PUBLISHER_ADMIN_FEE_RATE,
            MECHANICAL_ROYALTY_AMOUNT_SALE_CURRENCY,
            GROSS_REVENUE_AFTER_WITHHOLDING_TAX_SALE_CURRENCY,
            NET_REVENUE_SALE_CURRENCY,
            MECHANICAL_DEDUCTION_AMOUNT_SALE_CURRENCY,
            PUBLISHER_ADMIN_FEE_SALE_CURRENCY,
            NET_REVENUE_AFTER_MECHANICAL_SALE_CURRENCY,
            MECHANICAL_ROYALTY_AMOUNT_PAYEE_CURRENCY,
            GROSS_REVENUE_AFTER_WITHHOLDING_TAX_PAYEE_CURRENCY,
            NET_REVENUE_PAYEE_CURRENCY,
            MECHANICAL_DEDUCTION_AMOUNT_PAYEE_CURRENCY,
            PUBLISHER_ADMIN_FEE_PAYEE_CURRENCY,
            NET_REVENUE_AFTER_MECHANICAL_PAYEE_CURRENCY
        FROM ROYALTY_ACCOUNTING.{{environment}}.ACCOUNTING_RUN_RESULTS_DISTRO_MECH_STAGING AS source
            INNER JOIN  ROYALTY_ACCOUNTING.{{environment}}.CONTRACT_TRANSACTION_DISTRO as ct
                    ON source.CONTRACT_TXN_ID = ct.CONTRACT_TXN_ID
        WHERE ct.ACCOUNTING_RUN_ID = {{accounting_run_id}}
        """)