Removing the dependency of django-thumbs.

This commit is contained in:
Isaac Bythewood
2012-07-25 01:50:28 +00:00
parent 8389462c80
commit a178144da4
12 changed files with 83 additions and 229 deletions

View File

@@ -3,29 +3,54 @@ from django.core.files import File
from django.core.files.temp import NamedTemporaryFile
from django.contrib.auth.models import User
from thumbs import ImageWithThumbsField
import urllib2
import hashlib
from PIL import Image
class Pin(models.Model):
submitter = models.ForeignKey(User)
url = models.TextField(blank=True, null=True)
description = models.TextField(blank=True, null=True)
image = ImageWithThumbsField(upload_to='pins/pin', sizes=((200, 1000),))
image = models.ImageField(upload_to='pins/pin/originals/')
thumbnail = models.ImageField(upload_to='pins/pin/thumbnails/')
published = models.DateTimeField(auto_now_add=True)
def __unicode__(self):
return self.url
def save(self, *args, **kwargs):
if not self.image:
temp_img = NamedTemporaryFile()
temp_img.write(urllib2.urlopen(self.url).read())
temp_img.flush()
# pylint: disable-msg=E1101
self.image.save(self.url.split('/')[-1], File(temp_img))
else:
super(Pin, self).save()
if not self.thumbnail:
if not self.image:
image = Image.open(temp_img.name)
else:
image = Image.open(self.image.path)
size = image.size
prop = 200 / image.size[0]
size = (prop*image.size[0], prop*image.size[1])
image.resize(size, Image.ANTIALIAS)
temp_thumb = NamedTemporaryFile()
image.save(temp_thumb.name, 'JPEG')
if self.url:
name = self.url.split('/')[-1]
else:
name = self.image.name
self.thumbnail.save(name, File(temp_thumb))
super(Pin, self).save(*args, **kwargs)
class Meta:
ordering = ['-id']