Resizing badge to 50x50 when saving it

Luan
1 parent e56c1968
Showing 1 changed file with 8 additions and 0 deletions Show diff stats
src/badger/models.py
@@ -4,6 +4,7 @@ from django.conf import settings
 from django.contrib.auth import get_user_model
 from django.db import models
 from django.utils.translation import ugettext as _
+from PIL import Image
  
  
 class Badge(models.Model):
@@ -67,6 +68,13 @@ class Badge(models.Model):
     def get_badge_url(self):
         return u'{}{}'.format(settings.MEDIA_URL, self.image)
  
+    def save(self, *args, **kwargs):
+        img = Image.open(self.image)
+        (width, height) = img.size
+        img = img.resize((50, 50), Image.ANTIALIAS)
+        super(Badge, self).save(*args, **kwargs)
+        img.save(self.image.path)
+
     def __unicode__(self):
         return u'{} ({}, {})'.format(
             self.title,
...	...	@@ -4,6 +4,7 @@ from django.conf import settings
4	4	from django.contrib.auth import get_user_model
5	5	from django.db import models
6	6	from django.utils.translation import ugettext as _
	7	+from PIL import Image
7	8
8	9
9	10	class Badge(models.Model):
...	...	@@ -67,6 +68,13 @@ class Badge(models.Model):
67	68	def get_badge_url(self):
68	69	return u'{}{}'.format(settings.MEDIA_URL, self.image)
69	70
	71	+ def save(self, args, *kwargs):
	72	+ img = Image.open(self.image)
	73	+ (width, height) = img.size
	74	+ img = img.resize((50, 50), Image.ANTIALIAS)
	75	+ super(Badge, self).save(args, *kwargs)
	76	+ img.save(self.image.path)
	77	+
70	78	def __unicode__(self):
71	79	return u'{} ({}, {})'.format(
72	80	self.title,
...	...