121 lines
4.1 KiB
Python
121 lines
4.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
# vi:si:et:sw=4:sts=4:ts=4
|
|
import os.path
|
|
import hashlib
|
|
|
|
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
|
|
|
|
import oxweb.criterion
|
|
import oxweb.movieposterdb
|
|
import oxweb.karagarga
|
|
import oxweb.imdb
|
|
import oxweb.impawards
|
|
|
|
from oxdata.lookup.models import MovieId, Karagarga
|
|
|
|
def getPosters(movie_id):
|
|
getPosterUrls(movie_id)
|
|
posters = {}
|
|
for p in PosterCache.objects.all().filter(movie_id=movie_id):
|
|
if p.site not in posters:
|
|
posters[p.site] = []
|
|
poster = p.get()
|
|
if poster:
|
|
pjson = {}
|
|
pjson['url'] = poster._get_url()
|
|
pjson['width'] = poster.width
|
|
pjson['height'] = poster.height
|
|
posters[p.site].append(pjson)
|
|
for p in posters:
|
|
if not posters[p]:
|
|
del posters[p]
|
|
return posters
|
|
|
|
def poster_path(url, filename):
|
|
url_hash = hashlib.sha1(url).hexdigest()
|
|
ext = 'jpg'
|
|
if filename.endswith('.png'):
|
|
ext = 'png'
|
|
name = "%s.%s" % (url_hash, ext)
|
|
return os.path.join('posters', url_hash[:2], url_hash[2:4], url_hash[4:6], name)
|
|
|
|
class PosterCache(models.Model):
|
|
created = models.DateTimeField(auto_now_add=True)
|
|
modified = models.DateTimeField(auto_now=True)
|
|
|
|
movie_id = models.ForeignKey(MovieId, related_name='postercache')
|
|
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: poster_path(i.url, f))
|
|
failed = models.BooleanField(default=False)
|
|
|
|
def get(self):
|
|
if not self.image and not self.failed:
|
|
try:
|
|
import oxlib.net
|
|
name = hashlib.sha1(self.url).hexdigest()
|
|
data = oxlib.net.readUrl(self.url)
|
|
self.image.save(name, ContentFile(data))
|
|
except:
|
|
self.failed = True
|
|
self.save()
|
|
return self.image
|
|
|
|
|
|
class Poster(models.Model):
|
|
created = models.DateTimeField(auto_now_add=True)
|
|
modified = models.DateTimeField(auto_now=True)
|
|
|
|
movie_id = models.ForeignKey(MovieId, related_name='poster')
|
|
poster = models.ForeignKey(PosterCache, default=None, blank=True)
|
|
upload = models.ImageField(max_length=255, upload_to=lambda i, f: poster_path('upload/%s' % i.id, f))
|
|
oxdb = models.ImageField(max_length=255, upload_to=lambda i, f: poster_path('oxdb/%s' % i.id, f))
|
|
|
|
def get(self):
|
|
if self.image:
|
|
return self.image
|
|
if self.poster:
|
|
return self.poster.get()
|
|
#FIXME: select poster
|
|
return None
|
|
|
|
def getPosterUrls(m):
|
|
def addPoster(url, site, site_id):
|
|
if PosterCache.objects.all().filter(url=url).count() == 0:
|
|
p = PosterCache(url=url, site=site, site_id=site_id, movie_id=m)
|
|
p.save()
|
|
|
|
if m.imdb_id:
|
|
poster = oxweb.imdb.getMoviePoster(m.imdb_id)
|
|
if poster:
|
|
addPoster(poster, 'imdb.com', m.imdb_id)
|
|
for poster in oxweb.movieposterdb.getData(m.imdb_id)['posters']:
|
|
addPoster(poster, 'movieposterdb.com', m.imdb_id)
|
|
if m.criterion_id:
|
|
print 'criterion', m.criterion_id
|
|
for poster in oxweb.criterion.getData(m.criterion_id)['posters']:
|
|
addPoster(poster, 'criterion.com', m.criterion_id)
|
|
if m.wikipedia_id:
|
|
poster = oxweb.wikipedia.getPosterUrl(m.wikipedia_id)
|
|
if poster:
|
|
if PosterCache.objects.all().filter(url=poster).count() == 0:
|
|
addPoster(poster, 'wikipedia.org', m.wikipedia_id)
|
|
if m.impawards_id:
|
|
data = oxweb.impawards.getData(m.impawards_id)
|
|
if data and 'imdbId' in data:
|
|
for poster in data['posters']:
|
|
addPoster(poster, 'impawards.com', m.imdb_id)
|
|
|
|
for kg in Karagarga.objects.all().filter(movie_id=m):
|
|
data = oxweb.karagarga.getData(kg.karagarga_id)
|
|
if data:
|
|
for poster in data['posters']:
|
|
addPoster(poster, 'karagarga.net', kg.karagarga_id)
|
|
else:
|
|
kg.delete()
|
|
|