create Django object with FileField using `File` wrapper and in-memory-file?

Question

I am trying to build a functionality in my application where an In-memory PDF file will be created and that file will get saved on a FileField in my Document model.

Explaination of the process:


from io import BytesIO
from PyPDF2 import PdfWriter

in_memory_file = BytesIO()
writer = PdfWriter()


writer.add_blank_page(height=200, width=200)
writer.write(in_memory_file)

# The contents of the `writer` got saved to the `io` file

To save the above created file we can do the following

with open('new_local_file.pdf', 'wb') as local_file:
    local_file.write(in_memory_file.getbuffer())

The above code is working fine in terminal but since I cannot create local copies in my Django app, I have to do something like this to save the file

from django.core.files import File

obj = Document.objects.create(file_result=File(in_memory_file.getbuffer()))

The File wrapper accepts the python file object but still the above code is not working.

After execution no file is getting created in my Django Admin

Please comment if you need any more info.

Mukhtor Rasulov · Accepted Answer · 2022-07-28 18:35:55Z

1

If you want to save generated file to the model field you should save it in the media directory, And you can save the file by path:

from django.conf import settings


filename = "document.pdf"
directory = f"{settings.MEDIA_ROOT}/{filename}"
...
# in_memory_file generated here
...
with open(directory, 'wb') as local_file:
    local_file.write(in_memory_file.getbuffer())
obj = Document()
obj.file_result.name = filename
obj.save()

answered Jul 28, 2022 at 18:35

Mukhtor Rasulov

5873 silver badges10 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Vivek K. Singh Over a year ago

Thanks for the answer, but I wanted to ask if there is any other way where my memory-file converts directly to the FileField?

Collectives™ on Stack Overflow

create Django object with FileField using `File` wrapper and in-memory-file?

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related