Image resize fails when Froce flag is set

2012-08-18 17:27:22 +01:00
parent e990a6c70c
commit 676a7bf712
2 changed files with 27 additions and 3 deletions
--- a/mongoengine/fields.py
+++ b/mongoengine/fields.py
@@ -1108,6 +1108,7 @@ class ImageGridFsProxy(GridFSProxy):

        try:
            img = Image.open(file_obj)
+            img_format = img.format
        except:
            raise ValidationError('Invalid image')

@@ -1142,20 +1143,20 @@ class ImageGridFsProxy(GridFSProxy):

        if thumbnail:
            thumb_id = self._put_thumbnail(thumbnail,
-                                          img.format)
+                                          img_format)
        else:
            thumb_id = None

        w, h = img.size

        io = StringIO()
-        img.save(io, img.format)
+        img.save(io, img_format)
        io.seek(0)

        return super(ImageGridFsProxy, self).put(io,
                                                 width=w,
                                                 height=h,
-                                                 format=img.format,
+                                                 format=img_format,
                                                 thumbnail_id=thumb_id,
                                                 **kwargs)