独角兽企业重金招聘Python工程师标准>>>
本文介绍了Python操作MYSQL、执行SQL语句、获取结果集、遍历结果集、取得某个字段、获取表字段名、将图片插入数据库、执行事务等各种代码实例和详细介绍，代码居多，是一桌丰盛唯美的代码大餐。
实例1、取得MYSQL的版本
在windows环境下安装mysql模块用于python开发，请见另一篇文章：
MySQL-python Windows下EXE安装文件下载
# -*- coding: UTF-8 -*-#安装MYSQL DB for pythonimport MySQLdb as mdbcon = Nonetry:#连接mysql的方法：connect('ip','user','password','dbname')con = mdb.connect('localhost', 'root','root', 'test');#所有的查询，都在连接con的一个模块cursor上面运行的cur = con.cursor()#执行一个查询cur.execute("SELECT VERSION()")#取得上个查询的结果，是单个结果data = cur.fetchone()print "Database version : %s " % datafinally:if con:#无论如何，连接记得关闭con.close()
执行结果：
Database version : 5.5.25
实例2、创建一个表并且插入数据
主要还是在cursor上面执行execute方法来进行，请见源码：
# -*- coding: UTF-8 -*-# 来源于总结整理import MySQLdb as mdbimport sys#将con设定为全局连接con = mdb.connect('localhost', 'root', 'root', 'test');with con:#获取连接的cursor，只有获取了cursor，我们才能进行各种操作cur = con.cursor()#创建一个数据表 writers(id,name)cur.execute("CREATE TABLE IF NOT EXISTS \Writers(Id INT PRIMARY KEY AUTO_INCREMENT, Name VARCHAR(25))")#以下插入了5条数据cur.execute("INSERT INTO Writers(Name) VALUES('Jack London')")cur.execute("INSERT INTO Writers(Name) VALUES('Honore de Balzac')")cur.execute("INSERT INTO Writers(Name) VALUES('Lion Feuchtwanger')")cur.execute("INSERT INTO Writers(Name) VALUES('Emile Zola')")cur.execute("INSERT INTO Writers(Name) VALUES('Truman Capote')")
运行结果（在phpmyadmin中查看，在mysql命令行查看结果是一样的）：
实例3、python使用slect获取mysql的数据并遍历
这个恐怕是用的最多的了，请速看代码：
# -*- coding: UTF-8 -*-# 来源：整理总结import MySQLdb as mdbimport sys#连接mysql，获取连接的对象con = mdb.connect('localhost', 'root', 'root', 'test');with con:#仍然是，第一步要获取连接的cursor对象，用于执行查询cur = con.cursor()#类似于其他语言的query函数，execute是python中的执行查询函数cur.execute("SELECT * FROM Writers")#使用fetchall函数，将结果集（多维元组）存入rows里面rows = cur.fetchall()#依次遍历结果集，发现每个元素，就是表中的一条记录，用一个元组来显示for row in rows:print row
运行结果：
(1L, ‘Jack London’)
(2L, ‘Honore de Balzac’)
(3L, ‘Lion Feuchtwanger’)
(4L, ‘Emile Zola’)
(5L, ‘Truman Capote’)
上面的代码，用来将所有的结果取出，不过打印的时候是每行一个元祖打印，现在我们使用方法，取出其中的单个数据：
# -*- coding: UTF-8 -*-# 来源：疯狂的蚂蚁的博客总结整理import MySQLdb as mdbimport sys#获取mysql的链接对象con = mdb.connect('localhost', 'root', 'root', 'test');with con:#获取执行查询的对象cur = con.cursor()#执行那个查询，这里用的是select语句cur.execute("SELECT * FROM Writers")#使用cur.rowcount获取结果集的条数numrows = int(cur.rowcount)#循环numrows次，每次取出一行数据for i in range(numrows):#每次取出一行，放到row中，这是一个元组(id,name)row = cur.fetchone()#直接输出两个元素print row[0], row[1]
运行结果：
numrows = int(cur.rowcount)用于获取结果集的数目 row = cur.fetchone()每次取出一行数据，同时记录集的指针执行下一行1 Jack London
2 Honore de Balzac
3 Lion Feuchtwanger
4 Emile Zola
5 Truman Capote
实例4、使用字典cursor取得结果集（可以使用表字段名字访问值）
# -*- coding: UTF-8 -*-# 来源：疯狂的蚂蚁的博客总结整理import MySQLdb as mdbimport sys#获得mysql查询的链接对象con = mdb.connect('localhost', 'root', 'root', 'test')with con:#获取连接上的字典cursor，注意获取的方法，#每一个cursor其实都是cursor的子类cur = con.cursor(mdb.cursors.DictCursor)#执行语句不变cur.execute("SELECT * FROM Writers")#获取数据方法不变rows = cur.fetchall()#遍历数据也不变（比上一个更直接一点）for row in rows:#这里，可以使用键值对的方法，由键名字来获取数据print "%s %s" % (row["Id"], row["Name"])
实例5、获取单个表的字段名和信息的方法
# -*- coding: UTF-8 -*-# 来源：疯狂的蚂蚁的博客总结整理import MySQLdb as mdbimport sys#获取数据库的链接对象con = mdb.connect('localhost', 'root', 'root', 'test')with con:#获取普通的查询cursorcur = con.cursor()cur.execute("SELECT * FROM Writers")rows = cur.fetchall()#获取连接对象的描述信息desc = cur.descriptionprint 'cur.description:',desc#打印表头，就是字段名字print "%s %3s" % (desc[0][0], desc[1][0])for row in rows:#打印结果print "%2s %3s" % row
运行结果：
cur.description: ((‘Id’, 3, 1, 11, 11, 0, 0), (‘Name’, 253, 17, 25, 25, 0, 1))
Id Name
1 Jack London
2 Honore de Balzac
3 Lion Feuchtwanger
4 Emile Zola
5 Truman Capote
实例6、使用Prepared statements执行查询（更安全方便）
# -*- coding: UTF-8 -*-# 来源：疯狂的蚂蚁的博客总结整理import MySQLdb as mdbimport syscon = mdb.connect('localhost', 'root', 'root', 'test')with con: cur = con.cursor()#我们看到，这里可以通过写一个可以组装的sql语句来进行cur.execute("UPDATE Writers SET Name = %s WHERE Id = %s",("Guy de Maupasant", "4"))#使用cur.rowcount获取影响了多少行print "Number of rows updated: %d" % cur.rowcount
结果：
Number of rows updated: 1
实例7、把图片用二进制存入MYSQL
有人喜欢把图片存入MYSQL（这种做法貌似很少吧），我看大部分的程序，图片都是存放在服务器上的文件，数据库中存的只是图片的地址而已，不过MYSQL是支持把图片存入数据库的，也相应的有一个专门的字段BLOB(Binary Large Object)，即较大的二进制对象字段，请看如下程序，注意测试图片自己随便找一个，地址要正确：
首先，在数据库中创建一个表，用于存放图片：
CREATE TABLE Images(Id INT PRIMARY KEY AUTO_INCREMENT, Data MEDIUMBLOB);
然后运行如下PYTHON代码进行：
# -*- coding: UTF-8 -*-# 来源：疯狂的蚂蚁的博客总结整理import MySQLdb as mdbimport systry:#用读文件模式打开图片fin = open("../web.jpg")#将文本读入img对象中img = fin.read()#关闭文件fin.close()except IOError, e:#如果出错，打印错误信息print "Error %d: %s" % (e.args[0],e.args[1])sys.exit(1)try:#链接mysql，获取对象conn = mdb.connect(host='localhost',user='root',passwd='root', db='test')#获取执行cursorcursor = conn.cursor()#直接将数据作为字符串，插入数据库cursor.execute("INSERT INTO Images SET Data='%s'" % mdb.escape_string(img))#提交数据mit()#提交之后，再关闭cursor和链接cursor.close()conn.close()except mdb.Error, e:#若出现异常，打印信息print "Error %d: %s" % (e.args[0],e.args[1])sys.exit(1)
结果：
escape_string函数将插入数据库的字符串进行转义，这会指一些SQL注入的攻击
实例8、从数据库中把图片读出来
# -*- coding: UTF-8 -*-# 来源：疯狂的蚂蚁的博客总结整理import MySQLdb as mdbimport systry:#连接mysql，获取连接的对象conn = mdb.connect('localhost', 'root', 'root', 'test');cursor = conn.cursor()#执行查询该图片字段的SQLcursor.execute("SELECT Data FROM Images LIMIT 1")#使用二进制写文件的方法，打开一个图片文件，若不存在则自动创建fout = open('image.png','wb')#直接将数据如文件fout.write(cursor.fetchone()[0])#关闭写入的文件fout.close()#释放查询数据的资源cursor.close()conn.close()except IOError, e:#捕获IO的异常 ，主要是文件写入会发生错误print "Error %d: %s" % (e.args[0],e.args[1])sys.exit(1)
实例9、使用Transaction即事务（手动提交，自动回滚）
# -*- coding: UTF-8 -*-# 来源：疯狂的蚂蚁的博客总结整理import MySQLdb as mdbimport systry:#连接mysql，获取连接的对象conn = mdb.connect('localhost', 'root', 'root', 'test');cursor = conn.cursor()#如果某个数据库支持事务，会自动开启#这里用的是MYSQL，所以会自动开启事务（若是MYISM引擎则不会）cursor.execute("UPDATE Writers SET Name = %s WHERE Id = %s",("Leo Tolstoy", "1"))cursor.execute("UPDATE Writers SET Name = %s WHERE Id = %s",("Boris Pasternak", "2"))cursor.execute("UPDATE Writer SET Name = %s WHERE Id = %s",("Leonid Leonov", "3")) #事务的特性1、原子性的手动提交mit()cursor.close()conn.close()except mdb.Error, e:#如果出现了错误，那么可以回滚，就是上面的三条语句要么执行，要么都不执行conn.rollback()print "Error %d: %s" % (e.args[0],e.args[1])
结果：
1、因为不存在writer表（SQL第三条语句），所以出现错误：
Error 1146: Table ‘test.writer’ doesn’t exist
2、出现错误，出发异常处理，3条语句的前两条会自动变成了没有执行，结果不变
3、如果本代码放到一个MyISAM引擎表，前两句会执行，第三句不会；如果是INNDB引擎，则都不会执行。
全文完，本文翻译自：/databases/mysqlpythontutorial/
