本文介绍了Python操作MYSQL、执行SQL语句、获取结果集、遍历结果集、取得某个字段、获取表字段名、将图片插入数据库、执行事务等各种代码实例和详细介绍,代码居多,是一桌丰盛唯美的代码大餐。
实例1、取得MYSQL的版本
在windows环境下安装mysql模块用于python开发,请见我的另一篇文章:
MySQL-python Windows下EXE安装文件下载
# -*- coding: UTF-8 -*- #安装MYSQL DB for python import MySQLdb as mdb con = None try: #连接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 " % data finally: if con: #无论如何,连接记得关闭 con.close()
执行结果:
Database version : 5.5.25
实例2、创建一个表并且插入数据
主要还是在cursor上面执行execute方法来进行,请见源码:
# -*- coding: UTF-8 -*- # 来源于www.crazyant.net总结整理 import MySQLdb as mdb import 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 -*- # 来源:www.crazyant.net整理总结 import MySQLdb as mdb import 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 -*- # 来源:疯狂的蚂蚁的博客www.crazyant.net总结整理 import MySQLdb as mdb import 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]
运行结果:
1 Jack London
2 Honore de Balzac
3 Lion Feuchtwanger
4 Emile Zola
5 Truman Capote
- numrows = int(cur.rowcount)用于获取结果集的数目
- row = cur.fetchone()每次取出一行数据,同时记录集的指针执行下一行
实例4、使用字典cursor取得结果集(可以使用表字段名字访问值)
# -*- coding: UTF-8 -*- # 来源:疯狂的蚂蚁的博客www.crazyant.net总结整理 import MySQLdb as mdb import 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 -*- # 来源:疯狂的蚂蚁的博客www.crazyant.net总结整理 import MySQLdb as mdb import sys #获取数据库的链接对象 con = mdb.connect('localhost', 'root', 'root', 'test') with con: #获取普通的查询cursor cur = con.cursor() cur.execute("SELECT * FROM Writers") rows = cur.fetchall() #获取连接对象的描述信息 desc = cur.description print '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 -*- # 来源:疯狂的蚂蚁的博客www.crazyant.net总结整理 import MySQLdb as mdb import sys con = 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 -*- # 来源:疯狂的蚂蚁的博客www.crazyant.net总结整理 import MySQLdb as mdb import sys try: #用读文件模式打开图片 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') #获取执行cursor cursor = conn.cursor() #直接将数据作为字符串,插入数据库 cursor.execute("INSERT INTO Images SET Data='%s'" % mdb.escape_string(img)) #提交数据 conn.commit() #提交之后,再关闭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 -*- # 来源:疯狂的蚂蚁的博客www.crazyant.net总结整理 import MySQLdb as mdb import sys try: #连接mysql,获取连接的对象 conn = mdb.connect('localhost', 'root', 'root', 'test'); cursor = conn.cursor() #执行查询该图片字段的SQL cursor.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 -*- # 来源:疯狂的蚂蚁的博客www.crazyant.net总结整理 import MySQLdb as mdb import sys try: #连接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、原子性的手动提交 conn.commit() cursor.close() conn.close() except mdb.Error, e: #如果出现了错误,那么可以回滚,就是上面的三条语句要么执行,要么都不执行 conn.rollback() print "Error %d: %s" % (e.args[0],e.args[1])
结果:
Error 1146: Table ‘test.writer’ doesn’t exist
2、出现错误,出发异常处理,3条语句的前两条会自动变成了没有执行,结果不变
3、如果本代码放到一个MyISAM引擎表,前两句会执行,第三句不会;如果是INNDB引擎,则都不会执行。