Mysql connector封装

下面是脚本之家 jb51.cc 通过网络收集整理的代码片段。

脚本之家小编现在分享给大家，也给大家做个参考。

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

from mysql import connector

def connect():
    config={
        'host':'localhost',#默认127.0.0.1  
        'user':'root','password':'1111111','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)

以上是脚本之家(jb51.cc)为你收集整理的全部代码内容，希望文章能够帮你解决所遇到的程序开发问题。

如果觉得脚本之家网站内容还不错，欢迎将脚本之家网站推荐给程序员好友。

（编辑：安卓应用网）

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