import copy
from flask import jsonify, redirect
from flask_restx import inputs
from flexget.api import APIResource, api
from flexget.api.app import NotFoundError, etag
from . import db
from .api_trakt import ApiTrakt
trakt_api = api.namespace('trakt', description='Trakt lookup endpoint')
[docs]
class ObjectsContainer:
translation_object = {
'type': 'object',
'patternProperties': {
'^[/d]$': {
'type': 'object',
'properties': {
'overview': {'type': 'string'},
'tagline': {'type': 'string'},
'title': {'type': 'string'},
},
'required': ['overview', 'tagline', 'title'],
'additionalProperties': False,
}
},
}
actor_object = {
'type': 'object',
'patternProperties': {
'^[/d]$': {
'type': 'object',
'properties': {
'imdb_id': {'type': 'string'},
'name': {'type': 'string'},
'tmdb_id': {'type': 'integer'},
'trakt_id': {'type': 'integer'},
'trakt_slug': {'type': 'string'},
'birthday': {'type': 'string'},
'biography': {'type': ['string', 'null']},
'homepage': {'type': 'string'},
'death': {'type': ['string', 'null']},
},
'required': [
'imdb_id',
'name',
'tmdb_id',
'trakt_id',
'trakt_slug',
'birthday',
'biography',
'homepage',
'death',
],
'additionalProperties': False,
}
},
}
base_return_object = {
'type': 'object',
'properties': {
'translations': translation_object,
'actors': actor_object,
'cached_at': {'type': 'string', 'format': 'date-time'},
'genres': {'type': 'array', 'items': {'type': 'string'}},
'id': {'type': 'integer'},
'overview': {'type': ['string', 'null']},
'runtime': {'type': ['integer', 'null']},
'rating': {'type': ['number', 'null']},
'votes': {'type': ['integer', 'null']},
'language': {'type': ['string', 'null']},
'updated_at': {'type': 'string', 'format': 'date-time'},
'title': {'type': 'string'},
'year': {'type': ['integer', 'null']},
'homepage': {'type': ['string', 'null']},
'slug': {'type': ['string', 'null']},
'tmdb_id': {'type': ['integer', 'null']},
'imdb_id': {'type': ['string', 'null']},
},
'required': [
'cached_at',
'genres',
'id',
'overview',
'runtime',
'rating',
'votes',
'language',
'updated_at',
'title',
'year',
'homepage',
'slug',
'tmdb_id',
'imdb_id',
],
'additionalProperties': False,
}
series_return_object = copy.deepcopy(base_return_object)
series_return_object['properties']['tvdb_id'] = {'type': ['integer', 'null']}
series_return_object['properties']['tvrage_id'] = {'type': ['integer', 'null']}
series_return_object['properties']['first_aired'] = {
'type': ['string', 'null'],
'format': 'date-time',
}
series_return_object['properties']['air_day'] = {'type': ['string', 'null']}
series_return_object['properties']['air_time'] = {'type': ['string', 'null']}
series_return_object['properties']['certification'] = {'type': ['string', 'null']}
series_return_object['properties']['network'] = {'type': ['string', 'null']}
series_return_object['properties']['country'] = {'type': ['string', 'null']}
series_return_object['properties']['status'] = {'type': 'string'}
series_return_object['properties']['timezone'] = {'type': ['string', 'null']}
series_return_object['properties']['number_of_aired_episodes'] = {'type': ['integer', 'null']}
series_return_object['required'] += [
'tvdb_id',
'tvrage_id',
'first_aired',
'air_day',
'air_time',
'certification',
'network',
'country',
'status',
'timezone',
'number_of_aired_episodes',
]
movie_return_object = copy.deepcopy(base_return_object)
movie_return_object['properties']['tagline'] = {'type': 'string'}
movie_return_object['properties']['released'] = {'type': 'string'}
movie_return_object['properties']['trailer'] = {'type': ['string', 'null']}
movie_return_object['required'] += ['tagline', 'released', 'trailer']
series_return_schema = api.schema_model(
'series_return_schema', ObjectsContainer.series_return_object
)
movie_return_schema = api.schema_model('movie_return_schema', ObjectsContainer.movie_return_object)
lookup_parser = api.parser()
lookup_parser.add_argument('title', help='Movie Name')
lookup_parser.add_argument('year', type=int, help='Lookup year')
lookup_parser.add_argument('trakt_id', type=int, help='Trakt ID')
lookup_parser.add_argument('trakt_slug', help='Trakt slug')
lookup_parser.add_argument('tmdb_id', type=int, help='TMDB ID')
lookup_parser.add_argument('imdb_id', help='IMDB ID')
lookup_parser.add_argument('tvdb_id', type=int, help='TVDB ID')
lookup_parser.add_argument('tvrage_id', type=int, help='TVRage ID')
lookup_parser.add_argument(
'include_actors', type=inputs.boolean, help='Include actors in response'
)
lookup_parser.add_argument(
'include_translations', type=inputs.boolean, help='Include translations in response'
)
[docs]
@trakt_api.route('/series/')
class TraktSeriesSearchApi(APIResource):
[docs]
@etag(cache_age=3600)
@api.response(200, 'Successfully found show', series_return_schema)
@api.response(NotFoundError)
@api.doc(expect=[lookup_parser])
def get(self, session=None):
"""Trakt series lookup."""
args = lookup_parser.parse_args()
include_actors = args.pop('include_actors')
include_translations = args.pop('include_translations')
kwargs = args
try:
series = ApiTrakt.lookup_series(session=session, **kwargs)
except LookupError as e:
raise NotFoundError(e.args[0])
result = series.to_dict()
if include_actors:
result['actors'] = db.list_actors(series.actors)
if include_translations:
result['translations'] = db.get_translations_dict(series.translations, 'show')
return jsonify(result)
[docs]
@trakt_api.route('/series/<string:title>')
@api.doc(params={'title': 'Series name'})
class TraktSeriesWithTitleSearchApi(APIResource):
[docs]
@api.doc(expect=[lookup_parser])
@api.response(301)
def get(self, title, session=None):
kwargs = lookup_parser.parse_args()
kwargs['title'] = title
return redirect(api.url_for(TraktSeriesSearchApi, **kwargs), code=301)
[docs]
@trakt_api.route('/movies/')
class TraktMovieSearchApi(APIResource):
[docs]
@etag(cache_age=3600)
@api.response(200, 'Successfully found show', movie_return_schema)
@api.response(NotFoundError)
@api.doc(expect=[lookup_parser])
def get(self, session=None):
"""Trakt movie lookup."""
args = lookup_parser.parse_args()
include_actors = args.pop('include_actors')
include_translations = args.pop('include_translations')
kwargs = args
try:
movie = ApiTrakt.lookup_movie(session=session, **kwargs)
except LookupError as e:
raise NotFoundError(e.args[0])
result = movie.to_dict()
if include_actors:
result['actors'] = db.list_actors(movie.actors)
if include_translations:
result['translations'] = db.get_translations_dict(movie.translations, 'movie')
return jsonify(result)
[docs]
@trakt_api.route('/movies/<string:title>')
@api.doc(params={'title': 'Movie name'})
class TraktMovieWithTitleSearchApi(APIResource):
[docs]
@api.doc(expect=[lookup_parser])
@api.response(301)
def get(self, title, session=None):
kwargs = lookup_parser.parse_args()
kwargs['title'] = title
return redirect(api.url_for(TraktMovieSearchApi, **kwargs), code=301)
