随笔之pymysql查询结果转字典dict

直接上可用代码

import mysql

def get_sql_conn():
    """
    获取数据库连接
    """
    conn= pymysql.connect(host="*.*.*.*",user="username",password="password",db="dbname")
    cursor = conn.cursor()
    return conn,cursor
    
def get_index_dict(cursor):
    """
    获取数据库对应表中的字段名
    """
    index_dict=dict()
    index=0
    for desc in cursor.description:
        index_dict[desc[0]]=index
        index=index+1
    return index_dict
    
def get_dict_data_sql(cursor,sql):
    """
    运行sql语句，获取结果，并根据表中字段名，转化成dict格式（默认是tuple格式）
    """
    cursor.execute(sql)
    data=cursor.fetchall()
    index_dict=get_index_dict(cursor)
    res=[]
    for datai in data:
        resi=dict()
        for indexi in index_dict:
            resi[indexi]=datai[index_dict[indexi]]
        res.append(resi)
    return res

def main():
    con,cursor = get_sql_conn()
    sql = "SELECT * FROM user WHERE 1=1"
	print(get_dict_data_sql(cursor, sql))

if __name__ == '__main__':
    main()