"""Helper functions related to MySQL DB actions."""
from integration_scripts.connectors import mysql
from integration_scripts.db_utils import run_query

from integration_scripts.sql.select_upc_from_project_code import \
    SELECT_UPC_FROM_PROJECT_CODE


@mysql.ar_db_session_wrap
def get_ar_mysql_results(session, sql, params=None):
    """Return a dict fot run_query() to immediately return a dict object."""
    return run_query(session, sql=sql, params=params)


@mysql.rds_db_session_wrap
def get_rds_mysql_results(session, sql, params=None):
    """Return a dict fot run_query() to immediately return a dict object."""
    return run_query(session, sql=sql, params=params)


@mysql.ar_db_session_wrap
def get_upc_from_project_code(session, project_code):
    """Get all audio assets in ripper audio view."""
    sql = SELECT_UPC_FROM_PROJECT_CODE
    params = {
        'project_code': project_code
    }

    result = run_query(session, sql=sql, params=params)

    res = result.first()
    upc = res.upc

    return upc