dockeruser
/
CouchPotatoServer
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

Simplify tmdb provider

pull/4054/head
Ruud 11 years ago
parent
38a5d967dd
commit
06293dc0a2
1 changed files with 108 additions and 121 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 229
      couchpotato/core/media/movie/providers/info/themoviedb.py

229
couchpotato/core/media/movie/providers/info/themoviedb.py
View File

@ -1,8 +1,7 @@
import traceback import traceback
import time
from couchpotato.core.event import addEvent, fireEvent from couchpotato.core.event import addEvent, fireEvent
from couchpotato.core.helpers.encoding import simplifyString, toUnicode, ss, tryUrlencode from couchpotato.core.helpers.encoding import toUnicode, ss, tryUrlencode
from couchpotato.core.helpers.variable import tryInt from couchpotato.core.helpers.variable import tryInt
from couchpotato.core.logger import CPLog from couchpotato.core.logger import CPLog
from couchpotato.core.media.movie.providers.base import MovieProvider from couchpotato.core.media.movie.providers.base import MovieProvider
@ -14,7 +13,7 @@ autoload = 'TheMovieDb'
class TheMovieDb(MovieProvider): class TheMovieDb(MovieProvider):
http_time_between_calls = .3 http_time_between_calls = .35
configuration = { configuration = {
'images': { 'images': {
@ -23,6 +22,8 @@ class TheMovieDb(MovieProvider):
} }
def __init__(self): def __init__(self):
addEvent('info.search', self.search, priority = 3)
addEvent('movie.search', self.search, priority = 3)
addEvent('movie.info', self.getInfo, priority = 3) addEvent('movie.info', self.getInfo, priority = 3)
addEvent('movie.info_by_tmdb', self.getInfo) addEvent('movie.info_by_tmdb', self.getInfo)
addEvent('app.load', self.config) addEvent('app.load', self.config)
@ -32,49 +33,44 @@ class TheMovieDb(MovieProvider):
if configuration: if configuration:
self.configuration = configuration self.configuration = configuration
def search(self, q, limit = 12): def search(self, q, limit = 3):
""" Find movie by name """ """ Find movie by name """
if self.isDisabled(): if self.isDisabled():
return False return False
search_string = simplifyString(q) log.debug('Searching for movie: %s', q)
cache_key = 'tmdb.cache.%s.%s' % (search_string, limit)
results = None #self.getCache(cache_key)
if not results: raw = None
log.debug('Searching for movie: %s', q) try:
name_year = fireEvent('scanner.name_year', q, single = True)
raw = self.request('search/movie', {
'query': name_year.get('name', q),
'year': name_year.get('year'),
'search_type': 'ngram' if limit > 1 else 'phrase'
}, return_key = 'results')
except:
log.error('Failed searching TMDB for "%s": %s', (q, traceback.format_exc()))
raw = None results = []
if raw:
try: try:
nr = 0
#name_year = fireEvent('scanner.name_year', q, single = True) for movie in raw:
parsed_movie = self.parseMovie(movie, extended = False)
raw = self.request('search/movie', { results.append(parsed_movie)
'query': q
}, return_key = 'results')
except:
log.error('Failed searching TMDB for "%s": %s', (search_string, traceback.format_exc()))
results = []
if raw:
try:
nr = 0
for movie in raw: nr += 1
results.append(self.parseMovie(movie, extended = False)) if nr == limit:
break
nr += 1 log.info('Found: %s', [result['titles'][0] + ' (' + str(result.get('year', 0)) + ')' for result in results])
if nr == limit:
break
log.info('Found: %s', [result['titles'][0] + ' (' + str(result.get('year', 0)) + ')' for result in results]) return results
except SyntaxError as e:
self.setCache(cache_key, results) log.error('Failed to parse XML response: %s', e)
return results return False
except SyntaxError as e:
log.error('Failed to parse XML response: %s', e)
return False
return results return results
@ -91,89 +87,81 @@ class TheMovieDb(MovieProvider):
def parseMovie(self, movie, extended = True): def parseMovie(self, movie, extended = True):
cache_key = 'tmdb.cache.%s%s' % (movie.get('id'), '.ex' if extended else '') # Do request, append other items
movie_data = None #self.getCache(cache_key) movie = self.request('movie/%s' % movie.get('id'), {
'append_to_response': 'alternative_titles' + (',images,casts' if extended else '')
})
# Images
poster = self.getImage(movie, type = 'poster', size = 'w154')
poster_original = self.getImage(movie, type = 'poster', size = 'original')
backdrop_original = self.getImage(movie, type = 'backdrop', size = 'original')
extra_thumbs = self.getMultImages(movie, type = 'backdrops', size = 'original') if extended else []
images = {
'poster': [poster] if poster else [],
#'backdrop': [backdrop] if backdrop else [],
'poster_original': [poster_original] if poster_original else [],
'backdrop_original': [backdrop_original] if backdrop_original else [],
'actors': {},
'extra_thumbs': extra_thumbs
}
# Genres
try:
genres = [genre.get('name') for genre in movie.get('genres', [])]
except:
genres = []
# 1900 is the same as None
year = str(movie.get('release_date') or '')[:4]
if not movie.get('release_date') or year == '1900' or year.lower() == 'none':
year = None
if not movie_data: # Gather actors data
actors = {}
if extended:
# Full data # Full data
movie = self.request('movie/%s' % movie.get('id')) cast = movie.get('casts', {}).get('cast', [])
# Images for cast_item in cast:
poster = self.getImage(movie, type = 'poster', size = 'w154') try:
poster_original = self.getImage(movie, type = 'poster', size = 'original') actors[toUnicode(cast_item.get('name'))] = toUnicode(cast_item.get('character'))
backdrop_original = self.getImage(movie, type = 'backdrop', size = 'original') images['actors'][toUnicode(cast_item.get('name'))] = self.getImage(cast_item, type = 'profile', size = 'original')
extra_thumbs = self.getMultImages(movie, type = 'backdrops', size = 'original') except:
log.debug('Error getting cast info for %s: %s', (cast_item, traceback.format_exc()))
images = {
'poster': [poster] if poster else [], movie_data = {
#'backdrop': [backdrop] if backdrop else [], 'type': 'movie',
'poster_original': [poster_original] if poster_original else [], 'via_tmdb': True,
'backdrop_original': [backdrop_original] if backdrop_original else [], 'tmdb_id': movie.get('id'),
'actors': {}, 'titles': [toUnicode(movie.get('title'))],
'extra_thumbs': extra_thumbs 'original_title': movie.get('original_title'),
} 'images': images,
'imdb': movie.get('imdb_id'),
# Genres 'runtime': movie.get('runtime'),
try: 'released': str(movie.get('release_date')),
genres = [genre.get('name') for genre in movie.get('genres', [])] 'year': tryInt(year, None),
except: 'plot': movie.get('overview'),
genres = [] 'genres': genres,
'collection': getattr(movie.get('belongs_to_collection'), 'name', None),
# 1900 is the same as None 'actor_roles': actors
year = str(movie.get('release_date') or '')[:4] }
if not movie.get('release_date') or year == '1900' or year.lower() == 'none':
year = None movie_data = dict((k, v) for k, v in movie_data.items() if v)
# Gather actors data # Add alternative names
actors = {} if movie_data['original_title'] and movie_data['original_title'] not in movie_data['titles']:
if extended: movie_data['titles'].append(movie_data['original_title'])
# Full data # Add alternative titles
cast = self.request('movie/%s/casts' % movie.get('id'), return_key = 'cast') alternate_titles = movie.get('alternative_titles', {}).get('titles', [])
for cast_item in cast: for alt in alternate_titles:
try: alt_name = alt.get('title')
actors[toUnicode(cast_item.get('name'))] = toUnicode(cast_item.get('character')) if alt_name and alt_name not in movie_data['titles'] and alt_name.lower() != 'none' and alt_name is not None:
images['actors'][toUnicode(cast_item.get('name'))] = self.getImage(cast_item, type = 'profile', size = 'original') movie_data['titles'].append(alt_name)
except:
log.debug('Error getting cast info for %s: %s', (cast_item, traceback.format_exc()))
movie_data = {
'type': 'movie',
'via_tmdb': True,
'tmdb_id': movie.get('id'),
'titles': [toUnicode(movie.get('title'))],
'original_title': movie.get('original_title'),
'images': images,
'imdb': movie.get('imdb_id'),
'runtime': movie.get('runtime'),
'released': str(movie.get('release_date')),
'year': tryInt(year, None),
'plot': movie.get('overview'),
'genres': genres,
'collection': getattr(movie.get('belongs_to_collection'), 'name', None),
'actor_roles': actors
}
movie_data = dict((k, v) for k, v in movie_data.items() if v)
# Add alternative names
if movie_data['original_title'] and movie_data['original_title'] not in movie_data['titles']:
movie_data['titles'].append(movie_data['original_title'])
if extended:
# Full data
alternate_titles = self.request('movie/%s/alternative_titles' % movie.get('id'), return_key = 'titles')
for alt in alternate_titles:
alt_name = alt.get('title')
if alt_name and alt_name not in movie_data['titles'] and alt_name.lower() != 'none' and alt_name is not None:
movie_data['titles'].append(alt_name)
# Cache movie parsed
self.setCache(cache_key, movie_data)
return movie_data return movie_data
@ -192,23 +180,22 @@ class TheMovieDb(MovieProvider):
image_urls = [] image_urls = []
try: try:
for image in movie.get('images', {}).get(type, [])[1:5]:
# Full data
images = self.request('movie/%s/images' % movie.get('id'), return_key = type)
for image in images[1:5]:
image_urls.append(self.getImage(image, 'file', size)) image_urls.append(self.getImage(image, 'file', size))
except: except:
log.debug('Failed getting %s.%s for "%s"', (type, size, ss(str(movie)))) log.debug('Failed getting %s.%s for "%s"', (type, size, ss(str(movie))))
return image_urls return image_urls
def request(self, call = '', params = {}, return_key = None): def request(self, call = '', params = {}, return_key = None):
params = dict((k, v) for k, v in params.items() if v)
params = tryUrlencode(params) params = tryUrlencode(params)
url = 'http://api.themoviedb.org/3/%s?api_key=%s%s' % (call, self.conf('api_key'), '&%s' % params if params else '') url = 'http://api.themoviedb.org/3/%s?api_key=%s%s' % (call, self.conf('api_key'), '&%s' % params if params else '')
data = self.getJsonData(url, cache_timeout = 0) data = self.getJsonData(url)
if data and return_key and data.get(return_key): if data and return_key and return_key in data:
data = data.get(return_key) data = data.get(return_key)
return data return data

Write Preview
Loading…
Cancel
Save