Python-将数据表中数据导出到excel

Posted 2021-12-04 huoxn

‘‘‘
需求：写一个函数，随便输入一个表名，把这个表里面所有的数据，导出到excel里面
思路：
1、‘select * from %s‘ ,查出这个表所有的数据
2、再把所有的数据写到excel  xlwt
‘‘‘


import pymysql,hashlib,xlwt

def op_mysql(sql:str):
    mysql_info = {
        ‘host‘: ‘XXX.XXX.XXX.XXX‘,
        ‘port‘: XXXX,
        ‘password‘: ‘XXXX‘,
        ‘user‘: ‘XXXX‘,
        ‘db‘: ‘XXXX‘,
        ‘charset‘: ‘utf8‘,
        ‘autocommit‘: True
    }
    result = ‘执行完成‘
    conn = pymysql.connect(**mysql_info)
    # cur = conn.cursor(pymysql.cursors.DictCursor) #建立游标
    cur = conn.cursor() #建立游标
    cur.execute(sql)
    field = [ t[0] for t in cur.description ] #获取表里面的所有字段名
    if sql.strip().lower().startswith(‘select‘):
        # result  = cur.fetchone()
        result  = cur.fetchall()
    cur.close()
    conn.close()
    print(‘所有的字段‘,field)
    print(‘所有的数据‘,result)
    result = list(result) #因为返回的是元组，不能修改，所以转成list
    result.insert(0,field) #加入到最前面
    return result

def export_excel(table_name):
    sql =‘select * from %s;‘%table_name
    result = op_mysql(sql)
    book = xlwt.Workbook()
    sheet = book.add_sheet(‘sheet‘)
    for row, line in enumerate(result):
        for col, t in enumerate(line):
            sheet.write(row, col, t)
    book.save(‘%s.xls‘%table_name)
export_excel(‘CJ_test‘)
export_excel(‘app_myuser‘)


import xlwt
book = xlwt.Workbook()
sheet = book.add_sheet(‘sheet‘)l = [
    [1,‘dsk‘,‘xxdfsdfsd‘,0],
    [2,‘ldd‘,‘xxdfsdfsd‘,0],
    [3,‘lsdd‘,‘xxdfsdfsd‘,0],
    [4,‘lsd1‘,‘xxdfsdfsd‘,0],
]

# 方法一：
# row = 0#行号
# for line in l: #外面这个循环，每循环一次，就写excel里面一行
#     # col = 0#列号   [1,‘dsk‘,‘xxdfsdfsd‘,0]
#     col = 0
#     for t in line:#里面这个循环是控制列的，每循环一次就写一列
#         sheet.write(row,col,t)
#         col+=1
#     row+=1
#

# 方法二：
for row,line in enumerate(l):
    for col,t in enumerate(line):
        sheet.write(row,col,t)

book.save(‘user.xls‘)