python 快速生成EXCEL及追加数据

# -*- coding: utf-8 -*-
from __future__ import unicode_literals

import oss2
import xlrd
import xlwt
from xlutils.copy import copy


class export_excel(object):
    def __init__(self, p_path, rfilename, sheet_name="sheet_name", title_filed=None):
        self.p_path = p_path  # 文件名字 "/home/xxxx/"
        self.rfilename = rfilename  # 文件名字 "11111.xlsx"
        self.sheet_name = sheet_name
        self.title_filed = title_filed  # 标头

    # 新建excel-并写入表头
    def write_excel_xls(self):
        index = len(self.title_filed)  # 获取需要写入数据的行数
        workbook = xlwt.Workbook()  # 新建一个工作簿
        sheet = workbook.add_sheet(self.sheet_name)  # 在工作簿中新建一个表格
        for i in range(0, index):
            for j in range(0, len(self.title_filed[i])):
                sheet.write(i, j, self.title_filed[i][j])  # 像表格中写入数据（对应的行和列）
        try:
            workbook.save(self.p_path + self.rfilename)  # 保存工作簿
        except Exception as e:
            return {}
        return {"msg": "创建excel文件成功", "code": "200"}

    # 写入数据
    def write_excel_xls_append(self, value):
        index = len(value)  # 获取需要写入数据的行数
        workbook = xlrd.open_workbook(self.p_path + self.rfilename)  # 打开工作簿
        sheets = workbook.sheet_names()  # 获取工作簿中的所有表格
        worksheet = workbook.sheet_by_name(sheets[0])  # 获取工作簿中所有表格中的的第一个表格
        rows_old = worksheet.nrows  # 获取表格中已存在的数据的行数
        new_workbook = copy(workbook)  # 将xlrd对象拷贝转化为xlwt对象
        new_worksheet = new_workbook.get_sheet(0)  # 获取转化后工作簿中的第一个表格

        try:
            for i in range(0, index):
                for j in range(0, len(value[i])):
                    new_worksheet.write(i + rows_old, j, value[i][j])  # 追加写入数据，注意是从i+rows_old行开始写入
            new_workbook.save(str(self.p_path + self.rfilename))  # 保存工作簿
        except Exception as e:
            return {}
        return {"msg": "插入写入数据成功", "code": "200"}

    # 上传至oss-清除本地暂存文件
    def oss_post(self):
        subfilename = "xxxxx/" + datetime.datetime.now().strftime("%Y%m/%d/")  # oss 路径
        file_news = subfilename + self.rfilename
        auth = oss2.Auth(ACCESS_KEY_ID, ACCESS_KEY_SECRET)
        bucket = oss2.Bucket(auth, END_POINT, BUCKET_NAME)
        with open(self.p_path + self.rfilename, "rb") as fileobj:
            result1 = bucket.put_object(file_news, fileobj)
        if int(result1.status) == 200:
            return {"msg": "上传oss成功", "code": "200", "task_url": ""}
        else:
            return {"msg": "上传oss失败", "code": "101"}


if __name__ == "__main__":
    # 新建文件excel
    cls_obj_export = export_excel(p_path=r'E:/GreatZhou/HD-DOC/', rfilename="1.xlsx", sheet_name="娃哈哈", title_filed=[["姓名", "性别", "年龄"], ])
    totuch_ok = cls_obj_export.write_excel_xls()
    if totuch_ok.get("code") == "200":
        wr_result = cls_obj_export.write_excel_xls_append([["大头儿子", "男", "3"], ["小头爸爸", "男", "22"]])
        print wr_result