Как вернуть файл xlsx из памяти в fastapi?

Хочу дать xlsx по запросу. С помощью BytesIO и xlsxwriter я создаю файл.

Используя приведенный ниже код, я могу загрузить пустой (!) .txt файл:

@router.get("/payments/xlsx", response_description='xlsx')
async def payments():
    """sss"""
    output = BytesIO()
    workbook = xlsxwriter.Workbook(output)
    worksheet = workbook.add_worksheet()
    worksheet.write(0, 0, 'ISBN')
    worksheet.write(0, 1, 'Name')
    worksheet.write(0, 2, 'Takedown date')
    worksheet.write(0, 3, 'Last updated')
    workbook.close()
    output.seek(0)
    return StreamingResponse(output)

Если я добавлю headers={'Content-Type': 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'}, я получаю эту ошибку в браузере:

Unable to open file
You may be having a problem connecting with the server, or the file that you wanted to open was corrupted.

Как я могу это исправить?


python-3.x fastapi xlsxwriter bytesio
person Fyzzys    schedule 18.08.2020    source источник

Ответы (1)


arrow_upward
2
arrow_downward

Вы должны установить Content-Disposition заголовок в ответе

@router.get("/payments/xlsx", response_description='xlsx')
async def payments():
    output = BytesIO()
    workbook = xlsxwriter.Workbook(output)
    worksheet = workbook.add_worksheet()
    worksheet.write(0, 0, 'ISBN')
    worksheet.write(0, 1, 'Name')
    worksheet.write(0, 2, 'Takedown date')
    worksheet.write(0, 3, 'Last updated')
    workbook.close()
    output.seek(0)

    headers = {
        'Content-Disposition': 'attachment; filename="filename.xlsx"'
    }
    return StreamingResponse(output, headers=headers)
person JPG    schedule 19.08.2020