102 lines
3.3 KiB
Python
102 lines
3.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
# vi:si:et:sw=4:sts=4:ts=4
|
|
from __future__ import division
|
|
|
|
import os.path
|
|
import hashlib
|
|
|
|
from django.db import models
|
|
from django.core.files.base import ContentFile
|
|
|
|
from lookup.models import MovieId
|
|
|
|
def getCovers(isbn, url_prefix='', limit=lambda x, y: 0.3 < x/y < 1):
|
|
covers = {}
|
|
if url_prefix.endswith('/'): url_prefix = url_prefix[:-1]
|
|
for p in CoverCache.objects.all().filter(isbn=isbn, failed=False).order_by('id'):
|
|
if p.site not in covers:
|
|
covers[p.site] = []
|
|
cover = p.get()
|
|
if cover:
|
|
pjson = {}
|
|
pjson['url'] = url_prefix + cover._get_url()
|
|
pjson['width'] = cover.width
|
|
pjson['height'] = cover.height
|
|
if p.site not in ['other', 'wikipedia.org'] or limit(cover.width, cover.height):
|
|
covers[p.site].append(pjson)
|
|
for p in covers.keys():
|
|
if not covers[p]:
|
|
del covers[p]
|
|
covers = Cover.objects.filter(isbn=isbn).exclude(cover='')
|
|
if cover.count() > 0:
|
|
covers['local'] = [p.cover.url]
|
|
return covers
|
|
|
|
def cover_path(url, filename):
|
|
h = hashlib.sha1(url).hexdigest()
|
|
ext = 'jpg'
|
|
if filename.endswith('.png'):
|
|
ext = 'png'
|
|
name = "%s.%s" % (h, ext)
|
|
return os.path.join('covers', h[:2], h[2:4], h[4:6], name)
|
|
|
|
def covercache_image_path(i, f):
|
|
return cover_path(i.url.encode('utf-8'), f)
|
|
|
|
class CoverCache(models.Model):
|
|
class Meta:
|
|
unique_together = ("isbn", "url")
|
|
|
|
created = models.DateTimeField(auto_now_add=True)
|
|
modified = models.DateTimeField(auto_now=True)
|
|
|
|
isbn = models.CharField(max_length=13)
|
|
url = models.CharField(max_length=1024)
|
|
site = models.CharField(max_length=255)
|
|
site_id = models.CharField(max_length=42)
|
|
image = models.ImageField(max_length=255, upload_to=covercache_image_path)
|
|
status = models.CharField(max_length=1024, default='200')
|
|
failed = models.BooleanField(default=False)
|
|
|
|
def get(self):
|
|
if not self.image and not self.failed:
|
|
import ox.net
|
|
url = self.url.encode('utf-8')
|
|
name = hashlib.sha1(url).hexdigest()
|
|
try:
|
|
data = ox.net.read_url(url)
|
|
self.image.save(name, ContentFile(data))
|
|
except ox.net.urllib2.HTTPError, e:
|
|
#import traceback
|
|
#print traceback.print_exc()
|
|
self.status = e.code
|
|
self.failed = True
|
|
self.save()
|
|
except ox.net.urllib2.URLError, e:
|
|
#import traceback
|
|
#print traceback.print_exc()
|
|
self.status = e.reason
|
|
self.failed = True
|
|
self.save()
|
|
if self.image:
|
|
try:
|
|
self.image.width
|
|
except IOError:
|
|
self.image.delete()
|
|
except TypeError:
|
|
self.failed = True
|
|
self.status = "invalid image"
|
|
self.image.delete()
|
|
return self.image
|
|
|
|
def cover_image_path(i, f):
|
|
return cover_path('upload/%s' % i.id, f)
|
|
|
|
class Cover(models.Model):
|
|
created = models.DateTimeField(auto_now_add=True)
|
|
modified = models.DateTimeField(auto_now=True)
|
|
|
|
isbn = models.ForeignKey(MovieId, related_name='cover')
|
|
cover = models.ImageField(max_length=255, upload_to=cover_image_path)
|
|
|