Mysql connector封装

yinlin7937

对Mysql connector代码做了封装，可以直接插入dict对象

[Python]代码

1. #!/usr/bin/env python
   
2. #-*- coding:utf-8-*-
   
3. 
   
4. from mysql import connector
   
5. 
   
6. def connect():
   
7.     config={
   
8.         'host':'localhost',#默认127.0.0.1  
   
9.         'user':'root',  
   
10.         'password':'1111111',  
    
11.         'port':3306 ,#默认即为3306  
    
12.         'database':'test',  
    
13.         'charset':'utf8'#默认即为utf8  
    
14.     }
    
15.     try:  
    
16.         return connector.connect(**config)
    
17.     except connector.Error as e:
    
18.         print('connect fail! {}'.format(e))
    
19. 
    
20. def execSql(cur, sql, data, isCommit = False):
    
21.     cur.executemany(sql, data) if isinstance(data, list) else cur.execute(sql, data)
    
22.     if isCommit:
    
23.         conn.commit()
    
24. 
    
25. def insert(conn, tableName, datas):
    
26.     sql = 'insert into `{0}` ({1}) values ({2})'
    
27.     keys = []
    
28.     placeHolders = []
    
29.      
    
30.     data = datas
    
31.     if isinstance(datas, list):
    
32.         data = datas[0]
    
33.      
    
34.     for key in data:
    
35.         keys.append(key)
    
36.         placeHolders.append('%('+key+')s')
    
37.     sql = sql.format(tableName, ','.join(keys), ','.join(placeHolders))
    
38.      
    
39.     id = -1
    
40.     try:
    
41.         cur = conn.cursor()
    
42.         execSql(cur, sql, datas)
    
43.         if not isinstance(datas, list):
    
44.             id = cur.lastrowid #返回主键
    
45.         conn.commit()
    
46.     except connector.Error as e:
    
47.         raise e
    
48.     finally:
    
49.         cur.close()
    
50.     return id
    
51. 
    
52. def query(conn, sql, datas):
    
53.     try:
    
54.         cur = conn.cursor()
    
55.         cur.execute(sql, datas)
    
56.     except connector.Error as e:
    
57.         raise e
    
58.     return cur
    
59. 
    
60. if __name__ == '__main__':
    
61.     testDict = {
    
62.         'aa': 1,
    
63.         'bb': 'aaa',
    
64.         'cc': 'bbb',
    
65.         'dd': 'ccc'
    
66.     }
    
67.     conn = connect()
    
68.     print insert(conn, 'test_table', brandDict)
    

复制代码