mirror of
https://github.com/Hopiu/django-imagekit.git
synced 2026-03-17 05:40:25 +00:00
38 lines
1.6 KiB
Python
38 lines
1.6 KiB
Python
from ...files import GeneratedImageFile
|
|
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 = GeneratedImageFile(spec)
|
|
instance.__dict__[self.attname] = file
|
|
return file
|
|
|
|
def __set__(self, instance, value):
|
|
instance.__dict__[self.attname] = value
|