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

Use six.StringIO.

Browse source
This commit is contained in:
Jannis Leidel 2013-08-01 13:29:19 +02:00
parent 5d3a7dabd9
commit fc98737ce3
1 changed files with 2 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
avatar/models.py
View file

@ -13,11 +13,6 @@ from django.db.models import signals
from avatar.util import get_username
try:
from cStringIO import StringIO
except ImportError:
from StringIO import StringIO
try:
from PIL import Image
except ImportError:
@ -109,7 +104,7 @@ class Avatar(models.Model):
invalidate_cache(self.user, size)
try:
orig = self.avatar.storage.open(self.avatar.name, 'rb').read()
image = Image.open(StringIO(orig))
image = Image.open(six.StringIO(orig))
quality = quality or AVATAR_THUMB_QUALITY
(w, h) = image.size
if w != size or h != size:
@ -122,7 +117,7 @@ class Avatar(models.Model):
if image.mode != "RGB":
image = image.convert("RGB")
image = image.resize((size, size), AVATAR_RESIZE_METHOD)
thumb = StringIO()
thumb = six.StringIO()
image.save(thumb, AVATAR_THUMB_FORMAT, quality=quality)
thumb_file = ContentFile(thumb.getvalue())
else: