"""Query templates for snowflake_check_ingest_sales task."""

from jinja2 import Template


def check_ingested_sales() -> Template:
    """Check ingest sales data for batch id of a specific sales ingest type."""
    return Template("""
        SELECT
            COUNT(*) as count
        FROM
            ROYALTY_ACCOUNTING.{{env}}.STMT_DB_SALES_{{sales_ingest_type}}_STAGING
        WHERE
            BATCH_ID = '{{batch_id}}'
    """)