python可以使用多种方法对MySQL数据库进行操作,如MySQLDB、PyMySQL、MYSQL Connector 等,其中 MYSQL Connector是MYSQL官方的数据库驱动,我们可以在MYSQL 的官网上下载和安装,而且在官方网站上还有使用文档,使用起来非常便捷。
MySQL connector 有针对C、C++、ODBC、PHP 、Python等的驱动,我们只需要关注Python的驱动即可,官网网址如下:
目前,有两个 MySQL 驱动:
mysql-connector-python:是 MySQL 官方的纯 Python 驱动;
MySQL-python :是封装了 MySQL C驱动的 Python 驱动。
笔者一般使用 mysql-connector-python,当然,驱动可以在官网上下载安装,也可以使用pip进行安装。
pip install mysql-connector-python
pip install MySQL-python
使用方法也很简单,按照通常的数据库连接即可,当然,如果有其它的应用,如存储过程等的写法,可以参考官网的文档。
一般来说,通过Python读写数据库,要有以下步骤:连接数据库、获取游标、运行SQL语句、获取结果、关闭数据库。
连接数据库:
importmysql.connectordefconnectdb():print('连接到mysql服务器...')#打开数据库连接
#用户名:root, 密码:123456,用户名和密码需要改成你自己的mysql用户名和密码,并且要创建数据库TESTDB。
db = mysql.connector.connect(user="root", passwd="123456.", database="TESTDB", use_unicode=True)print('连接上了!')return db
当然,我们也可以使用字典方式来保存连接参数,使用try......except语句块来捕获错误。
importmysql.connectorfrom mysql.connector importerrorcode#连接参数
config ={'user': 'root','password': '123456','database': 'TESTDB','host': '127.0.0.1',
}defconnectdb():try:print('连接mysql数据库...')#打开数据库连接
db = mysql.connector.connect(**config)#返回数据库连接
returndbexceptmysql.connector.Error as err:if err.errno ==errorcode.ER_ACCESS_DENIED_ERROR:print("用户名或密码错误")elif err.errno ==errorcode.ER_BAD_DB_ERROR:print("数据库不存在")else:print(err)
db= connectdb()
在连上数据库后,我们就可以做相关的增、删、改、查动作了。
创建表:
defcreatetable(db):#使用cursor()方法获取操作游标
cursor =db.cursor()#如果存在表Sutdent先删除
cursor.execute("DROP TABLE IF EXISTS Student")
sql= """CREATE TABLE Student (
ID CHAR(10) NOT NULL,
Name CHAR(8),
Grade INT )"""
#创建Sutdent表
cursor.execute(sql)
插入数据:
definsertdb(db):#使用cursor()方法获取操作游标
cursor =db.cursor()#SQL 插入语句
sql = """INSERT INTO Student
VALUES ('001', 'CZQ', 70),
('002', 'LHQ', 80),
('003', 'MQ', 90),
('004', 'WH', 80),
('005', 'HP', 70),
('006', 'YF', 66),
('007', 'TEST', 100)"""
#sql = "INSERT INTO Student(ID, Name, Grade) \
#VALUES ('%s', '%s', '%d')" % \
#('001', 'HP', 60)
try:#执行sql语句
cursor.execute(sql)#提交到数据库执行
db.commit()except:#Rollback in case there is any error
print '插入数据失败!'db.rollback()
查询数据:
defquerydb(db):#使用cursor()方法获取操作游标
cursor =db.cursor()#SQL 查询语句
#sql = "SELECT * FROM Student \
#WHERE Grade > '%d'" % (80)
sql = "SELECT * FROM Student"
try:#执行SQL语句
cursor.execute(sql)#获取所有记录列表
results =cursor.fetchall()for row inresults:
ID=row[0]
Name= row[1]
Grade= row[2]#打印结果
print "ID: %s, Name: %s, Grade: %d" %\
(ID, Name, Grade)except:print "Error: unable to fecth data"
删除数据:
defdeletedb(db):#使用cursor()方法获取操作游标
cursor =db.cursor()#SQL 删除语句
sql = "DELETE FROM Student WHERE Grade = '%d'" % (100)try:#执行SQL语句
cursor.execute(sql)#提交修改
db.commit()except:print '删除数据失败!'
#发生错误时回滚
db.rollback()
更新数据:
defupdatedb(db):#使用cursor()方法获取操作游标
cursor =db.cursor()#SQL 更新语句
sql = "UPDATE Student SET Grade = Grade + 3 WHERE ID = '%s'" % ('003')try:#执行SQL语句
cursor.execute(sql)#提交到数据库执行
db.commit()except:print '更新数据失败!'
#发生错误时回滚
db.rollback()
关闭数据库连接:
defclosedb(db):
db.close()
然后在 main 主程序中,就可以这样应用了:
defmain():
db= connectdb() #连接MySQL数据库
createtable(db)#创建表
insertdb(db) #插入数据
print '\n插入数据后:'querydb(db)
deletedb(db)#删除数据
print '\n删除数据后:'querydb(db)
updatedb(db)#更新数据
print '\n更新数据后:'querydb(db)
closedb(db)#关闭数据库
if __name__ == '__main__':
main()