码迷,mamicode.com
首页 > 数据库 > 详细

python 连接数据库,查询结果写入数据到excel

时间:2019-11-04 16:01:21      阅读:122      评论:0      收藏:0      [点我收藏+]

标签:就是   成功   book   备份   title   字符串   row   hand   start   

使用Python链接数据库查询数据,并将查询结果写入到Excel中,实现方法上主要有两步,第一,查数据,第二,写Excel。

一、导入需要的包

import time
import xlwt
from commontool import dbtool
import os

 

二、查数据,并返回查询结果

  查询数据传入指定日期,使用指定日期在数据库中查询该日期区间的交易。因为传入的日期为字符串,第一步需要将传入的字符串转换成时间数组,第二步再将传入的日期转换成自己想要的时间格式。

class writefile:
    file = r"F:\python\PycharmProjects\pythonpractice\fileshandle\Files"
def querydata(self,date): """查询数据库结果""" self.date = date datestruct = time.strptime(self.date,"%Y%m%d") # 将字符串转换成时间数组 date = time.strftime("%Y-%m-%d",datestruct) # 将时间转换成其他格式 starttime = date + 00:00:00 endtime = date + 23:59:59 db = dbtool.db() orderquery = "select merchantid,payorderid,state,paytime,innerbank from ordersummary where paytime between ‘%s‘ and ‘%s‘ ; " %(starttime,endtime) res = db.dbquery(db.dbconnect()[3],orderquery).fetchall() return res

三、将查询结果写入到Excel

  将查询结果写入到Excel,这里使用的是xlwt模块,首先要做的是:创建一个Excel,再创建一个sheet页,需要表头的话,也需要写一个表头。表头的格式是一个列表。

  这里需要注意的是:

  (1)range函数是不包含stop数的,所以这里要写入查询的全部数据的话,需要给len(res)+1。比如:查询结果有5个,range(1,4)时,只能是1,2,3,4,就会漏掉5,所以需要给查询结果加上一个1。

  (2)row = 1代表Excel的第二行(第一行是表头),那么在第二行要写入查询的数据的话,应该写入查询数据的第一行,所以 sheet.write(row,col,res[row-1][col]) 中使用了row-1。

    def write_excel(self,res):
        """操作Excel"""
        book = xlwt.Workbook()                                     # 新建一个Excel
        sheet = book.add_sheet(导出数据)                          # 创建sheet
        title = [商户号,流水号,交易状态,交易时间,交易渠道]    # 写表头

        # 循环将表头写入到sheet页
        i=0
        for header in title:
            sheet.write(0,i,header)
            i+=1

        # 写数据
        for row in range(1,len(res)+1):
            for col in range(0,len(res[row-1])):
                sheet.write(row,col,res[row-1][col])
                col+=1
            row+=1
        book.save(self.file+"\交易导出.xls")
        print("导出成功")

四、判断生成Excel是否重复

  文件生成是在一个路径下,要重复生成需要判断该路径下是否存在该文件,如果存在,则删除,或者重命名,再或者移动到备份路径下,这里使用的是重复就删除。

  删除文件这里使用到了os模块。

    def judge_file_exist(self):
        if os.path.exists(self.file+"\交易导出.xls"):
            os.remove(self.file+"\交易导出.xls")

五、主要流程就是以上几步,现在需要主函数将其串起来。

  先判断文件是否存在,然后在调用写文件方法。

    def writefile(self):
        date = 20191028
        self.judge_file_exist()
        self.write_excel(self.querydata(date))

 

python 连接数据库,查询结果写入数据到excel

标签:就是   成功   book   备份   title   字符串   row   hand   start   

原文地址:https://www.cnblogs.com/deliaries/p/11792137.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!