from contextlib import contextmanager
from functools import wraps

import mysql.connector
from dynaconf import settings
from intelab_python_sdk.logger import log
from mysql.connector import errorcode
from retrying import retry


class MysqlClient(object):

    def __init__(self, config):
        self.config = dict(
            host=config.get('HOST'),
            port=config.get('PORT'),
            user=config.get('USER'),
            password=config.get('PASSWORD'),
            database=config.get('DATABASE'),
            auth_plugin='mysql_native_password'
        )

    @contextmanager
    def mysql_connector(self, cursor_dict=False):
        """
        构建上下文
        """

        conn = None
        try:
            conn = mysql.connector.connect(**self.config)

            if cursor_dict:
                cursor = conn.cursor(
                    cursor_class=mysql.connector.cursor.MySQLCursorDict)
            else:
                cursor = conn.cursor()

            yield cursor, conn
        except mysql.connector.Error as err:
            if err.errno == errorcode.ER_ACCESS_DENIED_ERROR:
                log.error(
                    "Something is wrong with your user name or password")
            elif err.errno == errorcode.ER_BAD_DB_ERROR:
                log.error("Database does not exist")
            else:
                log.error("cannot connect to mysql. err: {}".format(err))
            raise err

        finally:
            if conn:
                conn.close()


db = MysqlClient(settings.MYSQL)
version = settings.get('version', 'v3')


def query(cursor_dict=False):
    """ 执行sql 语句
    对被装饰的函数参数进行修改，传出mysql的connector和cursor
    默认重试5次

    :param cursor_dict: 查询结果为dict
    """
    def in_func(func):

        @wraps(func)
        @retry(stop_max_attempt_number=5, wait_random_min=100,
               wait_random_max=1000)
        def connect(*args, **kw):
            with db.mysql_connector(cursor_dict) as (cursor, conn):
                return func(cursor, conn, *args, **kw)

        return connect
    return in_func


@query()
def insert_video_info(cursor, conn, device_code, file_name, start_time, end_time,
                      recovered_time, date, video_url, video_resolution, status=0):
    sql = '''
        insert video_data_motion_test (
            device_code, file_name, start_date, end_date,
            recovered_time, date,
            video_url, video_resolution,
            biz_type, service_type,
            expired_time, status, update_time)
        value (%(device_code)s, %(file_name)s, %(start_date)s, %(end_date)s,
            %(recovered_time)s, %(date)s,
            %(video_url)s, %(video_resolution)s,
            1, 1,
            date_add(now(),interval 31 day), %(status)s, now()
        )
    '''
    cursor.execute(sql, {
        'device_code': device_code,
        'file_name': file_name,
        'start_date': start_time,
        'end_date': end_time,
        'video_url': video_url,
        'video_resolution': video_resolution,
        'recovered_time': recovered_time,
        'date': date,
        'status': status
    })
    video_id = cursor.lastrowid
    conn.commit()
    return video_id