Python MySQLdb 学习总结(转)
转自http://www.cnblogs.com/coser/archive/2012/01/12/2320741.html 感谢@糖拌咸鱼
任何应用都离不开数据,所以在学习python的时候,当然也要学习一个如何用python操作数据库了。MySQLdb就是python对mysql数据库操作的模块。官方Introduction : MySQLdb is an thread-compatible interface to the popular MySQL database server that provides the Python database API. 它其实相当于翻译了对应C的接口。
使用这种数据库接口大多是就是执行连接数据库->执行query->提取数据->关闭连接 这几个步骤。MySQLdb提供比较关键的对象,分别是Connection、Cursor、Result。具体使用步骤很简单先不写了,先写一些个人认为比较重要、值得注意的地方。
1、虽然在MySQLdb.Connect(host ,user , passw , db)函数中,我们经常使用的只是这几个参数,但是其实里面还有很多比如字符集、线程安全、ssl等也都是很重要的参数,使用时要身份注意。
2、当使用Connection.query()函数进行query后,connection 对象可以返回两种result,分别是store_result和use_result,store_result 将结果集存回client端,而use_result则是结果集保存在server端,并且维护了一个连接,会占用server资源。此时,不可以进行任何其他的查询。建议使用store_result,除非返回结果集(result set)过大或是无法使用limit的情形。
3、提取(fetch)数据的返回形式大多有三种情形。 as a tuple(how=0) ;as dictionaries, key=column or table.column if duplicated(how=1);as dictionaries, key=table.column (how=2)
4、每次fetch,在result内部都会产生数据位置的移动,也就是说假如有10行数据,执行result.fetch_row(3,0),会得到前三行,再执行result.fetch_row(3,0),则会得到中间的三行,所以说fetch会导致position的移动。另外值得注意的是,如果使用use_result,也就是数据存储在server时,在fetch所有的条目之前,不能进行任何的query操作。
5、mysql本身不支持游标(Cursor),但是MySQLdb对Cursor进行了仿真。重要的执行query方法有execute 和 executemany 。execute方法,执行单条sql语句,调用executemany方法很好用,数据库性能瓶颈很大一部分就在于网络IO和磁盘IO将多个insert放在一起,只执行一次IO,可以有效的提升数据库性能。游标cursor具有fetchone、fetchmany、fetchall三个方法提取数据,每个方法都会导致游标游动,所以必须关注游标的位置。游标的scroll(value, mode)方法可以使得游标进行卷动,mode参数指定相对当前位置(relative)还是以绝对位置(absolute)进行移动。
6、MySQLdb提供了很多函数方法,在官方指南里没有完全罗列,使用者可以用help去看看,里面提供了很多方便的东西。
7、对于mysql来说,如果使用支持事务的存储引擎,那么每次操作后,commit是必须的,否则不会真正写入数据库,对应rollback可以进行相应的回滚,但是commit后是无法再rollback的。commit() 可以在执行很多sql指令后再一次调用,这样可以适当提升性能。
8、executemany处理过多的命令也不见得一定好,因为数据一起传入到server端,可能会造成server端的buffer溢出,而一次数据量过大,也有可能产生一些意想不到的麻烦。合理,分批次executemany是个不错的办法。
最后,我自己写了个pyMysql模块,主要是对MySQLdb提供的常用方法进行了简单的再次封装,也借此机会好好学习下MySQLdb,以及练习python的编码。该程序使用的数据库表,采用myisam引擎,所以没加上commit(),一般最好还是要加上的。
测试输出:
******************** lines: 5 ********************
ID : 1 , Author_last : Greene , Author_First : Graham , Country : United Kingdom
ID : 4 , Author_last : Peter , Author_First : David , Country : China
ID : 5 , Author_last : mayday , Author_First : Feng , Country : France
ID : 6 , Author_last : zhang , Author_First : lixin , Country : France
ID : 9 , Author_last : zhang111 , Author_First : lixin , Country : France
******************** 3 行被插入 ********************
******************** lines: 8 ********************
-----------
country : United Kingdom
author_id : 1
author_first : Graham
author_last : Greene
-----------
country : China
author_id : 4
author_first : David
author_last : Peter
-----------
country : France
author_id : 5
author_first : Feng
author_last : mayday
-----------
country : France
author_id : 6
author_first : lixin
author_last : zhang
-----------
country : France
author_id : 9
author_first : lixin
author_last : zhang111
-----------
country : cccccc
author_id : 53
author_first : bbbbbb
author_last : aaaaaa
-----------
country : ffffff
author_id : 54
author_first : eeeeee
author_last : dddddd
-----------
country : iiiiii
author_id : 55
author_first : hhhhhh
author_last : gggggg
1 #-*- encoding:gb2312 -*-_ 2 ''' 3 Created on 2012-1-12 4 @author: xiaojay 5 ''' 6 import MySQLdb 7 import MySQLdb.cursors 8 9 STORE_RESULT_MODE = 0 10 USE_RESULT_MODE = 1 11 12 CURSOR_MODE = 0 13 DICTCURSOR_MODE = 1 14 SSCURSOR_MODE = 2 15 SSDICTCURSOR_MODE = 3 16 17 FETCH_ONE = 0 18 FETCH_MANY = 1 19 FETCH_ALL = 2 20 21 class PyMysql: 22 def __init__(self): 23 self.conn = None 24 pass 25 def newConnection(self,host,user,passwd,defaultdb): 26 """ 27 建立一个新连接,指定host、用户名、密码、默认数据库 28 """ 29 self.conn = MySQLdb.Connect(host,user,passwd,defaultdb) 30 if self.conn.open == False: 31 raise None 32 def closeConnnection(self): 33 """ 34 关闭当前连接 35 """ 36 self.conn.close() 37 38 def query(self,sqltext,mode=STORE_RESULT_MODE): 39 """ 40 作用:使用connection对象的query方法,并返回一个元组(影响行数(int),结果集(result)) 41 参数:sqltext:sql语句 42 mode=STORE_RESULT_MODE(0) 表示返回store_result,mode=USESTORE_RESULT_MODE(1) 表示返回use_result 43 返回:元组(影响行数(int),结果集(result) 44 """ 45 if self.conn==None or self.conn.open==False : 46 return -1 47 self.conn.query(sqltext) 48 if mode == 0 : 49 result = self.conn.store_result() 50 elif mode == 1 : 51 result = self.conn.use_result() 52 else : 53 raise Exception("mode value is wrong.") 54 return (self.conn.affected_rows(),result) 55 56 def fetch_queryresult(self,result,maxrows=1,how=0,moreinfo=False): 57 """ 58 参数:result: query后的结果集合 59 maxrows: 返回的最大行数 60 how: 以何种方式存储结果 61 (0:tuple,1:dictionaries with columnname,2:dictionaries with table.columnname) 62 moreinfo 表示是否获取更多额外信息(num_fields,num_rows,num_fields) 63 返回:元组(数据集,附加信息(当moreinfo=False)或单一数据集(当moreinfo=True) 64 """ 65 if result == None : return None 66 dataset = result.fetch_row(maxrows,how) 67 if moreinfo is False : 68 return dataset 69 else : 70 num_fields = result.num_fields() 71 num_rows = result.num_rows() 72 field_flags = result.field_flags() 73 info = (num_fields,num_rows,field_flags) 74 return (dataset,info) 75 76 def execute(self,sqltext,args=None,mode=CURSOR_MODE,many=False): 77 """ 78 作用:使用游标(cursor)的execute 执行query 79 参数:sqltext: 表示sql语句 80 args: sqltext的参数 81 mode:以何种方式返回数据集 82 CURSOR_MODE = 0 :store_result , tuple 83 DICTCURSOR_MODE = 1 : store_result , dict 84 SSCURSOR_MODE = 2 : use_result , tuple 85 SSDICTCURSOR_MODE = 3 : use_result , dict 86 many:是否执行多行操作(executemany) 87 返回:元组(影响行数(int),游标(Cursor)) 88 """ 89 if mode == CURSOR_MODE : 90 curclass = MySQLdb.cursors.Cursor 91 elif mode == DICTCURSOR_MODE : 92 curclass = MySQLdb.cursors.DictCursor 93 elif mode == SSCURSOR_MODE : 94 curclass = MySQLdb.cursors.SSCursor 95 elif mode == SSDICTCURSOR_MODE : 96 curclass = MySQLdb.cursors.SSDictCursor 97 else : 98 raise Exception("mode value is wrong") 99 100 cur = self.conn.cursor(cursorclass=curclass) 101 line = 0 102 if many == False : 103 if args == None : 104 line = cur.execute(sqltext) 105 else : 106 line = cur.execute(sqltext,args) 107 else : 108 if args == None : 109 line = cur.executemany(sqltext) 110 else : 111 line = cur.executemany(sqltext,args) 112 return (line , cur ) 113 114 def fetch_executeresult(self,cursor,mode=FETCH_ONE,rows=1): 115 """ 116 作用:提取cursor获取的数据集 117 参数:cursor:游标 118 mode:执行提取模式 119 FETCH_ONE: 提取一个; FETCH_MANY :提取rows个 ;FETCH_ALL : 提取所有 120 rows:提取行数 121 返回:fetch数据集 122 """ 123 if cursor == None : 124 return 125 if mode == FETCH_ONE : 126 return cursor.fetchone() 127 elif mode == FETCH_MANY : 128 return cursor.fetchmany(rows) 129 elif mode == FETCH_ALL : 130 return cursor.fetchall() 131 132 if __name__=="__main__" : 133 print help (PyMysql)
#-*- encoding:gb2312 -*- import PyMysql """ authors 这张表很简单。 +--------------+-------------+------+-----+---------+----------------+ | Field | Type | Null | Key | Default | Extra | +--------------+-------------+------+-----+---------+----------------+ | author_id | int(11) | NO | PRI | NULL | auto_increment | | author_last | varchar(50) | YES | | NULL | | | author_first | varchar(50) | YES | MUL | NULL | | | country | varchar(50) | YES | | NULL | | +--------------+-------------+------+-----+---------+----------------+ 本文主要的所有操作都针对该表。 """ def printAuthors(res,mode=0,lines=0): """ 格式化输出 """ print "*"*20, " lines: ",lines ," ","*"*20 if mode==0 : for author_id , author_last , author_first , country in res : print "ID : %s , Author_last : %s , Author_First : %s , Country : %s" \ % (author_id , author_last , author_first , country ) else : for item in res : print "-----------" for key in item.keys(): print key ," : ",item[key] #建立连接 mysql = PyMysql.PyMysql() mysql.newConnection( host="localhost", user="root", passwd="peterbbs", defaultdb="bookstore") "" #定义sql语句 sqltext = "select * from authors order by author_id " #调用query方法,得到result lines , res = mysql.query(sqltext, mode=PyMysql.STORE_RESULT_MODE) #提取数据 data = mysql.fetch_queryresult(res, maxrows=20, how=0, moreinfo=False) #打印 printAuthors(data,0,lines) #演示多行插入 sqltext = "insert into authors (author_last,author_first,country) values (%s,%s,%s)" args = [('aaaaaa','bbbbbb','cccccc'),('dddddd','eeeeee','ffffff'),('gggggg','hhhhhh','iiiiii')] lines ,cur = mysql.execute(sqltext,args,mode=PyMysql.DICTCURSOR_MODE,many=True) print "*"*20, lines ,"行被插入 ","*"*20 sqltext = "select * from authors order by author_id " #调用cursor.execute方法,得到result lines ,cur = mysql.execute(sqltext,mode=PyMysql.DICTCURSOR_MODE) #提取数据 data = mysql.fetch_executeresult(cur, mode=PyMysql.FETCH_MANY, rows=20) #打印 printAuthors(data,1,lines) #关闭连接 mysql.closeConnnection()