python3 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()

你可能感兴趣的:(技术,数据库,sql,mysql)