Ticket #961: thumbnailing.diff

core/meta/__init__.py

@@ -757 +757 @@
                 func.alters_data = True
                 setattr(new_class, 'save_%s_file' % f.name, func)
                 if isinstance(f, ImageField):
+                    if f.thumb_sizes:
+                        setattr(new_class, 'get_%s_thumbnail_url' % f.name, curry(method_get_image_thumbnail_url, f))
+                        setattr(new_class, 'get_%s_thumbnail_sizes' % f.name, curry(method_get_image_thumbnail_sizes, f))
                     # Add get_BLAH_width and get_BLAH_height methods, but only
                     # if the image field doesn't have width and height cache
                     # fields.
@@ -1215 +1218 @@
 
     # Write the file to disk.
     setattr(self, field.attname, filename)
-    fp = open(getattr(self, 'get_%s_filename' % field.name)(), 'wb')
+    path = getattr(self, 'get_%s_filename' % field.name)()
+    fp = open(path, 'wb')
     fp.write(raw_contents)
     fp.close()
 
     # Save the width and/or height, if applicable.
-    if isinstance(field, ImageField) and (field.width_field or field.height_field):
-        from django.utils.images import get_image_dimensions
-        width, height = get_image_dimensions(getattr(self, 'get_%s_filename' % field.name)())
-        if field.width_field:
-            setattr(self, field.width_field, width)
-        if field.height_field:
-            setattr(self, field.height_field, height)
+    if isinstance(field, ImageField):
+        if field.thumb_sizes:
+            from django.parts.media.photos import get_thumbnail_path
+            from PIL import Image
+            orig = Image.open(path)
+            for size in field.thumb_sizes:
+                thumb_path = get_thumbnail_path(path, str(size))
+                thumb = orig.copy()
+                thumb.thumbnail((size,size), Image.ANTIALIAS)
+                thumb.save(thumb_path)
+        if field.width_field or field.height_field:
+            from django.utils.images import get_image_dimensions
+            width, height = get_image_dimensions(path)
+            if field.width_field:
+                setattr(self, field.width_field, width)
+            if field.height_field:
+                setattr(self, field.height_field, height)
 
     # Save the object, because it has changed.
     self.save()
 
 # IMAGE FIELD METHODS ######################
 
+def method_get_image_thumbnail_url(field, self, size=None):
+    from django.parts.media.photos import get_thumbnail_url
+    # If they don't specify a size, or the size doesn't exist,
+    # just give them back the first available thumbnail url.
+    if (size == None) or (size not in field.thumb_sizes):
+        size = field.thumb_sizes[0]
+    url = getattr(self, "get_%s_url" % field.name)()
+    return get_thumbnail_url(url, str(size))
+
+def method_get_image_thumbnail_sizes(field, self):
+    return field.thumb_sizes
+
 def method_get_image_width(field, self):
     return _get_image_dimensions(field, self)[0]
 

core/meta/fields.py

@@ -549 +549 @@
         return [curry(formfields.FloatField, max_digits=self.max_digits, decimal_places=self.decimal_places)]
 
 class ImageField(FileField):
-    def __init__(self, verbose_name=None, name=None, width_field=None, height_field=None, **kwargs):
-        self.width_field, self.height_field = width_field, height_field
+    def __init__(self, verbose_name=None, name=None, width_field=None, height_field=None, thumb_sizes=None, **kwargs):
+        self.width_field, self.height_field, self.thumb_sizes = width_field, height_field, thumb_sizes
         FileField.__init__(self, verbose_name, name, **kwargs)
 
     def get_manipulator_field_objs(self):

parts/media/photos.py

@@ -1 @@
-import re
+import re, os
 
+def get_thumbnail_path(photo_url, width, sep=os.path.sep):
+    bits = photo_url.split(sep)
+    bits[-1] = re.sub(r'(?i)\.(gif|jpg)$', '_t%s.\\1' % width, bits[-1])
+    return sep.join(bits)
+
 def get_thumbnail_url(photo_url, width):
-    bits = photo_url.split('/')
-    bits[-1] = re.sub(r'(?i)\.(gif|jpg)$', '_t%s.\\1' % width, bits[-1])
-    return '/'.join(bits)
+    return get_thumbnail_path(photo_url, width, '/')