Hopiu/django-imagekit
1
0
Fork 0
mirror of https://github.com/Hopiu/django-imagekit.git synced 2026-03-17 05:40:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Special case serialization of ImageFieldFiles

Browse source 
Closes #168
This commit is contained in:
Matthew Tretter 2013-01-08 21:52:56 -05:00
parent c2dedaa2b8
commit 658bb22c78
1 changed files with 20 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
imagekit/specs/__init__.py
View file

@ -1,4 +1,5 @@
from django.conf import settings
from django.db.models.fields.files import ImageFieldFile
from hashlib import md5
import os
import pickle
@ -102,6 +103,25 @@ class ImageSpec(BaseImageSpec):
return os.path.join(settings.IMAGEKIT_CACHE_DIR,
'%s%s' % (hash, ext))
def __getstate__(self):
state = self.__dict__
# Unpickled ImageFieldFiles won't work (they're missing a storage
# object). Since they're such a common use case, we special case them.
if isinstance(self.source, ImageFieldFile):
field = getattr(self.source, 'field')
state['_field_data'] = {
'instance': getattr(self.source, 'instance', None),
'attname': getattr(field, 'name', None),
}
return state
def __setstate__(self, state):
field_data = state.pop('_field_data', None)
self.__dict__ = state
if field_data:
self.source = getattr(field_data['instance'], field_data['attname'])
def get_hash(self):
return md5(pickle.dumps([
self.source.name,