用Python操作MySQL数据库-详细指南

在工作中，我们需要经常对数据库进行操作，比如 Oracle、MySQL、SQL Sever 等，今天我们就学习如何利用Python来操作 MySQL 数据库。

1.安装第三方库

使用Python来操作MySQL，需要用到 PyMySQL 这个第三方库，具体安装如下：

pip install PyMySQL

我这里安装的版本是 0.9.3。

2.基本操作-增删改查

Python操作MySQL的步骤如下：

建立数据库连接
通过 cursor() 创建游标对象（pymysql通过游标来执行sql和获取结果）
使用 execute() 执行sql
获取结果（查询）/提交事务（增删改）
关闭游标
关闭连接

注意：用Python进行增删改查操作之前，最好先检查下sql语句是否正确，确保没有错误之后，再放到Python代码中。

查询操作

import pymysqldef select_db(select_sql):"""查询"""# 建立数据库连接db = pymysql.connect(host="192.168.89.128",port=3306,user="root",passwd="123456",db="test2020")# 通过 cursor() 创建游标对象，并让查询结果以字典格式输出cur = db.cursor(cursor=pymysql.cursors.DictCursor)# 使用 execute() 执行sqlcur.execute(select_sql)# 使用 fetchall() 获取所有查询结果data = cur.fetchall()# 关闭游标cur.close()# 关闭数据库连接db.close()return dataselect_sql = 'SELECT * FROM user WHERE username="张三"'
print(select_db(select_sql))

在上面，创建游标时使用了 cursor=pymysql.cursors.DictCursor ，目的是为了让查询结果以列表嵌套字典的格式输出，以方便后续操作。如果不使用该参数，那么返回的查询结果将是元组嵌套元组的形式。

更新操作

import pymysqldef update_db(update_sql):"""更新"""# 建立数据库连接db = pymysql.connect(host="192.168.89.128",port=3306,user="root",passwd="123456",db="test2020")# 通过 cursor() 创建游标对象cur = db.cursor()try:# 使用 execute() 执行sqlcur.execute(update_sql)# 提交事务db.commit()except Exception as e:print("操作出现错误：{}".format(e))# 回滚所有更改db.rollback()finally:# 关闭游标cur.close()# 关闭数据库连接db.close()update_sql = 'UPDATE user SET username = "张三2" WHERE id = 1'
update_db(update_sql)

在执行更新操作sql时，需要使用事务处理，即使用创建的连接进行提交事务 db.commit() ，使当前更改生效。当执行sql语句出现错误时，还应该支持事务回滚操作 db.rollback() 。

新增操作

import pymysqldef insert_db(insert_sql):"""插入"""# 建立数据库连接db = pymysql.connect(host="192.168.89.128",port=3306,user="root",passwd="123456",db="test2020")# 通过 cursor() 创建游标对象cur = db.cursor()try:# 使用 execute() 执行sqlcur.execute(insert_sql)# 提交事务db.commit()except Exception as e:print("操作出现错误：{}".format(e))# 回滚所有更改db.rollback()finally:# 关闭游标cur.close()# 关闭数据库连接db.close()insert_sql = 'INSERT INTO user(id, username, password) VALUES(11, "王五", "333333")'
insert_db(insert_sql)

在执行新增操作sql时，需要像更新操作一样，使用事务来处理。

删除操作

import pymysqldef delete_db(delete_sql):"""删除"""# 建立数据库连接db = pymysql.connect(host="192.168.89.128",port=3306,user="root",passwd="123456",db="test2020")# 通过 cursor() 创建游标对象cur = db.cursor()try:# 使用 execute() 执行sqlcur.execute(delete_sql)# 提交事务db.commit()except Exception as e:print("操作出现错误：{}".format(e))# 回滚所有更改db.rollback()finally:# 关闭游标cur.close()# 关闭数据库连接db.close()delete_sql = 'DELETE FROM user WHERE id = 11'
delete_db(delete_sql)

在执行删除操作sql时，需要像更新操作、插入操作一样，使用事务来处理。

3.简单封装数据库操作

从上面的代码中，我们可以发现存在不少问题，比如增删改操作的代码除了执行的sql不同外，其他代码都是一样的；同时每个操作时都写了一次建立数据库连接的代码，也就是说每次操作都会建立和删除连接的操作。

针对上面的问题，我们完全可以对上面的增删改查操作进行简单的封装，把Python操作MySQL的代码封装到一个类中，只在创建对象实例的时候建立数据库连接，只在对象删除时关闭连接。

以下是对数据库操作的简单封装：

'''
学习中遇到问题没人解答？小编创建了一个Python学习交流群：711312441
寻找有志同道合的小伙伴，互帮互助,群里还有不错的视频学习教程和PDF电子书！
'''
import pymysqlclass MysqlDb():def __init__(self, host, port, user, passwd, db):# 建立数据库连接self.conn = pymysql.connect(host=host,port=port,user=user,passwd=passwd,db=db)# 通过 cursor() 创建游标对象，并让查询结果以字典格式输出self.cur = self.conn.cursor(cursor=pymysql.cursors.DictCursor)def __del__(self): # 对象资源被释放时触发，在对象即将被删除时的最后操作# 关闭游标self.cur.close()# 关闭数据库连接self.conn.close()def select_db(self, sql):"""查询"""# 使用 execute() 执行sqlself.cur.execute(sql)# 使用 fetchall() 获取查询结果data = self.cur.fetchall()return datadef execute_db(self, sql):"""更新/插入/删除"""try:# 使用 execute() 执行sqlself.cur.execute(sql)# 提交事务self.conn.commit()except Exception as e:print("操作出现错误：{}".format(e))# 回滚所有更改self.conn.rollback()if __name__ == '__main__':db = MysqlDb("192.168.89.128", 3306, "root", "123456", "test2020")select_sql = 'SELECT * FROM user WHERE username="张三2"'update_sql = 'UPDATE user SET username = "张三2" WHERE id = 1'insert_sql = 'INSERT INTO user(id, username, password) VALUES(11, "王五", "333333")'delete_sql = 'DELETE FROM user WHERE id = 11'data = db.select_db(select_sql)print(data)db.execute_db(update_sql)db.execute_db(insert_sql)db.execute_db(delete_sql)