本章节讲解Python操作数据库，完成简单的增删改查工作，以MySQL数据库为例。
Python的MySQL数据库操作模块叫MySQLdb，需要额外的安装下。
通过pip工具安装：pip install MySQLdb
MySQLdb模块，我们主要就用到连接数据库的方法MySQLdb.Connect()，连接上数据库后，再使用一些方法做相应的操作。
MySQLdb.Connect(parameters...)方法提供了以下一些常用的参数：
| 参数 |
描述 |
| host | 数据库地址 |
| user | 数据库用户名， |
| passwd | 数据库密码，默认为空 |
| db | 数据库库名，没有默认库 |
| port | 数据库端口，默认3306 |
| connect_timeout | 连接超时时间，秒为单位 |
| use_unicode | 结果以unicode字符串返回 |
| charset | 插入数据库编码 |
连接对象返回的connect()函数：
| commit() | 提交事务。对支持事务的数据库和表，如果提交修改操作，不适用这个方法，则不会写到数据库中 |
| rollback() | 事务回滚。对支持事务的数据库和表，如果执行此方法，则回滚当前事务。在没有commit()前提下。 |
| cursor([cursorclass]) | 创建一个游标对象。所有的sql语句的执行都要在游标对象下进行。MySQL本身不支持游标，MySQLdb模块对其游标进行了仿真。 |
游标对象也提供了几种方法：
| close() | 关闭游标 |
| execute(sql) | 执行sql语句 |
| excutemany(sql) | 执行多条sql语句 |
| fetchone() | 从执行结果中取第一条记录 |
| fetchmany(n) | 从执行结果中取n条记录 |
| fetchall() | 从执行结果中取所有记录 |
| scroll(self, value, mode='relative') | 游标滚动 |
博客地址：http://lizhenliang.blog.51cto.com
QQ群：323779636（Shell/Python运维开发群）
13.1 数据库增删改查
13.1.1 在test库创建一张user表，并添加一条记录
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
>>> conn
=
MySQLdb.Connect(host
=
'192.168.1.244'
,user
=
'root'
,passwd
=
'QHyCTajI'
,db
=
'test'
,charset
=
'utf8'
)
>>> cursor
=
conn.cursor()
>>> sql
=
"create table user(id int,name varchar(30),password varchar(30))"
>>> cursor.execute(sql)
# 返回的数字是影响的行数
0L
>>> sql
=
"insert into user(id,name,password) values('1','xiaoming','123456')"
>>> cursor.execute(sql)
1L
>>> conn.commit()
# 提交事务，写入到数据库
>>> cursor.execute(
'show tables'
)
# 查看创建的表
1L
>>> cursor.fetchall()
# 返回上一个游标执行的所有结果，默认是以元组形式返回
((u
'user'
,),)
>>> cursor.execute(
'select * from user'
)
1L
>>> cursor.fetchall()
((
1L
, u
'xiaoming'
, u
'123456'
),)
|
13.1.2 插入多条数据
|
1
2
3
4
5
6
7
8
9
10
|
>>> sql
=
'insert into user(id,name,password) values(%s,%s,%s)'
>>> args
=
[(
'2'
,
'zhangsan'
,
'123456'
), (
'3'
,
'lisi'
,
'123456'
),(
'4'
,
'wangwu'
,
'123456'
)]
>>> cursor.executemany(sql, args)
3L
>>> conn.commit()
>>> sql
=
'select * from user'
>>> cursor.execute(sql)
4L
>>> cursor.fetchall()
((
1L
, u
'xiaoming'
, u
'123456'
), (
2L
, u
'zhangsan'
, u
'123456'
), (
3L
, u
'lisi'
, u
'123456'
), (
4L
, u
'wangwu'
, u
'123456'
))
|
args变量是一个包含多元组的列表，每个元组对应着每条记录。当查询多条记录时，使用此方法，可有效提高插入效率。
13.1.3 删除用户名xiaoming的记录
|
1
2
3
4
5
6
7
8
9
|
>>> sql
=
'delete from user where name="xiaoming"'
>>> cursor.execute(sql)
1L
>>> conn.commit()
>>> sql
=
'select * from user'
>>> cursor.execute(sql)
3L
>>> cursor.fetchall()
((
2L
, u
'zhangsan'
, u
'123456'
), (
3L
, u
'lisi'
, u
'123456'
), (
4L
, u
'wangwu'
, u
'123456'
))
|
13.1.4 查询记录
|
1
2
3
4
5
6
7
8
9
10
|
>>> sql
=
'select * from user'
>>> cursor.execute(sql)
3L
>>> cursor.fetchone()
# 获取第一条记录
(
2L
, u
'zhangsan'
, u
'123456'
)
>>> sql
=
'select * from user'
>>> cursor.execute(sql)
3L
>>> cursor.fetchmany(
2
)
# 获取两条记录
((
2L
, u
'zhangsan'
, u
'123456'
), (
3L
, u
'lisi'
, u
'123456'
))
|
13.1.4 以字典形式返回结果
默认显示是元组形式，要想返回字典形式，使得更易处理，就用到cursor([cursorclass])中的cusorclass参数。
传入MySQLdb.cursors.DictCursor类：
|
1
2
3
4
5
6
|
>>> cursor
=
conn.cursor(MySQLdb.cursors.DictCursor)
>>> sql
=
'select * from user'
>>> cursor.execute(sql)
3L
>>> cursor.fetchall()
({
'password'
: u
'123456'
,
'id'
:
2L
,
'name'
: u
'zhangsan'
}, {
'password'
: u
'123456'
,
'id'
:
3L
,
'name'
: u
'lisi'
}, {
'password'
: u
'123456'
,
'id'
:
4L
,
'name'
: u
'wangwu'
})
|
13.2 遍历查询结果
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import
MySQLdb
try
:
conn
=
MySQLdb.Connect(host
=
'127.0.0.1'
, port
=
3306
, user
=
'root'
, passwd
=
'123456'
, connect_timeout
=
3
, charset
=
'utf8'
)
cursor
=
conn.cursor()
sql
=
"select * from user"
cursor.execute(sql)
for
i
in
cursor.fetchall():
print
i
except
Exception, e:
print
(
"Connection Error: "
+
str
(e))
finally
:
conn.close()
# python test.py
(
2L
, u
'zhangsan'
, u
'123456'
)
(
3L
, u
'lisi'
, u
'123456'
)
(
4L
, u
'wangwu'
, u
'123456'
)
|
使用for循环遍历查询结果，并增加了异常处理。
本文转自 李振良OK 51CTO博客，原文链接：http://blog.51cto.com/lizhenliang/1874283，如需转载请自行联系原作者
