python封装数据库

封装DB，后续对数据库进行操作时，不需要再连一遍

    import pymysql
    def query_db(sql):
        """
        封装DB
        :return:
        """
        # 建立数据库连接
        
        db = pymysql.connect(host="xxx", port=xxx, user="xxx",
                             password="xxx", database="xxx", charset="utf8")

        # 创建游标
        cursor =db.cursor()
        # 使用游标执行sql
        cursor.execute(sql)
        print("行数：", cursor.rowcount)
        datas = cursor.fetchall()
        print("查询到的数据为：", datas)
        cursor.close()
        db.close()
        return datas

以上需替换连接串账号密码等信息~

调用

# 导包
from interface.po.base_api import BaseApi

# 查一条数据测试一下
def test_search():
    sql = f"SELECT * FROM xxx limit 1"
    dat = BaseApi.query_db(sql)
    print(dat)

image.png