import pymysql
from pymysql import cursors

import config


def get_connection(cursorclass=cursors.DictCursor):
    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 Exception as ex:
        print ex