Python操作数据库

前几天数据库课程的一个小project，需要接入MySQL数据库，导入数据，写了个脚本，做简单的CRUD操作，用Python实现，简单地记录一下，原文见Python操作数据库。

依赖

可以用MySQL-python来连MySQL，安装很简单，pip install MySQL-python，然后在脚本里引入import MySQLdb即可。有不止一个库实现类似的功能，API大同小异。

连接

首先要做的是链接数据库，当然要确保你MySQL Server是安装运行的，用homebrew安装的话brew install mysql。

链接数据库之后，会返回一个cursor，主要通过这个cursor执行SQL语句，操作数据库。比如有一个数据库叫MusicDB，链接的函数如下。

import MySQLdb
def connect_db():
    """Connect database and return db and cursor"""
    db = MySQLdb.connect(host="localhost",user='root',
                     	 passwd='PASSWD'',db="MusicDB")
    cursor = db.cursor()
    return db, cursor

当然如果想确认下有没有链接成功的话，可以用如下代码。

db, cursor = connect_db()
cursor.execute("SELECT VERSION()")
data = cursor.fetchone()
print "Database version : %s " % data
db.close()

运行脚本，如果看到类似这样的输出Database versio