2012-10-16 03:53:05 +00:00
|
|
|
from django.db.models.fields.files import ImageFieldFile
|
2012-10-17 03:38:44 +00:00
|
|
|
import os
|
2013-01-24 03:37:00 +00:00
|
|
|
from ...utils import suggest_extension, generate
|
2012-07-19 02:24:39 +00:00
|
|
|
|
2012-02-14 02:44:29 +00:00
|
|
|
|
|
|
|
|
class ProcessedImageFieldFile(ImageFieldFile):
|
|
|
|
|
def save(self, name, content, save=True):
|
2012-10-17 03:38:44 +00:00
|
|
|
filename, ext = os.path.splitext(name)
|
2013-01-24 03:37:00 +00:00
|
|
|
spec = self.field.get_spec(source=content)
|
2012-10-21 03:21:01 +00:00
|
|
|
ext = suggest_extension(name, spec.format)
|
2012-10-17 03:38:44 +00:00
|
|
|
new_name = '%s%s' % (filename, ext)
|
2013-01-24 03:37:00 +00:00
|
|
|
content = generate(spec)
|
2012-10-17 03:38:44 +00:00
|
|
|
return super(ProcessedImageFieldFile, self).save(new_name, content, save)
|