使用 openpyxl 庫從 HTTP 響應解析 xlsx 表

Question

我正在編寫一個測試用例來測試 Excel 表解析。 我嘗試使用 openpyxl 將 response.content 解析為對象列表。 我已經從響應頭中提取了文件名並轉換為類似文件的對象。 load_workbook() 不接受文件名。

def test_export_timesheet(self):
        change_url = '/admin/core/timesheet/'

        #response contains the generated file using openpyxl
        response = self.client.post(change_url, {'action': 'export_xlsx', '_selected_action': [x.id for x in timesheets]})

    content = response._headers.get('content-disposition')[1]
    start = content.find('=') + 1
    end = content.find('.xlsx')
    content_path = (content[start:end]+'.xlsx')

    #Passing file like object
    wb = load_workbook(BytesIO(filename="'"+content_path+"'"))
    ws = wb.get_sheet_by_name(name="'" + content[start:end] + "'")
    for row in ws.iter_rows():
        for cell in row:
            print cell.value

基本上我試圖驗證我的測試用例中文件的內容。 有沒有辦法做到這一點？

Answer 1

# response contains the generated file using openpyxl
response = self.client.post(change_url, ・・・・・

當你得到上面的響應時，“response.content”是字節類型的，所以你可以用 BytesIO 將它加載到緩沖區中。 從上面繼續寫：

from io import BytesIO
file_like_object = BytesIO(response.content)
(from openpyxl import load_workbook) # if this line is needed...
wb = load_workbook(file_like_object)

現在您可以使用這個“wb”進行一般的 openpyxl 操作