"""Query templates for update_sales_file_metadata task."""

from jinja2 import Template


def get_distro_sales_file_metadata():
    """Query stmt_db_sales_distro_temp to get total amount (pre tax) and row_count."""
    return Template("""
        SELECT
            SUM(UNIT_PRICE_USD * QUANTITY) AS amount_usd,
            COUNT(*) AS row_count
        FROM ROYALTY_ACCOUNTING.{{env}}.STMT_DB_SALES_DISTRO_TEMP
        WHERE SALES_FILE_ID = {{sales_file_id}};
    """)


def get_nr_sales_file_metadata():
    """Query stmt_db_sales_nr_temp to get total amount (pre tax) and row_count."""
    return Template("""
        SELECT
            SUM(ZEROIFNULL(UNIT_PRICE_USD) * ZEROIFNULL(QUANTITY)) AS amount_usd,
            COUNT(*) AS row_count
        FROM ROYALTY_ACCOUNTING.{{env}}.STMT_DB_SALES_NR_TEMP
        WHERE SALES_FILE_ID = {{sales_file_id}};
    """)