"""Tax Report."""

from pathlib import Path

from snowflake_connector.etl_connector import SnowflakeSQLExecutor


def execute_tax_report_query(
    sf_config,
    period_ids,
    territory_filter,
    payment_interval,
    schema
):
    """Get tax report data from snowflake."""
    raw_sql = Path('royalties/queries/tax_report.sql').read_text()
    territory_condition = ''
    if territory_filter == 'domestic':
        territory_condition = 'AND v.country = 1'
    elif territory_filter == 'foreign':
        territory_condition = 'AND v.country != 1'
    else:
        assert territory_filter == 'all'
    params = {
        'schema': schema,
        'period_ids': period_ids,
        'payment_interval': payment_interval,
        'territory_condition': territory_condition
    }
    with SnowflakeSQLExecutor(sf_config) as sf_executor:
        sql = raw_sql.format(**params)
        return sf_executor.fetchall(sql)