利用Python操作MySQL数据库

前言

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

本人环境：Python 3.7.0 、MySQL 5.7

安装第三方库

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

pip install PyMySQL

我这里安装的版本是 0.9.3。

PyMySQL版本号

基本操作-增删改查

Python操作MySQL的步骤如下：

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

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

user表

查询操作

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)
最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
  • 人面猴
    序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
    沈念sama阅读 199,830评论 5赞 468
  • 死咒
    序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
    沈念sama阅读 83,992评论 2赞 376
  • 救了他两次的神仙让他今天三更去死
    文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
    开封第一讲书人阅读 146,875评论 0赞 331
  • 道士缉凶录：失踪的卖姜人
    文/不坏的土叔 我叫张陵，是天一观的道长。 经常有香客问我，道长，这世上最难降的妖魔是什么？ 我笑而不...
    开封第一讲书人阅读 53,837评论 1赞 271
  • 港岛之恋（遗憾婚礼）
    正文 为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
    茶点故事阅读 62,734评论 5赞 360
  • 恶毒庶女顶嫁案：这布局不是一般人想出来的
    文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
    开封第一讲书人阅读 48,091评论 1赞 277
  • 城市分裂传说
    那天，我揣着相机与录音，去河边找鬼。 笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。 我是一名探鬼主播，决...
    沈念sama阅读 37,550评论 3赞 390
  • 双鸳鸯连环套：你想象不到人心有多黑
    文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
    开封第一讲书人阅读 36,217评论 0赞 254
  • 万荣杀人案实录
    序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
    沈念sama阅读 40,368评论 1赞 294
  • 护林员之死
    正文 独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,298评论 2赞 317
  • 白月光启示录
    正文 我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,350评论 1赞 329
  • 活死人
    序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
    沈念sama阅读 33,027评论 3赞 315
  • 日本核电站爆炸内幕
    正文 年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
    茶点故事阅读 38,623评论 3赞 303
  • 男人毒药：我在死后第九天来索命
    文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,706评论 0赞 19
  • 一桩弑父案，背后竟有这般阴谋
    文/苍兰香墨 我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,940评论 1赞 255
  • 情欲美人皮
    我被黑心中介骗来泰国打工， 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。 一个月前我还...
    沈念sama阅读 42,349评论 2赞 346
  • 代替公主和亲
    正文 我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
    茶点故事阅读 41,936评论 2赞 341