数据库连接、配置文件、日志封装

最新推荐文章于 2025-05-06 18:00:38 发布

原创

最新推荐文章于 2025-05-06 18:00:38 发布 · 690 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数据库 #sql #database

本文详细介绍了如何进行数据库连接的封装，包括在测试用例中的应用，以及配置文件和日志系统的封装。通过yaml文件管理配置，利用log封装实现日志记录，并在测试中整合使用这些封装技术。

一、数据库封装并在用例中使用

1、数据库封装

import pymysql
class DBHandler:

    def __init__(self,
                 host='api.lemonban.com',
                 port=3306,
                 username='future',
                 password='123456',
                 db_name='futureloan'):
        self.connection = pymysql.connect(host=host,
                                     port=port,
                                     user=username,
                                     password=password,
                                     db=db_name)
    def query_one(self,sql):
        cursor = self.connection.cursor()
        self.connection.commit()
        cursor.execute(sql)
        data = cursor.fetchone()
        cursor.close()
        return data
    
    def query_all(self,sql):
        cursor = self.connection.cursor()
        self.connection.commit()
        cursor.execute(sql)
        data = cursor.fetchall()
        cursor.close()
        return dat