def create_image(files, user): """Given an uploaded file, a user, and other data, it creates an Image""" up_file = files.values()[0] check_file_size(up_file, settings.IMAGE_MAX_FILESIZE) try: image = Image.objects.filter(creator=user, is_draft=True) # Delete other drafts, if any: image.exclude(pk=image[0].pk).delete() image = image[0] except IndexError: # No drafts, create one image = Image(creator=user, is_draft=True) # Async uploads fallback to these defaults. image.title = get_draft_title(user) image.description = u'Autosaved draft.' image.locale = settings.WIKI_DEFAULT_LANGUAGE up_file = _image_to_png(up_file) # Finally save the image along with uploading the file. image.file.save(os.path.splitext(up_file.name)[0] + '.png', File(up_file), save=True) (width, height) = _scale_dimensions(image.file.width, image.file.height) delete_url = reverse('gallery.delete_media', args=['image', image.id]) return {'name': up_file.name, 'url': image.get_absolute_url(), 'thumbnail_url': image.thumbnail_url_if_set(), 'width': width, 'height': height, 'delete_url': delete_url}
def create_image(files, user): """Given an uploaded file, a user, and other data, it creates an Image""" up_file = files.values()[0] check_file_size(up_file, settings.IMAGE_MAX_FILESIZE) try: image = Image.objects.filter(creator=user, is_draft=True) # Delete other drafts, if any: image.exclude(pk=image[0].pk).delete() image = image[0] except IndexError: # No drafts, create one image = Image(creator=user, is_draft=True) # Async uploads fallback to these defaults. image.title = get_draft_title(user) image.description = u'Autosaved draft.' image.locale = settings.WIKI_DEFAULT_LANGUAGE up_file = _image_to_png(up_file) # Finally save the image along with uploading the file. image.file.save(os.path.splitext(up_file.name)[0] + '.png', File(up_file), save=True) (width, height) = _scale_dimensions(image.file.width, image.file.height) delete_url = reverse('gallery.delete_media', args=['image', image.id]) return { 'name': up_file.name, 'url': image.get_absolute_url(), 'thumbnail_url': image.thumbnail_url_if_set(), 'width': width, 'height': height, 'delete_url': 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, 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_video(files, user): """Given an uploaded file, a user, and other data, it creates a Video""" # Async uploads fallback to these defaults. title = get_draft_title(user) description = u'Autosaved draft.' # Use default locale to make sure a user can only have one draft locale = settings.WIKI_DEFAULT_LANGUAGE try: vid = Video.objects.get(title=title, locale=locale) except Video.DoesNotExist: vid = Video(title=title, creator=user, description=description, locale=locale) for name in files: up_file = files[name] check_file_size(up_file, settings.VIDEO_MAX_FILESIZE) # name is in (webm, ogv, flv) sent from upload_video(), below getattr(vid, name).save(up_file.name, up_file, save=False) if 'thumbnail' == name: # Save poster too, we shrink it later. vid.poster.save(up_file.name, up_file, save=False) vid.save() if 'thumbnail' in files: thumb = vid.thumbnail (width, height) = _scale_dimensions(thumb.width, thumb.height) else: width = settings.THUMBNAIL_PROGRESS_WIDTH height = settings.THUMBNAIL_PROGRESS_HEIGHT delete_url = reverse('gallery.delete_media', args=['video', vid.id]) return {'name': up_file.name, 'url': vid.get_absolute_url(), 'thumbnail_url': vid.thumbnail_url_if_set(), 'width': width, 'height': height, 'delete_url': 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 create_image(files, user): """Given an uploaded file, a user, and other data, it creates an Image""" up_file = files.values()[0] check_file_size(up_file, settings.IMAGE_MAX_FILESIZE) # Async uploads fallback to these defaults. title = get_draft_title(user) description = u'Autosaved draft.' # Use default locale to make sure a user can only have one draft locale = settings.WIKI_DEFAULT_LANGUAGE image = Image(title=title, creator=user, locale=locale, description=description) image.file.save(up_file.name, File(up_file), save=True) (width, height) = _scale_dimensions(image.file.width, image.file.height) delete_url = reverse('gallery.delete_media', args=['image', image.id]) return {'name': up_file.name, 'url': image.get_absolute_url(), 'thumbnail_url': image.thumbnail_url_if_set(), 'width': width, 'height': height, 'delete_url': delete_url}
def create_video(files, user): """Given an uploaded file, a user, and other data, it creates a Video""" try: vid = Video.objects.filter(creator=user, is_draft=True) # Delete other drafts, if any: vid.exclude(pk=vid[0].pk).delete() vid = vid[0] except IndexError: # No drafts, create one vid = Video(creator=user, is_draft=True) # Async uploads fallback to these defaults. vid.title = get_draft_title(user) vid.description = u'Autosaved draft.' vid.locale = settings.WIKI_DEFAULT_LANGUAGE for name in files: up_file = files[name] check_file_size(up_file, settings.VIDEO_MAX_FILESIZE) # name is in (webm, ogv, flv) sent from upload_video(), below getattr(vid, name).save(up_file.name, up_file, save=False) if 'thumbnail' == name: # Save poster too, we shrink it later. vid.poster.save(up_file.name, up_file, save=False) vid.save() if 'thumbnail' in files: thumb = vid.thumbnail (width, height) = _scale_dimensions(thumb.width, thumb.height) else: width = settings.THUMBNAIL_PROGRESS_WIDTH height = settings.THUMBNAIL_PROGRESS_HEIGHT delete_url = reverse('gallery.delete_media', args=['video', vid.id]) return { 'name': up_file.name, 'url': vid.get_absolute_url(), 'thumbnail_url': vid.thumbnail_url_if_set(), 'width': width, 'height': height, 'delete_url': delete_url }
def create_video(files, user): """Given an uploaded file, a user, and other data, it creates a Video""" try: vid = Video.objects.filter(creator=user, is_draft=True) # Delete other drafts, if any: vid.exclude(pk=vid[0].pk).delete() vid = vid[0] except IndexError: # No drafts, create one vid = Video(creator=user, is_draft=True) # Async uploads fallback to these defaults. vid.title = get_draft_title(user) vid.description = u'Autosaved draft.' vid.locale = settings.WIKI_DEFAULT_LANGUAGE for name in files: up_file = files[name] check_file_size(up_file, settings.VIDEO_MAX_FILESIZE) # name is in (webm, ogv, flv) sent from upload_video(), below getattr(vid, name).save(up_file.name, up_file, save=False) if 'thumbnail' == name: # Save poster too, we shrink it later. vid.poster.save(up_file.name, up_file, save=False) vid.save() if 'thumbnail' in files: thumb = vid.thumbnail (width, height) = _scale_dimensions(thumb.width, thumb.height) else: width = settings.THUMBNAIL_PROGRESS_WIDTH height = settings.THUMBNAIL_PROGRESS_HEIGHT delete_url = reverse('gallery.delete_media', args=['video', vid.id]) return {'name': up_file.name, 'url': vid.get_absolute_url(), 'thumbnail_url': vid.thumbnail_url_if_set(), 'width': width, 'height': height, 'delete_url': delete_url}
def test_small(self): """A small image is not scaled.""" ts = settings.THUMBNAIL_SIZE / 2 (width, height) = _scale_dimensions(ts, ts) eq_(ts, width) eq_(ts, height)
def test_scale_dimensions_default(self): """A square image of exact size is not scaled.""" ts = settings.THUMBNAIL_SIZE (width, height) = _scale_dimensions(ts, ts, ts) eq_(ts, width) eq_(ts, height)
def test_large_both_width(self): """An image with both large is scaled to the largest - width.""" ts = 150 (width, height) = _scale_dimensions(ts * 20 + 8, ts * 4 + 36, ts) eq_(ts, width) eq_(31, height)
def test_large_both_height(self): """An image with both large is scaled to the largest - height.""" ts = 150 (width, height) = _scale_dimensions(ts * 2 + 13, ts * 5 + 30, ts) eq_(60, width) eq_(ts, height)
def test_large_height(self): """An image with large height is scaled to height=MAX.""" ts = 150 (width, height) = _scale_dimensions(ts - 2, ts * 2 + 9, ts) eq_(71, width) eq_(ts, height)
def test_basic(self): """A square image of exact size is not scaled.""" ts = settings.THUMBNAIL_SIZE (width, height) = _scale_dimensions(ts, ts, ts) eq_(ts, width) eq_(ts, height)
def test_width_large(self): """An image with large width is scaled to width=MAX.""" ts = 120 (width, height) = _scale_dimensions(ts * 3 + 10, ts - 1, ts) eq_(ts, width) eq_(38, height)