"""MySQL utils."""

from typing import List

import pymysql

import config
import constants
from utils import execute_chunks


sql_connection = pymysql.connect(
    host=config.MySQL.HOST,
    port=config.MySQL.PORT,
    user=config.MySQL.USER,
    passwd=config.MySQL.PASSWORD,
    db=config.MySQL.DATABASE
)
sql_cursor = sql_connection.cursor()


@execute_chunks(config.MySQL.BATCH_SIZE)
def insert_user_market(users: List[tuple]):
    """Batch insert user home market ID relation into MySQL.

    Args:
        users (List[dict]):List of users data.

    """
    sql_cursor.executemany(constants.SQL_INSERT_USER_MARKET, users)
    sql_connection.commit()