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