Python连接MySQL数据库按条件查询返回键值对格式

import mysql.connector
# MySQL数据库连接参数
config = {
    'user': '',
    'password': '',
    'host': '',
    'database': ''
}

def stock_info(request):
    try:
        code = request.GET.get('code')
        # 连接到MySQL数据库
        cnx = mysql.connector.connect(**config)
        # 创建cursor对象
        cursor = cnx.cursor()
        select_query = f"""SELECT * FROM stock_info WHERE code='{code}'"""
        # 执行SQL命令
        cursor.execute(select_query)
        # 获取查询结果
        results = cursor.fetchall()
        # 查询表中的列信息
        column_query = "SELECT COLUMN_NAME FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_SCHEMA = '%s' AND TABLE_NAME = '%s';" % (config['database'], 'table_name')
        # 执行SQL命令
        cursor.execute(column_query)
        # 获取表头信息
        columns = cursor.fetchall()
        # 将查询结果转换为键值对格式
        results = common.results_to_json(columns, results)
        # 关闭cursor和cnx
        cursor.close()
        cnx.close()
        return common.res_ok(200, results, '')
    except mysql.connector.Error as err:
        print(err)

def results_to_json(columns, results):
    data = []
    keys = []
    for column in columns:
        keys.append(column[0])
    for row in results:
        key_value_pairs = {k: v for k, v in zip(keys, row)}
        data.append(key_value_pairs)
    return data