Как используя BytesIO создать эксель файл и отправить его клиенту через FastAPI?

Question

[gidwin]

Не совсем понимаю как правильно переписать код так, чтобы генерировался эксель на "лету" и отправлялся клиенту через фастапи. Пока что генерирую эксель сохраняя физически, мне это не нужно, вот мой код по генерации эксель используя библиотеку openpyxl:

def create_xl(append_data):
    one_list = 0
    wb = Workbook()
    with NamedTemporaryFile(delete=True) as tmp:
        tmp.close()
        wb2 = wb.active
        add_first_column(wb2)
        while one_list < len(append_data):
            for index in cells:
                
                number = 2
                for elem in append_data[one_list]:
                    wb2[f'{index}{number}'] = elem
                    number += 1
                one_list += 1
        wb.save(tmp.name)

    return tmp.name

А вот код, где отправляю клиенту этот файл через фастапи:

def historydatapnils(pnils = Form()):
    append_data_fio = history_pnils(pnils)

    strams = create_xl(append_data_fio)

    return FileResponse(strams, 
                        filename=f'{pnils}.xlsx', 
                        media_type="application/octet-stream")

Понимаю, что скорее всего надо использовать StreamingResponse при отправке сгенерированного на лету экселя, но вот вообще не получалось никак его таким образом сгенерировать, подскажите пожалуйста, как это можно реализовать?

Comments

[Алан Гибизов]

Сам не пробовал, но может посмотреть что-нибудь вроде этого:
https://stackoverflow.com/questions/67796261/pytho...

Answer 1

[Сергей Соловьев]

1. В ответе нужно указать Content-Disposition заголовок с названием файла
2. Возвращаешь StreamingResponse в который передаешь bytes

Т.е. тебе не нужно создавать временный файл (который ты ай-ай-ай не удаляешь после отправки)

StackOverflow с конкректным примером

Comments

[gidwin]

Спасибо за ответ, понимаю как отправить, но ключевая проблема, что я не понимаю как правильно создавать файл, ну то есть, каким образом мне его генерировать и как это можно сделать.

[Сергей Соловьев]

я дал ссылку, там пример

[gidwin]

Сергей Соловьев, там используется другая библиотека, к сожалению, условно под копирку я не могу написать, и к тому же, я находил этот пример и пытался переписать генерацию, но у меня ничего не выходило.. Да и вообще не понимаю как именно надо использовать BytesIO отсюда и все проблемы.

[Сергей Соловьев]

тогда хотя бы библиотеку указал, которую используешь

[gidwin]

Сергей Соловьев, я указал) но не в коде. Openpyxl использую

[Сергей Соловьев]

Тогда вот

def create_xl(append_data):
        one_list = 0
        wb = Workbook()
        wb2 = wb.active
        add_first_column(wb2)
        while one_list < len(append_data):
            for index in cells:
                
                number = 2
                for elem in append_data[one_list]:
                    wb2[f'{index}{number}'] = elem
                    number += 1
                one_list += 1
        result = BytesIO()
        wb.save(result)

        return result

[gidwin]

Сергей Соловьев, Спасибо, да и правда такой способ работает - скачивает, но пустой файл, можешь подсказать куда копать?

[Сергей Соловьев]

как в примере StackOverflow сделай не FileResponse, а StreamingResponse

[gidwin]

Сергей Соловьев, именно так я и сделал

[gidwin]

Сергей Соловьев, Если вдруг интересно, я разобрался. Достаточно было добавить к этому коду одну строчку в конце

def create_xl(append_data):
        one_list = 0
        wb = Workbook()
        wb2 = wb.active
        add_first_column(wb2)
        while one_list < len(append_data):
            for index in cells:
                
                number = 2
                for elem in append_data[one_list]:
                    wb2[f'{index}{number}'] = elem
                    number += 1
                one_list += 1
        result = BytesIO()
        wb.save(result)

        return result

Вот эту - result.seek(0)