前言

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

　　本人环境：Python 3.7.0 、MySQL 5.7

　　安装第三方库

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

　　pip install PyMySQL

　　我这里安装的版本是 0.9.3。

　　基本操作-增删改查

　　Python操作MySQL的步骤如下：

　　建立数据库连接

　　通过 cursor() 创建游标对象(pymysql通过游标来执行sql和获取结果)

　　使用 execute() 执行sql

　　获取结果(查询)/提交事务(增删改)

　　关闭游标

　　关闭连接

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

　　查询操作

　　import pymysql

　　def 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() 执行sql

　　cur.execute(select_sql)

　　# 使用 fetchall() 获取所有查询结果

　　data = cur.fetchall()

　　# 关闭游标

　　cur.close()

　　# 关闭数据库连接

　　db.close()

　　return data

　　select_sql = 'SELECT * FROM user WHERE username="张三"'

　　print(select_db(select_sql))

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

　　更新操作

　　import pymysql

　　def 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() 执行sql

　　cur.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 pymysql

　　def 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() 执行sql

　　cur.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 pymysql

　　def 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() 执行sql

　　cur.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时，需要像 更新操作、插入操作 一样，使用事务来处理。

　　简单封装数据库操作

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

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

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

　　import pymysql

　　class 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() 执行sql

　　self.cur.execute(sql)

　　# 使用 fetchall() 获取查询结果

　　data = self.cur.fetchall()

　　return data

　　def execute_db(self, sql):

　　"""更新/插入/删除"""

　　try:

　　# 使用 execute() 执行sql

　　self.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)