def get(self, session=None):
args = lookup_parser.parse_args()
include_actors = args.pop('include_actors')
try:
series = at.lookup_series(session=session, **args)
except LookupError as e:
return {'status': 'error',
'message': e.args[0]
}, 404
result = series.to_dict()
if include_actors:
result["actors"] = list_actors(series.actors),
return jsonify(result)
def get(self, title, session=None):
args = lookup_parser.parse_args()
include_actors = args.pop('include_actors')
include_translations = args.pop('include_translations')
kwargs = args
kwargs['title'] = title
try:
movie = at.lookup_movie(session=session, **kwargs)
except LookupError as e:
return {'status': 'error',
'message': e.args[0]
}, 404
result = movie.to_dict()
if include_actors:
result["actors"] = list_actors(movie.actors)
if include_translations:
result["translations"] = get_translations(movie.translate)
return jsonify(result)
class PluginTraktLookup(object):
"""Retrieves trakt information for entries. Uses series_name,
series_season, series_episode from series plugin.
Example:
trakt_lookup: yes
Primarily used for passing trakt information to other plugins.
Among these is the IMDB url for the series.
This information is provided (via entry):
series info:
trakt_series_name
trakt_series_runtime
trakt_series_first_aired_epoch
trakt_series_first_aired_iso
trakt_series_air_time
trakt_series_content_rating
trakt_series_genres
trakt_series_banner_url
trakt_series_fanart_url
trakt_series_imdb_url
trakt_series_trakt_url
trakt_series_images
imdb_id
tvdb_id
trakt_series_actors
trakt_series_country
trakt_series_year
trakt_series_tvrage_id
trakt_series_status
trakt_series_overview
trakt_ep_name
trakt_ep_season
trakt_ep_number
trakt_ep_overview
trakt_ep_first_aired_epoch
trakt_ep_first_aired_iso
trakt_ep_image_url
trakt_ep_id
trakt_ep_tvdb_id
trakt_ep_images
"""
# Series info
series_map = {
'trakt_series_name':
'title',
'trakt_series_year':
'year',
'imdb_id':
'imdb_id',
'tvdb_id':
'tvdb_id',
'tmdb_id':
'tmdb_id',
'trakt_show_id':
'id',
'trakt_slug':
'slug',
'tvrage_id':
'tvrage_id',
'trakt_trailer':
'trailer',
'trakt_homepage':
'homepage',
'trakt_series_runtime':
'runtime',
'trakt_series_first_aired':
'first_aired',
'trakt_series_air_time':
'air_time',
'trakt_series_air_day':
'air_day',
'trakt_series_content_rating':
'certification',
'trakt_genres':
lambda i: [db_genre.name for db_genre in i.genres],
'trakt_series_network':
'network',
'imdb_url':
lambda series: series.imdb_id and 'http://www.imdb.com/title/%s' %
series.imdb_id,
'trakt_series_url':
lambda series: series.slug and 'http://trakt.tv/shows/%s' % series.
slug,
'trakt_series_country':
'country',
'trakt_series_status':
'status',
'trakt_series_overview':
'overview',
'trakt_series_rating':
'rating',
'trakt_series_votes':
'votes',
'trakt_series_language':
'language',
'trakt_series_aired_episodes':
'aired_episodes',
'trakt_series_episodes':
lambda show: [episodes.title for episodes in show.episodes],
'trakt_series_images':
lambda im: list_images(im.images),
'trakt_languages':
lambda i: [t.name for t in i.translations],
}
series_actor_map = {
'trakt_actors': lambda show: list_actors(show.actors),
}
show_translate_map = {
'trakt_translations': lambda show: get_translations(show.translate),
}
# Episode info
episode_map = {
'trakt_ep_name': 'title',
'trakt_ep_imdb_id': 'imdb_id',
'trakt_ep_tvdb_id': 'tvdb_id',
'trakt_ep_tmdb_id': 'tmdb_id',
'trakt_ep_tvrage': 'tvrage_id',
'trakt_episode_id': 'id',
'trakt_ep_first_aired': 'first_aired',
'trakt_ep_overview': 'overview',
'trakt_ep_abs_number': 'number_abs',
'trakt_season': 'season',
'trakt_episode': 'number',
'trakt_ep_id': lambda ep: 'S%02dE%02d' % (ep.season, ep.number),
'trakt_ep_images': lambda im: list_images(im.images),
}
# Movie info
movie_map = {
'movie_name': 'title',
'movie_year': 'year',
'trakt_movie_name': 'title',
'trakt_movie_year': 'year',
'trakt_movie_id': 'id',
'trakt_slug': 'slug',
'imdb_id': 'imdb_id',
'tmdb_id': 'tmdb_id',
'trakt_tagline': 'tagline',
'trakt_overview': 'overview',
'trakt_released': 'released',
'trakt_runtime': 'runtime',
'trakt_rating': 'rating',
'trakt_votes': 'votes',
'trakt_homepage': 'homepage',
'trakt_trailer': 'trailer',
'trakt_language': 'language',
'trakt_genres': lambda i: [db_genre.name for db_genre in i.genres],
'trakt_images': lambda im: list_images(im.images),
'trakt_languages': lambda i: [t.name for t in i.translations]
}
movie_translate_map = {
'trakt_translations': lambda movie: get_translations(movie.translate),
}
movie_actor_map = {
'trakt_actors': lambda movie: list_actors(movie.actors),
}
schema = {
'oneOf': [{
'type':
'object',
'properties': {
'account': {
'type': 'string'
},
'username': {
'type': 'string'
},
},
'anyOf': [{
'required': ['username']
}, {
'required': ['account']
}],
'error_anyOf':
'At least one of `username` or `account` options are needed.',
'additionalProperties':
False
}, {
'type': 'boolean'
}]
}
def lazy_series_lookup(self, entry):
"""Does the lookup for this entry and populates the entry fields."""
with Session() as session:
lookupargs = {
'title': entry.get('series_name', eval_lazy=False),
'year': entry.get('year', eval_lazy=False),
'trakt_id': entry.get('trakt_show_id', eval_lazy=False),
'tvdb_id': entry.get('tvdb_id', eval_lazy=False),
'tmdb_id': entry.get('tmdb_id', eval_lazy=False),
'session': session
}
try:
series = lookup_series(**lookupargs)
except LookupError as e:
log.debug(e.args[0])
else:
entry.update_using_map(self.series_map, series)
return entry
def lazy_series_actor_lookup(self, entry):
"""Does the lookup for this entry and populates the entry fields."""
with Session() as session:
lookupargs = {
'trakt_id': entry.get('trakt_show_id', eval_lazy=False),
'title': entry.get('series_name', eval_lazy=False),
'session': session
}
try:
series = lookup_series(**lookupargs)
except LookupError as e:
log.debug(e.args[0])
else:
entry.update_using_map(self.series_actor_map, series)
return entry
def lazy_series_translate_lookup(self, entry):
"""Does the lookup for this entry and populates the entry fields."""
with Session() as session:
lookupargs = {
'trakt_id': entry.get('trakt_movie_id', eval_lazy=False),
'title': entry.get('series_name', eval_lazy=False),
'session': session
}
try:
series = lookup_series(**lookupargs)
except LookupError as e:
log.debug(e.args[0])
else:
entry.update_using_map(self.show_translate_map, series)
return entry
def lazy_episode_lookup(self, entry):
with Session(expire_on_commit=False) as session:
lookupargs = {
'title': entry.get('series_name', eval_lazy=False),
'trakt_id': entry.get('trakt_show_id', eval_lazy=False),
'session': session
}
try:
series = lookup_series(**lookupargs)
episode = series.get_episode(entry['series_season'],
entry['series_episode'], session)
except LookupError as e:
log.debug(
'Error looking up trakt episode information for %s: %s',
entry['title'], e.args[0])
else:
entry.update_using_map(self.episode_map, episode)
return entry
def lazy_movie_lookup(self, entry):
"""Does the lookup for this entry and populates the entry fields."""
with Session() as session:
lookupargs = {
'title': entry.get('title', eval_lazy=False),
'year': entry.get('year', eval_lazy=False),
'trakt_id': entry.get('trakt_movie_id', eval_lazy=False),
'trakt_slug': entry.get('trakt_movie_slug', eval_lazy=False),
'tmdb_id': entry.get('tmdb_id', eval_lazy=False),
'imdb_id': entry.get('imdb_id', eval_lazy=False),
'session': session
}
try:
movie = lookup_movie(**lookupargs)
except LookupError as e:
log.debug(e.args[0])
else:
entry.update_using_map(self.movie_map, movie)
return entry
def lazy_movie_actor_lookup(self, entry):
"""Does the lookup for this entry and populates the entry fields."""
with Session() as session:
lookupargs = {
'trakt_id': entry.get('trakt_movie_id', eval_lazy=False),
'title': entry.get('title', eval_lazy=False),
'session': session
}
try:
movie = lookup_movie(**lookupargs)
except LookupError as e:
log.debug(e.args[0])
else:
entry.update_using_map(self.movie_actor_map, movie)
return entry
def lazy_movie_translate_lookup(self, entry):
"""Does the lookup for this entry and populates the entry fields."""
with Session() as session:
lookupargs = {
'trakt_id': entry.get('trakt_movie_id', eval_lazy=False),
'title': entry.get('movie_name', eval_lazy=False),
'session': session
}
try:
movie = lookup_movie(**lookupargs)
except LookupError as e:
log.debug(e.args[0])
else:
entry.update_using_map(self.movie_translate_map, movie)
return entry
def lazy_collected_lookup(self, config, style, entry):
"""Does the lookup for this entry and populates the entry fields."""
if style == 'show' or style == 'episode':
lookup = lookup_series
trakt_id = entry.get('trakt_show_id', eval_lazy=True)
else:
lookup = lookup_movie
trakt_id = entry.get('trakt_movie_id', eval_lazy=True)
with Session() as session:
lookupargs = {'trakt_id': trakt_id, 'session': session}
try:
item = lookup(**lookupargs)
if style == 'episode':
item = item.get_episode(entry['series_season'],
entry['series_episode'], session)
collected = ApiTrakt.collected(style,
item,
entry.get('title'),
username=config.get('username'),
account=config.get('account'))
except LookupError as e:
log.debug(e.args[0])
else:
entry['trakt_collected'] = collected
return entry
def lazy_watched_lookup(self, config, style, entry):
"""Does the lookup for this entry and populates the entry fields."""
if style == 'show' or style == 'episode':
lookup = lookup_series
trakt_id = entry.get('trakt_show_id', eval_lazy=True)
else:
lookup = lookup_movie
trakt_id = entry.get('trakt_movie_id', eval_lazy=True)
with Session() as session:
lookupargs = {'trakt_id': trakt_id, 'session': session}
try:
item = lookup(**lookupargs)
if style == 'episode':
item = item.get_episode(entry['series_season'],
entry['series_episode'], session)
watched = ApiTrakt.watched(style,
item,
entry.get('title'),
username=config.get('username'),
account=config.get('account'))
except LookupError as e:
log.debug(e.args[0])
else:
entry['trakt_watched'] = watched
return entry
# Run after series and metainfo series
@plugin.priority(110)
def on_task_metainfo(self, task, config):
if not config:
return
if isinstance(config, bool):
config = dict()
for entry in task.entries:
if entry.get('series_name') or entry.get('tvdb_id',
eval_lazy=False):
style = 'show'
entry.register_lazy_func(self.lazy_series_lookup,
self.series_map)
# TODO cleaner way to do this?
entry.register_lazy_func(self.lazy_series_actor_lookup,
self.series_actor_map)
entry.register_lazy_func(self.lazy_series_translate_lookup,
self.show_translate_map)
if 'series_season' in entry and 'series_episode' in entry:
entry.register_lazy_func(self.lazy_episode_lookup,
self.episode_map)
style = 'episode'
if config.get('username') or config.get('account'):
collected_lookup = functools.partial(
self.lazy_collected_lookup, config, style)
watched_lookup = functools.partial(
self.lazy_watched_lookup, config, style)
entry.register_lazy_func(collected_lookup,
['trakt_collected'])
entry.register_lazy_func(watched_lookup, ['trakt_watched'])
else:
entry.register_lazy_func(self.lazy_movie_lookup,
self.movie_map)
# TODO cleaner way to do this?
entry.register_lazy_func(self.lazy_movie_actor_lookup,
self.movie_actor_map)
entry.register_lazy_func(self.lazy_movie_translate_lookup,
self.movie_translate_map)
if config.get('username') or config.get('account'):
collected_lookup = functools.partial(
self.lazy_collected_lookup, config, 'movie')
watched_lookup = functools.partial(
self.lazy_watched_lookup, config, 'movie')
entry.register_lazy_func(collected_lookup,
['trakt_collected'])
entry.register_lazy_func(watched_lookup, ['trakt_watched'])
@property
def series_identifier(self):
"""Returns the plugin main identifier type"""
return 'trakt_show_id'
@property
def movie_identifier(self):
"""Returns the plugin main identifier type"""
return 'trakt_movie_id'
