Python - 实现数据库的数据导出到 Excel

写一个函数,传入一个表名,将表中所有数据导出到一个 Excel 中。

我们先捋捋这个代码的过程:

  1. 先链接数据库
  2. 通过函数传入的表名,执行 sql 语句读取表中所有数据
  3. 将 sql 执行结果赋值,然后关闭游标和数据库的链接
  4. 实例化一个 Excel 对象,并激活
  5. 将执行 sql 得到的结果写入 Excel 中
  6. 保存 Excel
import pymysql
import openpyxl


# 下边链接信息为虚拟的,大家只要对号入座填写自己的数据库连接信息即可
host = "127.127.127.127”
username = "Mac"
passwd = "123456"
db = "Mac"
port = 3306
charset = "utf8"
excel_name = "students.xlsx"


def wr_Excel(table_name):
    # 连数据库,读取数据
    conn = pymysql.connect(host=host, user=username,
                           passwd=passwd, db=db,
                           port=port, charset=charset)
    cur = conn.cursor()

    sql = "select * from %s;" % table_name
    cur.execute(sql)

    # 使用 %s 占位符可以占位 where 条件,但是不能占位表名
    # 实际为了防止 sql 注入,在where 条件使用占位符时通常使用下述方法
    # sql = "select * from table_name where %s;" (此时假如要输入的变量为 age)
    # cur.execute(sql, age)

    sql_result = cur.fetchall()
    cur.close()
    conn.close()

    # 写 Excel
    book = openpyxl.Workbook()
    sheet = book.active
    fff = [filed[0] for filed in cur.description]  # 获取表头信息
    sheet.append(fff)
    for i in sql_result:
        sheet.append(i)
    book.save("%s" % excel_name)


wr_Excel('faker_user')

  • 8
    点赞
  • 54
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值