Python接口自动化测试框架（实战篇）-- 数据库操作MySQL

import pymysql
from DBUtils import PooledDB
 
# 配置数据库连接信息
POOL = PooledDB(
    creator=pymysql,  # 使用链接数据库的模块
    maxconnections=6,  # 连接池允许的最大连接数
    mincached=2,  # 初始化时，链接池中至少创建的空闲的链接
    maxcached=5,  # 链接池中最多闲置的链接
    maxshared=3,  # 链接池中最多共享的链接数量
    blocking=True,  # 连接池中如果没有可用连接后，是否阻塞等待
    host='localhost',  # 数据库服务器地址
    port=3306,  # 数据库服务器端口
    user='root',  # 数据库用户名
    password='123456',  # 数据库密码
    database='test',  # 数据库名
    charset='utf8'  # 数据库编码
)
 
# 获取数据库连接
def get_conn():
    return POOL.connection()
 
# 查询数据库
def query_db(sql):
    conn = get_conn()
    cursor = conn.cursor()
    cursor.execute(sql)
    result = cursor.fetchall()
    cursor.close()
    conn.close()
    return result
 
# 插入数据库
def insert_db(sql):
    conn = get_conn()
    cursor = conn.cursor()
    cursor.execute(sql)
    conn.commit()
    cursor.close()
    conn.close()
 
# 更新数据库
def update_db(sql):
    conn = get_conn()
    cursor = conn.cursor()
    cursor.execute(sql)
    conn.commit()
    cursor.close()
    conn.close()
 
# 删除数据库
def delete_db(sql):
    conn = get_conn()
    cursor = conn.cursor()
    cursor.execute(sql)
    conn.commit()
    cursor.close()
    conn.close()
 
# 示例：查询数据库
result = query_db('SELECT * FROM user')
print(result)
 
# 示例：插入数据库
insert_sql = "INSERT INTO user (username, password) VALUES ('test', '123456')"
insert_db(insert_sql)
 
# 示例：更新数据库
update_sql = "UPDATE user SET password='654321' WHERE username='test'"
update_db(update_sql)
 
# 示例：删除数据库
delete_sql = "DELETE FROM user WHERE username='test'"
delete_db(delete_sql)

这段代码展示了如何使用pymysql和DBUtils.PooledDB模块来管理MySQL数据库连接，并提供了查询、插入、更新和删除数据的基本操作。这样的设计可以提高代码的可维护性和可重用性，并且通过连接池的方式管理数据库连接，可以提高应用程序的性能。