目的

将数据库连接的步骤封装成一个方法，在需要连接数据库时，传入指定的参数（SQL）即可实现对数据查询和修改

代码实现

1、编写数据库连接方法
2、导入其他方法中使用

步骤一

import pymysql


def mysqlConnetion(Sql):
    # 数据库连接信息
    host = 'localhost'
    user = 'root'
    password = '12345678'
    database = 'testplat'

    # 建立数据库连接
    try:
        connection = pymysql.connect(
            host=host,
            user=user,
            password=password,
            database=database
        )
        # 创建游标对象
        with connection.cursor() as cursor:
            # 执行 SQL 查询
            sql = Sql
            cursor.execute(sql)
            # 获取查询结果
            result = cursor.fetchall()
            for row in result:
                print(row)
    except pymysql.MySQLError as e:
        print(f"Error connecting to MySQL database: {e}")
    finally:
        # 关闭连接
        if connection:
            connection.close()

步骤二

from DButil.MySqlConnection import mysqlConnetion


# Press the green button in the gutter to run the script.
if __name__ == '__main__':
    Sql = "SELECT * FROM user where id = 1"
    mysqlConnetion(Sql)

实现效果

开发连接方法思考

1、每执行一个自动化测试用例，都需要连接和关闭一次数据库，这部分操作可以简化吗？
2、多个模块的自动化用例执行，会不会对数据库造成很大压力？连接和执行SQL的效率还能保持？

这里就可以引入数据库连接池的概念，可以解决以上问题