当前位置：首页 > news >正文

软件测试/测试开发丨Python 常用第三方库 pymysql

news 2025/8/30 0:47:50

pymysql 概述

Python 的数据库接口标准是 Python DB-API
PyMySQL 是从 Python 连接到 MySQL 数据库服务器的接口
PyMySQL 的目标是成为 MySQLdb 的替代品
官方文档：pymysql.readthedocs.io/

pymysql 安装

使用 pip 安装
使用 Pycharm 界面安装

pip install pymysql

pymysql 连接数据库

host：MySQL 服务器地址
user：用户名
password：密码
database：数据库名称
charset：编码方式，推荐使用 utf8mb4

# 1.导入库
import pymysql# 2.建立连接
conn = pymysql.connect(host='服务器地址',user='用户名',password='密码',database='数据库名',charset="utf8mb4")
# 3.关闭连接
conn.close()

pymysql 连接数据库

封装获取连接的函数

import pymysql# 1.封装建立连接的对象
def get_conn():conn = pymysql.connect(host="服务器地址",user="root",password="123456",database="数据库名",charset="utf8mb4")return conn

pymysql 入门实例

获取连接对象
- 打开
- 关闭
获取游标对象
- 执行 SQL
- 查询记录

from . import get_conndef test_demo():# 1.获取连接对象conn = get_conn()# 2.获取游标对象cursor = conn.cursor()# 3.执行SQLcursor.execute("SELECT VERSION()")# 4.查询结果version = cursor.fetchone()print(f"数据库的版本是：{version}")# 5.关闭连接conn.close()

pymysql 创建表

创建表 testcase

from . import get_conndef test_create():conn = get_conn()  # 获取连接cursor = conn.cursor()  # 获取游标sql = """CREATE TABLE `testcase` (`id` int(11) NOT NULL AUTO_INCREMENT,`title` varchar(255) COLLATE utf8_bin NOT NULL,`expect` varchar(255) COLLATE utf8_bin NOT NULL,`owner` varchar(255) COLLATE utf8_bin NOT NULL,PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_bin;"""cursor.execute(sql)  # 执行SQLconn.close()  # 关闭连接

pymysql 插入操作

插入记录

from . import get_conndef test_insert():conn = get_conn()  # 获取连接cursor = conn.cursor()  # 获取游标sql = """INSERT INTO testcase(id, title, expect, owner)values (1, 'S11总决赛', '冠军', 'EDG');"""cursor.execute(sql)  # 执行SQLconn.commit()  # 提交

执行事务

提交操作：commit()
回滚操作：rollback()
try-catch-finally

from . import get_conndef test_insert():conn = get_conn()  # 获取连接cursor = conn.cursor()  # 获取游标sql = """INSERT INTO testcase(id, title, expect, owner)values (2, 'S11全球总决赛', '冠军', 'EDG');"""try:cursor.execute(sql)  # 执行SQLconn.commit()  # 提交事务except:conn.rollback()  # 回滚事务finally:conn.close()  # 关闭连接

pymysql 查询操作

查询操作
- fetchone()：获取单条记录
- fetchmany(n)：获取 n 条记录
- fetchall()：获取所有结果记录

import sys
from . import get_conndef test_retrieve():conn = get_conn()  # 获取连接cursor = conn.cursor()  # 获取游标sql = "SELECT * FROM testcase;"# 捕获异常try:cursor.execute(sql)  # 执行SQLrecord = cursor.fetchone()  # 查询记录print(record)except Exception as e:print(sys.exc_info())  # 打印错误信息finally:conn.close()  # 关闭连接

pymysql 更新操作

更新数据表的数据

from . import get_conndef test_update():conn = get_conn()cursor = conn.cursor()sql = "UPDATE testcase SET owner='hogwarts' WHERE id=2;"try:cursor.execute(sql)  # 执行SQLconn.commit()  # 提交事务except:conn.rollback()  # 回滚事务finally:conn.close()  # 关闭连接

pymysql 删除操作

删除数据表的数据

from . import get_conndef test_delete():conn = get_conn()  # 获取连接cursor = conn.cursor()  # 获取游标sql = "DELETE FROM testcase WHERE id=3;"try:cursor.execute(sql)  # 执行SQLconn.commit()  # 提交事务except:conn.rollback()  # 回滚事务finally:conn.close()  # 关闭连接

最后感谢每一个认真阅读我文章的人，礼尚往来总是要有的，虽然不是什么很值钱的东西，如果你用得到的话可以直接拿走：