注意修改数据库的操作要 执行conn.commit(),否则无法将数据提交给数据库


实例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')
#在python 2.6之后将with改为保留字了,在2.5之前可以使用
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')")
con.commit()
运行结果(在phpmyadmin中查看,在mysql命令行查看结果是一样的):

QQ截图20120608134716

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

结果:

QQ截图20120608144047

 

  • 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引擎,则都不会执行。

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐