Saving a matplotlib graph as an image field in database

Question

I just started learning matplotlib and I want to use it in one of my django apps. So I wanted to know how I can save the graph generated in an image field of my models, so that I can retrive whenever needed.

falsetru · Accepted Answer · 2013-12-14 06:42:40Z

4

matplotlib.pyplot.savefig accepts file-like object as the first parameter. You can pass StringIO/BytesIO (according to your python version).

f = StringIO()
plt.savefig(f)

Then, use django.core.files.ContentFile to convert the string to django.core.files.File (because FieldFile.save accepts only accept an instance of django.core.files.File).

content_file = ContentFile(f.getvalue())
model_object = Model(....)
model_object.image_field.save('name_of_image', content_file)
model_object.save()

edited Dec 14, 2013 at 6:42

answered Dec 14, 2013 at 6:06

falsetru

371k69 gold badges770 silver badges660 bronze badges

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

10 Comments

Aswin Murugesh Over a year ago

Am using python 2.7 and I get a not defined error for both StringIO and BytesIO

falsetru Over a year ago

@AswinMurugesh, Import it. from io import BytesIO / from StringIO import StringIO / from cStringIO import StringIO.

falsetru Over a year ago

@AswinMurugesh, docs.python.org/2/library/stringio.html / docs.python.org/3/library/io.html#io.BytesIO

falsetru Over a year ago

@AswinMurugesh, Please read Managing files | Django documentation

Chris Over a year ago

Here it says: For Python2.6 or better, use io.BytesIO instead of cStringIO.StringIO for forward-compatibility. In Python3, the cStringIO, StringIO modules are gone. Their functionality is all in the io module.

|

Collectives™ on Stack Overflow

Saving a matplotlib graph as an image field in database

1 Answer 1

10 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

10 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related