Python 操作 MySQL 实例代码教程

Question

取得 MYSQL 的版本

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

创建一个表并且插入数据

主要还是在cursor上面执行execute方法来进行，请见源码：

# -*- coding: UTF-8 -*-
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')")

使用slect获取mysql的数据并遍历

这个恐怕是用的最多的了，请速看代码：

# -*- coding: UTF-8 -*-
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 -*- # 
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() 每次取出一行数据，同时记录集的指针执行下一行

使用字典cursor取得结果集

（可以使用表字段名字访问值）

# -*- coding: UTF-8 -*- #
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"])

获取单个表的字段名和信息的方法

# -*- coding: UTF-8 -*- #
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

使用Prepared statements执行查询

（更安全方便）

# -*- coding: UTF-8 -*- #
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

把图片用二进制存入 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 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 注入的攻击.

从数据库中把图片读出来

# -*- coding: UTF-8 -*- #
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)

使用 Transaction 即事务

手动提交，自动回滚

# -*- coding: UTF-8 -*- #
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])

1、因为不存在 writer 表（SQL 第三条语句），所以出现错误：

Error 1146: Table ‘test.writer’ doesn’t exist

2、出现错误，出发异常处理，3 条语句的前两条会自动变成了没有执行，结果不变

3、如果本代码放到一个 MyISAM 引擎表，前两句会执行，第三句不会；如果是 INNDB 引擎，则都不会执行。