python连接mysql及其注意事项

本文主要讲述利用 python 连接数据库的过程和部分注意事项。文章不会涉及到连接的原理,只是简单介绍连接的步骤,以及 mysql 不同的引擎连接的过程的细微区别。

基本上任何语言与数据库进行交互都需要引入外部的数据库驱动,在 python 操作mysql数据库中常用的就是MySQLdb这个驱动,后面也会以这个为例子进行讲解。

数据库的最常见的操作就是“增删查改”,实现这几个功能需要对应的SQL语句。而通过程序连接数据库实际上就是获得与数据库的连接,通过这个连接执行SQL命令,得到返回结果(如果有返回结果的话)。

因此,通过 python 操作 mysql 的步骤可以概括为下面3个步骤: 1.获得连接 2.获取游标 3.通过游标执行SQL语句,获取返回结果(如果有)

** 获得连接**

获得连接的代码如下

1
2
3
4
5
6
7
8
9
10
11
import MySQLdb

HOST = XXXX
PORT = XXXX
USER = XXXX
PASSWD = XXXX
DB = XXXX
CHARSET = XXXX

conn = MySQLdb.connect(host=HOST,port=PORT,user=USER,passwd=PASSWD,db=DB,charset=CHARSET)

一开始声明的几个常量表示要连接到哪台机器的哪个数据库以及采用的编码,注意除了PORT为整数类型,其余都为字符类型。

假如在某个工程中有多个地方需要操作数据库,可以将这个写成一个函数。如下所示:

1
2
3
4
5
6
def connectDB():
try:
conn = MySQLdb.connect(passwd=PASSWD, host=HOST, user=USER, port=PORT, db=DB, charset='utf8')
return conn
except:
return 0

可通过返回值是不是0判断是否建立了正常连接。

获取游标

获得连接后,我们希望做的就是执行我们的 SQL 语句,但是在MySQLdb中,conn并不能执行此操作。需要通过游标(cursor)来执行命令并保存执行的结果,而游标可通过第一步得到的连接获取。代码如下:

1
cursor = conn.cursor()

执行SQL语句,获得返回结果

可以通过cursor.execute(SQL)来执行SQL语句,通过cursor.fetchall()获取返回的结果(针对select语句) 这里获取返回结果针对的是查询(select)语句。

查询的代码代码如下

1
2
3
4
5
6
7
8
9
10
11
SQL = "select * from table"
try:
cursor.execute(SQL)
resultSet = cursor.fetchall()
for row in resultSet:
for i in row:
print i
except:
print 'error while querying'
return 0
finally:
执行的SQL语句返回的结果集可以认为是一个嵌套的两级列表,数据库中每一条记录是一级列表中的一个元素。

插入新纪录的代码如下:

1
2
3
4
5
6
SQL = 'insert into table(field) values(%s) ' %record
try:
cursor.execute(SQL)
except:
print 'error while querying'
return 0

另外删除和更新的代码也类似,只是SQL语句不同而已。

这里需要注意的一点就是上面的代码有可能执行成功,但是数据库中不会更新记录。

原因是mysql的引擎问题,假如mysql的引擎是MyISAM,那么上面的代码就没问题,但是假如mysql的引擎是InnoDB,那么上面的插入新纪录的代码将不会执行成功。因为InnoDB的是支持事务处理的,执行更新的操作会在mysql事先分配的缓存中进行,只有提交后,修改才能生效。提交的操作也很简单,就是在cursor.execute()后加上这句:

1
conn.commit()
关于这个问题,MySQLdb官网说明如下:

commit() If the database and the tables support transactions, this commits the current transaction; otherwise this method successfully does nothing. rollback() If the database and tables support transactions, this rolls back (cancels) the current transaction; otherwise a NotSupportedError is raised.

从说明中可以看到,commit()函数对于支持事务的引擎生效,对于不支持事务的引擎也不会报错,所以建议在代码中均使用。除了commit()函数外,对于支持事务的引擎还有一个rollback()函数用于执行失败后的回滚,但是这个只能在支持事务的引擎上使用。

最后,上个完整的代码

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
# encoding:utf-8
# 将数据文件的数据导入到mysql数据库中

import MySQLdb
import ReadDirFiles

HOST = XXXX
PORT = XXXX
USER = XXXX
PASSWD = XXXX
DB = XXXX
CHARSET = XXXX

def connectDB():
try:
conn = MySQLdb.connect(host=HOST, user=USER, passwd=PASSWD, port=PORT, db=DB, charset='utf8')
return conn
except:
print 'error connecting the database'
return 0

def importData():
conn = connectDB()
if conn == 0:
return 0
cursor = conn.cursor()
SQL = 'insert into `ad_log` values("%s","%s","%s","%s","%s","%s","%s","%s","%s","%s","%s","%s","%s","%s","%s","%s","%s","%s") ' %tuple
# print SQL # 检查SQL语句是否正确
# 下面的代码是一个事务
try:
cursor.execute(SQL)
conn.commit()
print 'successfully insert the record'
except:
conn.rollback() #引擎不支持事务时会报错
finally:
cursor.close()
conn.close()