django-imagekit/imagekit/models/fields/utils.py

from ...cachefiles import ImageCacheFile
from django.db.models.fields.files import ImageField


class ImageSpecFileDescriptor(object):
    def __init__(self, field, attname):
        self.attname = attname
        self.field = field

    def __get__(self, instance, owner):
        if instance is None:
            return self.field
        else:
            field_name = getattr(self.field, 'source', None)
            if field_name:
                source = getattr(instance, field_name)
            else:
                image_fields = [getattr(instance, f.attname) for f in
                        instance.__class__._meta.fields if
                        isinstance(f, ImageField)]
                if len(image_fields) == 0:
                    raise Exception('%s does not define any ImageFields, so your'
                            ' %s ImageSpecField has no image to act on.' %
                            (instance.__class__.__name__, self.attname))
                elif len(image_fields) > 1:
                    raise Exception('%s defines multiple ImageFields, but you'
                            ' have not specified a source for your %s'
                            ' ImageSpecField.' % (instance.__class__.__name__,
                            self.attname))
                else:
                    source = image_fields[0]
            spec = self.field.get_spec(source=source)
            file = ImageCacheFile(spec)
            instance.__dict__[self.attname] = file
            return file

    def __set__(self, instance, value):
        instance.__dict__[self.attname] = value