Hopiu/django-avatar
1
0
Fork 0
mirror of https://github.com/jazzband/django-avatar.git synced 2026-03-16 22:20:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

check if image is not corrupted during upload (#218)

Browse source
This commit is contained in:
Petr Dlouhý 2023-02-22 12:39:28 -06:00 committed by GitHub
parent dfb2cb67e7
commit 43e052ebd5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
avatar/forms.py
View file

@ -5,6 +5,7 @@ from django.forms import widgets
from django.template.defaultfilters import filesizeformat from django.template.defaultfilters import filesizeformat
from django.utils.safestring import mark_safe from django.utils.safestring import mark_safe
from django.utils.translation import gettext_lazy as _ from django.utils.translation import gettext_lazy as _
from PIL import Image, ImageOps
from avatar.conf import settings from avatar.conf import settings
from avatar.models import Avatar from avatar.models import Avatar
@ -82,6 +83,12 @@ class UploadAvatarForm(forms.Form):
} }
) )
try:
image = Image.open(data)
ImageOps.exif_transpose(image)
except TypeError:
raise forms.ValidationError(_("Corrupted image"))
count = Avatar.objects.filter(user=self.user).count() count = Avatar.objects.filter(user=self.user).count()
if 1 < settings.AVATAR_MAX_AVATARS_PER_USER <= count: if 1 < settings.AVATAR_MAX_AVATARS_PER_USER <= count:
error = _( error = _(