You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
97 lines
3.3 KiB
97 lines
3.3 KiB
from __future__ import with_statement
|
|
from couchpotato.core.event import addEvent
|
|
from couchpotato.core.helpers.encoding import simplifyString, toUnicode
|
|
from couchpotato.core.logger import CPLog
|
|
from couchpotato.core.providers.base import Provider
|
|
from couchpotato.environment import Env
|
|
from urllib import quote_plus
|
|
import copy
|
|
import simplejson as json
|
|
import urllib2
|
|
|
|
log = CPLog(__name__)
|
|
|
|
class TMDB(Provider):
|
|
"""Api for theMovieDb"""
|
|
|
|
type = 'movie'
|
|
apiUrl = 'http://api.themoviedb.org/2.1'
|
|
imageUrl = 'http://hwcdn.themoviedb.org'
|
|
|
|
def __init__(self):
|
|
addEvent('provider.movie.search', self.search)
|
|
|
|
def conf(self, attr):
|
|
return Env.setting(attr, 'themoviedb')
|
|
|
|
def search(self, q, limit = 12, alternative = True):
|
|
''' Find movie by name '''
|
|
|
|
if self.isDisabled():
|
|
return False
|
|
|
|
log.debug('TheMovieDB - Searching for movie: %s' % q)
|
|
|
|
url = "%s/%s/%s/json/%s/%s" % (self.apiUrl, 'Movie.search', 'en', self.conf('api_key'), quote_plus(simplifyString(q)))
|
|
|
|
log.info('Searching: %s' % url)
|
|
|
|
data = urllib2.urlopen(url)
|
|
jsn = json.load(data)
|
|
|
|
return self.parse(jsn, limit, alternative = alternative)
|
|
|
|
def parse(self, data, limit, alternative = True):
|
|
if data:
|
|
log.debug('TheMovieDB - Parsing RSS')
|
|
try:
|
|
results = []
|
|
nr = 0
|
|
for movie in data:
|
|
|
|
year = str(movie['released'])[:4]
|
|
|
|
# Poster url
|
|
poster = ''
|
|
for p in movie['posters']:
|
|
p = p['image']
|
|
if(p['size'] == 'thumb'):
|
|
poster = p['url']
|
|
break
|
|
|
|
# 1900 is the same as None
|
|
if year == '1900' or year.lower() == 'none':
|
|
year = None
|
|
|
|
movie_data = {
|
|
'id': int(movie['id']),
|
|
'name': toUnicode(movie['name']),
|
|
'poster': poster,
|
|
'imdb': movie['imdb_id'],
|
|
'year': year,
|
|
'tagline': 'This is the tagline of the movie',
|
|
}
|
|
results.append(copy.deepcopy(movie_data))
|
|
|
|
alternativeName = movie['alternative_name']
|
|
if alternativeName and alternative:
|
|
if alternativeName.lower() != movie['name'].lower() and alternativeName.lower() != 'none' and alternativeName != None:
|
|
movie_data['name'] = toUnicode(alternativeName)
|
|
results.append(copy.deepcopy(movie_data))
|
|
nr += 1
|
|
if nr == limit:
|
|
break
|
|
|
|
log.info('TheMovieDB - Found: %s' % [result['name'] + u' (' + str(result['year']) + ')' for result in results])
|
|
return results
|
|
except SyntaxError, e:
|
|
log.error('TheMovieDB - Failed to parse XML response from TheMovieDb: %s' % e)
|
|
return False
|
|
|
|
|
|
def isDisabled(self):
|
|
if self.conf('api_key') == '':
|
|
log.error('TheMovieDB - No API key provided for TheMovieDB')
|
|
True
|
|
else:
|
|
False
|
|
|