Mysql connector封装

以下代码由PHP站长网 52php.cn收集自互联网

现在PHP站长网小编把它分享给大家，仅供参考

#!/usr/bin/env python
#-*- coding:utf-8-*-

from mysql import connector

def connect():
    config={
        'host':'localhost',#默认127.0.0.1  
        'user':'root','password':'2222211','port':3306,#默认即为3306  
        'database':'test','charset':'utf8'#默认即为utf8  
    }
    try:  
        return connector.connect(**config)
    except connector.Error as e:
        print('connect fail! {}'.format(e))

def execSql(cur,sql,data,isCommit = False):
    cur.executemany(sql,data) if isinstance(data,list) else cur.execute(sql,data)
    if isCommit:
        conn.commit()

def insert(conn,tableName,datas):
    sql = 'insert into `{0}` ({1}) values ({2})'
    keys = []
    placeHolders = []
    
    data = datas
    if isinstance(datas,list):
        data = datas[0]
    
    for key in data:
        keys.append(key)
        placeHolders.append('%('+key+')s')
    sql = sql.format(tableName,','.join(keys),'.join(placeHolders))
    
    id = -1
    try:
        cur = conn.cursor()
        execSql(cur,datas)
        if not isinstance(datas,list):
            id = cur.lastrowid #返回主键
        conn.commit()
    except connector.Error as e:
        raise e
    finally:
        cur.close()
    return id

def query(conn,datas):
    try:
        cur = conn.cursor()
        cur.execute(sql,datas)
    except connector.Error as e:
        raise e
    return cur

if __name__ == '__main__':
    testDict = {
        'aa': 1,'bb': 'aaa','cc': 'bbb','dd': 'ccc'
    }
    conn = connect()
    print insert(conn,'test_table',brandDict)

以上内容由PHP站长网【52php.cn】收集整理供大家参考研究

如果以上内容对您有帮助，欢迎收藏、点赞、推荐、分享。

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!