Python使用MySQLdb切换数据库

转自：http://hi.baidu.com/_yuan0518/blog/item/696907a5da6355eb9052eee5.html

为了实现把博客中的用户帐号和密码导入FTP数据库中，以实现直接使用博客的帐号登陆FTP的功能。我决定使用python协助我处理。这篇日志就是简单介绍，使用python连接Mysql的方法。

一、安装MySQLdb模块
使用python连接Mysql的前提，就是需要一个让python连接到Mysql的接口，这就是MySQLdb模块。
MySQLdb功能介绍：

引用

MySQL support for Python. MySQL versions3.23-5.1; and Python versions 2.3-2.5 are supported. MySQLdb is the Python DBAPI-2.0 interface. _mysql is a low-level API similiar to the MySQL C API.ZMySQLDA is a Database Adapter for Zope2.

在红旗的操作系统中，默认是带有1.0.0版本的MySQL-python模块的：

引用

# rpm -qa|grep MySQL
MySQL-python-1.0.0-1.RHEL4.1

但实际使用发现，该模块版本太旧，对Mysql 4.1以上的支持不完整，甚至连help也存在Bug。所以，建议升级到新的版本。

1、安装新版本
MySQLdb的官方网站是：这里
你可以自行下载编译。也可以使用我已经编译好的rpm包：

                              下载文件

点击这里下载文件

源码包：

下载文件

点击这里下载文件

※该rpm包适用与红旗DC Server 5.0 for x86 SP2版本，python为2.3的系统，其他系统请自行编译使用。
另外，该1.2.2版本MySQLdb模块不能CentOS 5上成功编译，在CentOS 5上使用，可用其自带的1.2.1版本即可。

2、编译安装
新版本1.2.2的MySQLdb所支持的python及Mysql版本与旧版本有所不同。请参考源码中带的README文件。
若不使用rpm方式安装，可按下述命令手动编译和安装：

# cd /usr/local/src/MySQL-python-1.2.2/
# python setup.py build
# python setup.py install

若您的python环境中没有setuptools模块，编译时，请连通互联网，其会从网上下载setuptools模块，并生成.egg压缩模块文件供以后使用。

二、使用MySQLdb模块
MySQLdb模块使用相当简单，只要import MySQLdb环境即可使用。帮助可见：

>>> help ('MySQLdb')
>>> help ('MySQLdb.cursors')

1、导入模块

import MySQLdb

2、连接数据库

conn = MySQLdb.Connection('host', 'user','password', 'dbname')

含义是：

引用

host：Mysql主机
user：连接使用的用户名
password：连接使用的用户名密码
dbname：默认打开的数据库

若该用户可打开其他数据库，可切换数据库：

conn.select_db('database name')

※若打开数据库信息不对，或失败，会触发OperationalError，类似：

引用

>>>conn=MySQLdb.Connection('localhost','linuxing','redflag','proftpd')
Traceback (most recent call last):
File "<stdin>", line 1, in ?
File "build/bdist.linux-i686/egg/MySQLdb/__init__.py", line 74, inConnect
File "build/bdist.linux-i686/egg/MySQLdb/connections.py", line 170,in __init__
_mysql_exceptions.OperationalError: (1045, "Access denied for user'linuxing'@'localhost' (using password: YES)")

请执行处理异常，并确认参数后，再尝试。

3、获取数据库指针
因该模块底层其实是调用C API的，所以，需要先得到当前指向数据库的指针：

cur = conn.cursor()

4、获取数据库信息（select）
先使用指针对象执行SQL语句：

cur.execute('select * from tables')

其返回值为SQL语句得到的行数，如：2L，表示2行。
然后，可以从该对象的fetchone或fetchall方法得到行信息。

5、获取行信息
指针对象的fetchone()方法，是每次得到一行的tuple返回值：

引用

>>> row=cur.fetchone()
>>> print row
('user1', '52c69e3a57331081823331c4e69d3f2e', 1000L, 1000L, '/home/FTP/user1','')

指针对象的fetchall()方法，是得到一组tuple，其内容为由行信息组成的tuple值：

引用

>>> cur.scroll(0,'absolute')
>>> row=cur.fetchall()
>>> print row
(('user1', '52c69e3a57331081823331c4e69d3f2e', 1000L, 1000L, '/home/FTP/user1',''), ('user2', '7e58d63b60197ceb55a1c487989a3720', 1000L, 1000L,'/home/FTP/user2', None))

6、移动指针
当使用fetchone()方法是，指针是会发生移动的。所以，若不重置指针，那么使用fetchall的信息将只会包含指针后面的行内容。
手动移动指针使用：

cur.scroll(int,parm)

含义为：

引用

int：移动的行数，整数；在相对模式下，正数向下移动，负值表示向上移动。
parm：移动的模式，默认是relative，相对模式；可接受absoulte，绝对模式。

※实际上，我觉得应该尽量避免移动指针；而应从select语句得到全部结果后，交给python处理tuple，比较方便。但我们应该明白该指针的含义。

7、修改数据
修改数据，包括插入、更新、删除。它们都是使用指针对象的execute()方法执行：

cur.execute("insert into table (row1,row2) values ('111', '222')")
cur.execute("update table set row1 = 'test' where row2 = 'row2' ")
cur.execute("delete from table where row1 = 'row1' ")

因单引号“'”用于SQL语句中的标识，所以，python中的字符串需使用双引号括住。
此外，也可以使用python的“格式化字符串”写法，简化命令，例如：

cur.execute("update table set row1 ='%s' where row2 = '%s' " %('value1','value2'))

※请注意，'%s'的单引号是SQL语句的间隔符，'value1'的单引号是python的字符串间隔符，其含义是不同的。是否需要间隔符，以及使用双引号还是单引号作为间隔，需根据其含义决定。例如，还有：

cur.execute("update FTPUSERS setpasswd=%s where userid='%s' " %("md5('123')",'user2'))

这里，paswd=%s是因SQL的md5()函数是不需要单引号间隔的；"md5('123')"是python的字符串中含有单引号，所以用双引号括住。

8、提交修改
一般情况下，MySQLdb模块会自动提交修改。但我们在更新数据后，手动运行一次：

conn.commit()

三、注意事项
1、生成rpm包
可以使用：

# python setup.py bdist_rpm

把MySQLdb模块通过setuptools生成rpm包。若你的系统同时存在多个python版本，例如2.4、2.5等，可以增加--python=XXX参数增加python的版本，默认就是python的执行路径。
另外，当我在红旗DC 5.0上生成rpm时，最后执行setup(**metadata)是会报错，但并不影响rpm的生成，最终文件放在dist目录下：

/usr/local/src/MySQL-python-1.2.2/dist

这个错误应该是bdist_rpm参数与setup(**metadata)不兼容的Bug（因为直接build是可以的）。既然不影响使用，后来就没有理会咯。

2、导入模块错误
当我在MySQLdb的源码路径build及install完，并导入模块时，发生警告：

引用

>>> import MySQLdb
/usr/lib/python2.3/site-packages/MySQL_python-1.2.2-py2.3-linux-i686.egg/_mysql.py:3:UserWarning: Module _mysql was already imported from/usr/lib/python2.3/site-packages/MySQL_python-1.2.2-py2.3-linux-i686.egg/_mysql.pyc,but /usr/local/src/MySQL-python-1.2.2 is being added to sys.path

原因是，install后MySQLdb模块已经被放到python的site-pachages目录中；但我在当前目录也存在相同的模块，所以可能会重复导入。只要切换到其他目录运行就可以了。