def default_get_mysql_connection(
user_name, user_pass, socket, dbname='',
timeout=60, connect_timeout=10, charset=None):
"""
Default method for connection to a MySQL instance.
You can override this behaviour by define/import in cli.py and pass it to
Payload at instantiation time.
The function should return a valid Connection object just as
MySQLdb.Connect does.
"""
connection_config = {
'user': user_name,
'passwd': user_pass,
'unix_socket': socket,
'db': dbname,
'use_unicode': True,
'connect_timeout': connect_timeout
}
if charset:
connection_config['charset'] = charset
dbh = MySQLdb.Connect(**connection_config)
dbh.autocommit(True)
if timeout:
cursor = dbh.cursor()
cursor.execute("SET SESSION WAIT_TIMEOUT = %s", (timeout,))
return dbh
评论列表
文章目录