import pytest
from paramecio2.libraries.db.webmodel import WebModel
from paramecio2.libraries.db import corefields


def pytest_addoption(parser):
    parser.addoption("--mysql_host", action="store", default="localhost", help="MySQL host: default localhost")
    parser.addoption("--mysql_user", action="store", default="root", help="Mysql User for make the test: default root")
    parser.addoption("--mysql_password", action="store", default="", help="Mysql password for make the test: default empty value")
    parser.addoption("--mysql_db", action="store", default="test_paramecio_db", help="Mysql Database for execute the test: default test_paramecio_db")
    parser.addoption("--mysql_type", action="store", default="pymysql", help="MySQL Python library used: options pymysql mysqldb")  
    
@pytest.fixture
def webmodel_conn(request):
    options={'mysql_host': '', 'mysql_user': '', 'mysql_password': '', 'mysql_db': '', 'mysql_type': ''}
    
    for k in options:
        options[k]=request.config.getoption(k)
    
    # Get connection 
    #'db': options['mysql_db'],
    
    WebModel.connections={'default': {'name': 'default', 'host': options['mysql_host'], 'user': options['mysql_user'], 'password': options['mysql_password'], 'db': options['mysql_db'], 'charset': 'utf8mb4', 'set_connection': False, 'db_type': options['mysql_type']} }
    
    #conn=WebModel.connection()
    
    # Create database using raw mysql methods
    
    if options['mysql_type']=='pymysql':

        import pymysql.cursors
        pymysql.install_as_MySQLdb

        conn=pymysql.connect(host=options['mysql_host'],
                    user=options['mysql_user'],
                    passwd=options['mysql_password'],
                    charset='utf8mb4',
                    cursorclass=pymysql.cursors.DictCursor)
    else:
        import MySQLdb.cursors
        
        conn=MySQLdb.connect(options['mysql_host'],
            user=options['mysql_user'],
            passwd=options['mysql_password'],
            charset='utf8mb4',
            cursorclass=MySQLdb.cursors.DictCursor)
            
    conn.query(
            '''CREATE DATABASE {name}
            DEFAULT CHARACTER SET {charset}'''
            .format(
                name=options['mysql_db'], charset='utf8mb4'
            )
        )
    
    #conn.close()
    
    def drop_database():
        print('Finish him')
        conn.query('DROP DATABASE IF EXISTS %s' % options['mysql_db'])
        conn.close()
        
    request.addfinalizer(drop_database)
    
    final_conn=WebModel.connection()
    
    return final_conn