Flask HttpResponse 응답체에서 openpyxl에서 생성한 Excel 문서를 되돌려줍니다.

4569 단어

Flask HttpResponse 응답체에서 openpyxl에서 생성한 Excel 문서를 되돌려줍니다.


방문할 때 Openpyxl을 통해 Excel 문서를 동적으로 생성하고 응답체로 되돌아오는 링크를 제공해야 합니다.
동적으로 생성된 Excel 문서를 서버에 저장하고 정적 파일 링크로 바꾸는 것은 분명 적당한 방법이 아니다.
따라서 Openpyx에서 생성한 Excel 문서를 Flask의 HttpResponse 대상 응답체에 직접 기록해야 합니다.
원본 코드를 연구한 결과 openpyxl이 excel 파일을 저장할 때save_workbook 이 함수, 이 함수는openpyxl에 있습니다.writer.excel 모듈에서 정의합니다.이 파일을 열면save_라는 파일이 정의되어 있습니다virtual_workbook의 함수, 보아하니 이것이 바로 우리가 찾는 물건인 것 같다.
def save_virtual_workbook(workbook,):
    """Return an in-memory workbook, suitable for a Django response."""
    temp_buffer = BytesIO()
    archive = ZipFile(temp_buffer, 'w', ZIP_DEFLATED, allowZip64=True)

    writer = ExcelWriter(workbook, archive)

    try:
        writer.write_data()
    finally:
        archive.close()

    virtual_workbook = temp_buffer.getvalue()
    temp_buffer.close()
    return virtual_workbook

이 함수는 Excel 문서를 메모리에 쓰고 바이트 그룹을 되돌려줍니다.Flask의 HttpResponse 클래스 구조 방법은 바이트 그룹을 응답체로 받아들일 수 있기 때문에 우리는 이렇게 쓸 수 있다.
from openpyxl import Workbook
from openpyxl.writer.excel import save_virtual_workbook
from flask import make_response

wb = Workbook()
ws = wb.create_sheet('test1')
ws.append(' ')
content = save_virtual_workbook(wb)
resp = make_response(content)
resp.headers["Content-Disposition"] = 'attachment; filename=finance_info.xlsx'
resp.headers['Content-Type'] = 'application/x-xlsx'

그 다음에resp로 돌아가면 됩니다.
만약 이 문장이 당신에게 도움이 된다면 좋아요를 눌러 주십시오. 감사합니다!

좋은 웹페이지 즐겨찾기