def create_imageattachment(files, user, obj):
"""
Given an uploaded file, a user and an object, it creates an ImageAttachment
owned by `user` and attached to `obj`.
"""
up_file = files.values()[0]
check_file_size(up_file, settings.IMAGE_MAX_FILESIZE)
(up_file, is_animated) = _image_to_png(up_file)
image = ImageAttachment(content_object=obj, creator=user)
image.file.save(up_file.name, File(up_file), save=True)
# Compress and generate thumbnail off thread
generate_thumbnail.delay(image, 'file', 'thumbnail')
if not is_animated:
compress_image.delay(image, 'file')
(width, height) = _scale_dimensions(image.file.width, image.file.height)
return {
'name': up_file.name,
'url': image.file.url,
'thumbnail_url': image.thumbnail_if_set().url,
'width': width,
'height': height,
'delete_url': image.get_delete_url()
}
def create_imageattachment(files, user, obj):
"""
Given an uploaded file, a user and an object, it creates an ImageAttachment
owned by `user` and attached to `obj`.
"""
up_file = files.values()[0]
check_file_size(up_file, settings.IMAGE_MAX_FILESIZE)
up_file = _image_to_png(up_file)
image = ImageAttachment(content_object=obj, creator=user)
image.file.save(os.path.splitext(up_file.name)[0] + '.png',
File(up_file),
save=True)
# Compress and generate thumbnail off thread
generate_thumbnail.delay(image, 'file', 'thumbnail')
compress_image.delay(image, 'file')
(width, height) = _scale_dimensions(image.file.width, image.file.height)
return {
'name': up_file.name,
'url': image.file.url,
'thumbnail_url': image.thumbnail_if_set().url,
'width': width,
'height': height,
'delete_url': image.get_delete_url()
}
def test_thumbnail_if_set(self):
"""thumbnail_if_set() returns self.thumbnail if set, or else returns
self.file"""
image = ImageAttachment(content_object=self.obj, creator=self.user)
with open('apps/upload/tests/media/test.jpg') as f:
up_file = File(f)
image.file.save(up_file.name, up_file, save=True)
eq_(image.file, image.thumbnail_if_set())
generate_thumbnail(image, 'file', 'thumbnail')
eq_(image.thumbnail, image.thumbnail_if_set())
def test_thumbnail_if_set(self):
"""thumbnail_if_set() returns self.thumbnail if set, or else returns
self.file"""
image = ImageAttachment(content_object=self.obj, creator=self.user)
with open('apps/upload/tests/media/test.jpg') as f:
up_file = File(f)
image.file.save(up_file.name, up_file, save=True)
eq_(image.file, image.thumbnail_if_set())
generate_thumbnail(image, 'file', 'thumbnail')
eq_(image.thumbnail, image.thumbnail_if_set())
def _image_with_thumbnail(self):
image = ImageAttachment(content_object=self.obj, creator=self.user)
with open('apps/upload/tests/media/test.jpg') as f:
up_file = File(f)
image.file.save(up_file.name, up_file, save=True)
generate_thumbnail(image, 'file', 'thumbnail')
return image
def test_generate_deleted_file(self):
"""generate_thumbnail does not fail if file doesn't actually exist."""
image = ImageAttachment(content_object=self.obj, creator=self.user)
with open('apps/upload/tests/media/test.jpg') as f:
up_file = File(f)
image.file.save(up_file.name, up_file, save=True)
# The field will be set but the file isn't there.
os.remove(image.file.path)
generate_thumbnail(image, 'file', 'thumbnail')
def test_generate_no_file(self):
"""generate_thumbnail does not fail when no file is provided."""
image = ImageAttachment(content_object=self.obj, creator=self.user)
generate_thumbnail(image, 'file', 'thumbnail')
def test_compress_no_file(self, call):
"""compress_image does not fail when no file is provided."""
image = ImageAttachment(content_object=self.obj, creator=self.user)
compress_image(image, 'file')
assert not call.called
def _uploaded_image(self):
image = ImageAttachment(content_object=self.obj, creator=self.user)
with open('apps/upload/tests/media/test.jpg') as f:
up_file = File(f)
image.file.save(up_file.name, up_file, save=True)
return image
