Python中操作myslq的方法

实例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])

结果：

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

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

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

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