本文介绍了Python操作MYSQL、执行SQL语句、获取结果集、遍历结果集、取得某个字段、获取表字段名、将图片插入数据库、执行事务等各种代码实例和详细介绍,代码居多,是一桌丰盛唯美的代码大餐。
实例1、取得MYSQL的版本
在windows环境下安装mysql模块用于python开发,请见我的另一篇文章:
MySQL-python Windows下EXE安装文件下载
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
# -*- 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方法来进行,请见源码:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
# -*- 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的数据并遍历
这个恐怕是用的最多的了,请速看代码:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
# -*- 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’)
上面的代码,用来将所有的结果取出,不过打印的时候是每行一个元祖打印,现在我们使用方法,取出其中的单个数据:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
# -*- 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取得结果集(可以使用表字段名字访问值)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
# -*- 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、获取单个表的字段名和信息的方法
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
# -*- 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执行查询(更安全方便)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
# -*- 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),即较大的二进制对象字段,请看如下程序,注意测试图片自己随便找一个,地址要正确:
首先,在数据库中创建一个表,用于存放图片:
1
|
CREATE
TABLE
Images
(
Id
INT
PRIMARY
KEY
AUTO_INCREMENT
,
Data
MEDIUMBLOB
)
;
|
然后运行如下PYTHON代码进行:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
# -*- 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、从数据库中把图片读出来
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
# -*- 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即事务(手动提交,自动回滚)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
# -*- 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引擎,则都不会执行。