利用Python操做MySQL数据库

前言

在工做中，咱们须要常常对数据库进行操做，好比 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)