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, is_animated) = _image_to_png(up_file)
# Finally save the image along with uploading the file.
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_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, is_animated) = _image_to_png(up_file)
# Finally save the image along with uploading the file.
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 clean_avatar(self):
if not ("avatar" in self.cleaned_data and self.cleaned_data["avatar"]):
return self.cleaned_data["avatar"]
try:
check_file_size(self.cleaned_data["avatar"], settings.MAX_AVATAR_FILE_SIZE)
except FileTooLargeError as e:
raise forms.ValidationError(e.args[0])
return self.cleaned_data["avatar"]
def clean_avatar(self):
if not ('avatar' in self.cleaned_data and self.cleaned_data['avatar']):
return self.cleaned_data['avatar']
try:
check_file_size(self.cleaned_data['avatar'],
settings.MAX_AVATAR_FILE_SIZE)
except FileTooLargeError as e:
raise forms.ValidationError(e.args[0])
return self.cleaned_data['avatar']
def clean_avatar(self):
if not ("avatar" in self.cleaned_data and self.cleaned_data["avatar"]):
return self.cleaned_data["avatar"]
try:
check_file_size(self.cleaned_data["avatar"], settings.MAX_AVATAR_FILE_SIZE)
except FileTooLargeError as e:
raise forms.ValidationError(e.args[0])
clean_image_extension(self.cleaned_data.get("avatar"))
return self.cleaned_data["avatar"]
def clean_avatar(self):
if not ('avatar' in self.cleaned_data and self.cleaned_data['avatar']):
return self.cleaned_data['avatar']
try:
check_file_size(self.cleaned_data['avatar'],
settings.MAX_AVATAR_FILE_SIZE)
except FileTooLargeError as e:
raise forms.ValidationError(e.args[0])
clean_image_extension(self.cleaned_data.get('avatar'))
return self.cleaned_data['avatar']
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_check_file_size_over(self):
"""FileTooLargeError should be raised"""
with open('kitsune/upload/tests/media/test.jpg') as f:
up_file = File(f)
# This should raise
check_file_size(up_file, 0)
def test_check_file_size_under(self):
"""No exception should be raised"""
with open('kitsune/upload/tests/media/test.jpg') as f:
up_file = File(f)
check_file_size(up_file, settings.IMAGE_MAX_FILESIZE)
def test_check_file_size_over(self):
"""FileTooLargeError should be raised"""
with open('kitsune/upload/tests/media/test.jpg') as f:
up_file = File(f)
fn = lambda: check_file_size(up_file, 0)
self.assertRaises(FileTooLargeError, fn)
def test_check_file_size_under(self):
"""No exception should be raised"""
with open('kitsune/upload/tests/media/test.jpg') as f:
up_file = File(f)
check_file_size(up_file, settings.IMAGE_MAX_FILESIZE)
def test_check_file_size_over(self):
"""FileTooLargeError should be raised"""
with open('kitsune/upload/tests/media/test.jpg') as f:
up_file = File(f)
# This should raise
check_file_size(up_file, 0)
def test_check_file_size_over(self):
"""FileTooLargeError should be raised"""
with open('kitsune/upload/tests/media/test.jpg') as f:
up_file = File(f)
fn = lambda: check_file_size(up_file, 0)
self.assertRaises(FileTooLargeError, fn)
