Home > Enterprise >  Python: writing file and using buffer
Python: writing file and using buffer

Time:03-04

I'm using django to generate personalized file, but while doing so a file is generated, and in terms of space using it is quite a poor thing to do.

this is how i do it right now:

with open(filename, 'wb') as f:
    pdf.write(f) #pdf is an object of pyPDF2 library

with open(filename, 'rb') as f:
    return send_file(data=f, filename=filename) #send_file is a HTTPResponse parametted to download file data

So in the code above a file is generated.

The easy fix would be to deleted the file after downloading it, but i remember in java using stream object to handle this case.

Is it possible to do so in Python?

EDIT:

def send_file(data, filename, mimetype=None, force_download=False):
    disposition = 'attachment' if force_download else 'inline'
    filename = os.path.basename(filename)
    response = HttpResponse(data, content_type=mimetype or 'application/octet-stream')
    response['Content-Disposition'] = '%s; filename="%s"' % (disposition, filename)
    return response

CodePudding user response:

Without knowing the exact details of the pdf.write and send_file functions, I expect in both cases they will take an object that conforms to the BinaryIO interface. So, you could try using a BytesIO to store the content in an in-memory buffer, rather than writing out to a file:

with io.BytesIO() as buf:
    pdf.write(buf)
    buf.seek(0)
    send_file(data=buf, filename=filename)

Depending on the exact nature of the above-mentioned functions, YMMV.

  • Related