Python MySQLdb 学习总结(转)

时间 2019-11-21

原文原文链接

转自http://www.cnblogs.com/coser/archive/2012/01/12/2320741.html 感谢@糖拌咸鱼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的接口。python

使用这种数据库接口大可能是就是执行链接数据库->执行query->提取数据->关闭链接这几个步骤。MySQLdb提供比较关键的对象，分别是Connection、Cursor、Result。具体使用步骤很简单先不写了，先写一些我的认为比较重要、值得注意的地方。mysql

一、虽然在MySQLdb.Connect(host ,user , passw , db)函数中，咱们常用的只是这几个参数，可是其实里面还有不少好比字符集、线程安全、ssl等也都是很重要的参数，使用时要身份注意。sql

二、当使用Connection.query()函数进行query后，connection 对象能够返回两种result，分别是store_result和use_result，store_result 将结果集存回client端，而use_result则是结果集保存在server端，而且维护了一个链接，会占用server资源。此时，不能够进行任何其余的查询。建议使用store_result，除非返回结果集（result set）过大或是没法使用limit的情形。数据库

三、提取（fetch）数据的返回形式大多有三种情形。 as a tuple(how=0) ；as dictionaries, key=column or table.column if duplicated(how=1)；as dictionaries, key=table.column (how=2)安全

四、每次fetch，在result内部都会产生数据位置的移动，也就是说假若有10行数据，执行result.fetch_row(3,0)，会获得前三行，再执行result.fetch_row(3,0)，则会获得中间的三行，因此说fetch会致使position的移动。另外值得注意的是，若是使用use_result，也就是数据存储在server时，在fetch全部的条目以前，不能进行任何的query操做。网络

五、mysql自己不支持游标（Cursor），可是MySQLdb对Cursor进行了仿真。重要的执行query方法有execute 和 executemany 。execute方法，执行单条sql语句，调用executemany方法很好用，数据库性能瓶颈很大一部分就在于网络IO和磁盘IO将多个insert放在一块儿，只执行一次IO，能够有效的提高数据库性能。游标cursor具备fetchone、fetchmany、fetchall三个方法提取数据，每一个方法都会致使游标游动，因此必须关注游标的位置。游标的scroll(value, mode)方法可使得游标进行卷动，mode参数指定相对当前位置(relative)仍是以绝对位置(absolute)进行移动。dom

六、MySQLdb提供了不少函数方法，在官方指南里没有彻底罗列，使用者能够用help去看看，里面提供了不少方便的东西。ide

七、对于mysql来讲，若是使用支持事务的存储引擎，那么每次操做后，commit是必须的，不然不会真正写入数据库，对应rollback能够进行相应的回滚，可是commit后是没法再rollback的。commit() 能够在执行不少sql指令后再一次调用，这样能够适当提高性能。函数

八、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)

View Code

#-*- 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()

测试代码