"""MySQL connection."""

import pymysql
from pymysql import cursors
from pymysql import err

from accounting import config


def get_connection(cursorclass=cursors.DictCursor):
    """Get connection to RDS MySQL.

    Args:
        cursorclass (Cursor): PyMysql cursor class to use

    Returns:
        Connection: Mysql connection object

    """
    try:
        connection = pymysql.connect(
            host=config.MYSQL_HOST,
            port=config.MYSQL_PORT,
            user=config.MYSQL_USER,
            password=config.MYSQL_PASSWORD,
            db=config.MYSQL_DB,
            charset='utf8',
            cursorclass=cursorclass
        )
        return connection
    except err.MySQLError as ex:
        config.log.exception('Failed to connect to MySQL')
        raise ex