add covers

This commit is contained in:
j 2011-02-23 10:52:30 +01:00
parent 8b45dd8460
commit c2586bb61d
5 changed files with 130 additions and 0 deletions

0
oxdata/cover/__init__.py Normal file
View file

10
oxdata/cover/admin.py Normal file
View file

@ -0,0 +1,10 @@
# -*- coding: utf-8 -*-
# vi:si:et:sw=4:sts=4:ts=4
import models
from django.contrib import admin
class CoverAdmin(admin.ModelAdmin):
search_fields = ['url', 'isbn']
admin.site.register(models.Cover, CoverAdmin)

96
oxdata/cover/models.py Normal file
View file

@ -0,0 +1,96 @@
# -*- coding: utf-8 -*-
# vi:si:et:sw=4:sts=4:ts=4
from __future__ import division
import os.path
import hashlib
from django.conf import settings
from django.db import models
from django.db.models import Q
from django.contrib.auth.models import User
from django.core.files.base import ContentFile
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)
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=lambda i, f: cover_path(i.url.encode('utf-8'), f))
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.readUrl(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
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=lambda i, f: cover_path('upload/%s' % i.id, f))

6
oxdata/cover/urls.py Normal file
View file

@ -0,0 +1,6 @@
from django.conf.urls.defaults import *
urlpatterns = patterns('oxdata.cover.views',
(r'^$', 'cover'),
)

18
oxdata/cover/views.py Normal file
View file

@ -0,0 +1,18 @@
# -*- coding: utf-8 -*-
# vi:si:et:sw=4:sts=4:ts=4
import os.path
from django.db import models
from django.db.models import Q
from django.contrib.auth.models import User
from django.shortcuts import render_to_response, get_object_or_404, get_list_or_404
from django.template import RequestContext
from oxdjango.shortcuts import render_to_json_response
import models
def cover(request):
isbn = request.GET['isbn']
json = models.getCovers(isbn, request.build_absolute_uri('/'))
return render_to_json_response(json)