"""Used in UPC remapping."""
from ddex_ingester_common.constants.catalog_ingestion import \
    DDEX_PROVIDER_TO_SOURCE_ID
from ddex_ingester_common.helpers.rds import run_rds_query
from ddex_ingester_common.models.state_machine.body import \
    Body as StateMachineBody

import config
from constants.sql_queries import SELECT_ORIGINAL_UPC


def get_remapped_upc_rows(logger, context: StateMachineBody) -> list:
    """Get the UPC remap rows."""
    logger.info('Reading UPC remapping from RDS.')

    upc = context.product.upc
    source_id = DDEX_PROVIDER_TO_SOURCE_ID.get(context.ddex_provider)
    query_args = (upc, source_id)

    return run_rds_query(
        logger,
        config.RDS_HOST,
        config.RDS_DB_NAME,
        config.RDS_RW_USER,
        config.secrets_manager_client.get_cred('rds_read_write_password'),
        SELECT_ORIGINAL_UPC,
        query_args,
    )