473 lines
16 KiB
Python
473 lines
16 KiB
Python
# !/usr/bin/env python
|
|
#-*- coding:utf-8 -*-
|
|
#author:doganaydin
|
|
#project:themoviedb
|
|
#repository:http://github.com/doganaydin/themoviedb
|
|
#license: LGPLv3 http://www.gnu.org/licenses/lgpl.html
|
|
"""An interface to the themoviedb.org API"""
|
|
|
|
__author__ = "doganaydin"
|
|
__version__ = "1.0b"
|
|
import os
|
|
|
|
os.sys.path.insert(0, os.path.join(os.path.dirname(os.path.abspath(__file__))))
|
|
try:
|
|
import simplejson
|
|
except:
|
|
import json as simplejson
|
|
|
|
import fuzzywuzzy.fuzz
|
|
import requests
|
|
|
|
config = {}
|
|
|
|
|
|
def configure(api_key, language='en'):
|
|
config['apikey'] = api_key
|
|
config['language'] = language
|
|
config['urls'] = {}
|
|
config['urls'][
|
|
'movie.search'] = "https://api.themoviedb.org/3/search/movie?query=%%s&year=%%s&api_key=%(apikey)s&page=%%s" % (
|
|
config)
|
|
config['urls']['movie.info'] = "https://api.themoviedb.org/3/movie/%%s?api_key=%(apikey)s" % (config)
|
|
config['urls'][
|
|
'people.search'] = "https://api.themoviedb.org/3/search/person?query=%%s&api_key=%(apikey)s&page=%%s" % (config)
|
|
config['urls']['collection.info'] = "https://api.themoviedb.org/3/collection/%%s&api_key=%(apikey)s" % (config)
|
|
config['urls'][
|
|
'movie.alternativetitles'] = "https://api.themoviedb.org/3/movie/%%s/alternative_titles?api_key=%(apikey)s" % (
|
|
config)
|
|
config['urls']['movie.casts'] = "https://api.themoviedb.org/3/movie/%%s/casts?api_key=%(apikey)s" % (config)
|
|
config['urls']['movie.images'] = "https://api.themoviedb.org/3/movie/%%s/images?api_key=%(apikey)s" % (config)
|
|
config['urls']['movie.keywords'] = "https://api.themoviedb.org/3/movie/%%s/keywords?api_key=%(apikey)s" % (config)
|
|
config['urls']['movie.releases'] = "https://api.themoviedb.org/3/movie/%%s/releases?api_key=%(apikey)s" % (config)
|
|
config['urls']['movie.trailers'] = "https://api.themoviedb.org/3/movie/%%s/trailers?api_key=%(apikey)s" % (config)
|
|
config['urls']['movie.translations'] = "https://api.themoviedb.org/3/movie/%%s/translations?api_key=%(apikey)s" % (
|
|
config)
|
|
config['urls'][
|
|
'person.info'] = "https://api.themoviedb.org/3/person/%%s?api_key=%(apikey)s&append_to_response=images,credits" % (
|
|
config)
|
|
config['urls']['latestmovie'] = "https://api.themoviedb.org/3/latest/movie?api_key=%(apikey)s" % (config)
|
|
config['urls']['config'] = "https://api.themoviedb.org/3/configuration?api_key=%(apikey)s" % (config)
|
|
config['urls']['request.token'] = "https://api.themoviedb.org/3/authentication/token/new?api_key=%(apikey)s" % (
|
|
config)
|
|
config['urls'][
|
|
'session.id'] = "https://api.themoviedb.org/3/authentication/session/new?api_key=%(apikey)s&request_token=%%s" % (
|
|
config)
|
|
config['urls'][
|
|
'movie.add.rating'] = "https://api.themoviedb.org/3/movie/%%s/rating?session_id=%%s&api_key=%(apikey)s" % (
|
|
config)
|
|
config['api'] = {}
|
|
config['api']['backdrop.sizes'] = ""
|
|
config['api']['base.url'] = ""
|
|
config['api']['poster.sizes'] = ""
|
|
config['api']['profile.sizes'] = ""
|
|
config['api']['session.id'] = ""
|
|
|
|
|
|
class Core(object):
|
|
def getJSON(self, url, language=None):
|
|
language = language or config['language']
|
|
page = requests.get(url, params={'language': language}).content
|
|
try:
|
|
return simplejson.loads(page)
|
|
except:
|
|
try:
|
|
return simplejson.loads(page.decode('utf-8'))
|
|
except:
|
|
return
|
|
|
|
def escape(self, text):
|
|
if len(text) > 0:
|
|
try:
|
|
return requests.utils.quote(text)
|
|
except:
|
|
return requests.utils.quote(text.encode('utf-8'))
|
|
return False
|
|
|
|
def update_configuration(self):
|
|
c = self.getJSON(config['urls']['config'])
|
|
try:
|
|
config['api']['backdrop.sizes'] = c['images']['backdrop_sizes']
|
|
config['api']['base.url'] = c['images']['base_url']
|
|
config['api']['poster.sizes'] = c['images']['poster_sizes']
|
|
config['api']['profile.sizes'] = c['images']['profile_sizes']
|
|
return "ok"
|
|
except:
|
|
return 'not ok'
|
|
|
|
def backdrop_sizes(self, img_size):
|
|
size_list = {'s': 'w300', 'm': 'w780', 'l': 'w1280', 'o': 'original'}
|
|
return size_list[img_size]
|
|
|
|
def poster_sizes(self, img_size):
|
|
size_list = {'s': 'w92', 'm': 'w185', 'l': 'w500', 'o': 'original'}
|
|
return size_list[img_size]
|
|
|
|
def profile_sizes(self, img_size):
|
|
size_list = {'s': 'w45', 'm': '185', 'l': 'w632', 'o': 'original'}
|
|
return size_list[img_size]
|
|
|
|
def request_token(self):
|
|
req = self.getJSON(config['urls']['request.token'])
|
|
r = req["request_token"]
|
|
return {"url": "http://themoviedb.org/authenticate/%s" % r, "request_token": r}
|
|
|
|
def session_id(self, token):
|
|
sess = self.getJSON(config['urls']['session.id'] % token)
|
|
config['api']['session.id'] = sess["session_id"]
|
|
return sess["session_id"]
|
|
|
|
|
|
class Movies(Core):
|
|
def __init__(self, title="", year="", limit=True, language=None):
|
|
self.limit = limit
|
|
self.update_configuration()
|
|
self.searched = title
|
|
title = self.escape(title)
|
|
self.movies = self.getJSON(config['urls']['movie.search'] % (title, str(year), str(1)), language=language)
|
|
#pages = self.movies.get('total_pages')
|
|
#if not self.limit and pages:
|
|
# if int(pages) > 1: #
|
|
# for i in range(2,int(pages)+1): # Thanks @tBuLi
|
|
# self.movies["results"].extend(self.getJSON(config['urls']['movie.search'] % (title,str(i)), language=language)["results"])
|
|
|
|
def __iter__(self):
|
|
if self.movies.get('results'):
|
|
for i in self.movies["results"]:
|
|
yield Movie(i["id"])
|
|
|
|
def get_total_results(self):
|
|
if self.limit:
|
|
return len(self.movies["results"])
|
|
return self.movies["total_results"]
|
|
|
|
def iter_results(self):
|
|
for i in self.movies["results"]:
|
|
yield i
|
|
|
|
def get_ordered_matches(self):
|
|
"""
|
|
Return a list of tuples. Each tuples first element is a percentage similarity
|
|
between the search term and the tuples second element 'title' value.
|
|
|
|
Ordered, descending, by the percentage similarity.
|
|
"""
|
|
our_results = []
|
|
for movie in self.iter_results():
|
|
ratio = fuzzywuzzy.fuzz.ratio(self.searched, movie['title'])
|
|
our_results.append((ratio, movie))
|
|
return sorted(our_results, reverse=True)
|
|
|
|
def get_best_match(self):
|
|
"""
|
|
Returns a tuple whose first element is the percent similarity between the search
|
|
term and the tuple's second element's 'title' value.
|
|
|
|
The result is the best-matching result from all the results we get from TMDb.
|
|
"""
|
|
for show in Data:
|
|
itemIndex = itemIndex + 1
|
|
Shref = show['link']
|
|
Syear = show['year'][0]
|
|
|
|
Stitle = show['title']
|
|
|
|
rate = scoreMediaTitleMatch(title, ruTitle, year, Stitle.encode('utf-8'), Syear, itemIndex)
|
|
shows.append({'rate': rate, 'href': Shref, 'title': Stitle, 'year': Syear})
|
|
|
|
shows = sorted(shows, key=lambda x: x['rate'], reverse=True)
|
|
#print str(shows)
|
|
if shows:
|
|
return shows[0]
|
|
try:
|
|
return self.get_ordered_matches()[0]
|
|
except IndexError:
|
|
return
|
|
|
|
|
|
class Movie(Core):
|
|
def __init__(self, movie_id, language=None):
|
|
self.movie_id = movie_id
|
|
self.update_configuration()
|
|
self.movies = self.getJSON(config['urls']['movie.info'] % self.movie_id, language=language)
|
|
self.casts = self.getJSON(config['urls']['movie.casts'] % self.movie_id, language=language)
|
|
|
|
def is_adult(self):
|
|
return self.movies['adult']
|
|
|
|
def get_collection_id(self):
|
|
return self.movies['belongs_to_collection']["id"]
|
|
|
|
def get_collection_name(self):
|
|
return self.movies['belongs_to_collection']["name"]
|
|
|
|
# Sizes = s->w300 m->w780 l->w1280 o->original(default)
|
|
def get_collection_backdrop(self, img_size="o"):
|
|
try:
|
|
img_path = self.movies["belongs_to_collection"]["backdrop_path"]
|
|
except KeyError:
|
|
return
|
|
return config['api']['base.url'] + self.poster_sizes(img_size) + img_path
|
|
|
|
# Sizes = s->w92 m->w185 l->w500 o->original(default)
|
|
def get_collection_poster(self, img_size="o"):
|
|
try:
|
|
img_path = self.movies["belongs_to_collection"]["poster_path"]
|
|
except KeyError:
|
|
return
|
|
return config['api']['base.url'] + self.poster_sizes(img_size) + img_path
|
|
|
|
def get_budget(self):
|
|
return self.movies['budget']
|
|
|
|
def get_genres(self):
|
|
genres = []
|
|
for i in self.movies['genres']:
|
|
genres.append({"id": i["id"], "name": i["name"]})
|
|
return genres
|
|
|
|
def get_homepage(self):
|
|
return self.movies['homepage']
|
|
|
|
def get_imdb_id(self):
|
|
return self.movies['imdb_id']
|
|
|
|
def get_overview(self):
|
|
return self.movies['overview']
|
|
|
|
def get_production_companies(self):
|
|
companies = []
|
|
for i in self.movies['production_companies']:
|
|
companies.append({"id": i["id"], "name": i["name"]})
|
|
return companies
|
|
|
|
def get_productions_countries(self):
|
|
countries = []
|
|
for i in self.movies['production_countries']:
|
|
countries.append({"iso_3166_1": i["iso_3166_1"], "name": i["name"]})
|
|
return countries
|
|
|
|
def get_revenue(self):
|
|
return self.movies['revenue']
|
|
|
|
def get_runtime(self):
|
|
return self.movies['runtime']
|
|
|
|
def get_spoken_languages(self):
|
|
langs = []
|
|
for i in self.movies['spoken_languages']:
|
|
langs.append({"iso_639_1": i["iso_639_1"], "name": i["name"]})
|
|
return langs
|
|
|
|
def get_tagline(self):
|
|
return self.movies['tagline']
|
|
|
|
def get_vote_average(self):
|
|
return self.movies['vote_average']
|
|
|
|
def get_vote_count(self):
|
|
return self.movies['vote_count']
|
|
|
|
def get_id(self):
|
|
return self.movie_id
|
|
|
|
# Sizes = s->w300 m->w780 l->w1280 o->original(default)
|
|
def get_backdrop(self, img_size="o"):
|
|
img_path = self.movies.get("backdrop_path")
|
|
if not img_path:
|
|
return
|
|
return config['api']['base.url'] + self.backdrop_sizes(img_size) + img_path
|
|
|
|
def get_original_title(self):
|
|
return self.movies["original_title"]
|
|
|
|
def get_popularity(self):
|
|
return self.movies["popularity"]
|
|
|
|
def get_release_date(self):
|
|
return self.movies["release_date"]
|
|
|
|
def get_title(self):
|
|
return self.movies["title"]
|
|
|
|
# Sizes = s->w92 m->w185 l->w500 o->original(default)
|
|
def get_poster(self, img_size="o"):
|
|
img_path = self.movies.get("poster_path")
|
|
if not img_path:
|
|
return
|
|
return config['api']['base.url'] + self.poster_sizes(img_size) + img_path
|
|
|
|
def get_trailers(self, language=None):
|
|
return self.getJSON(config['urls']['movie.trailers'] % self.movie_id, language=language)
|
|
|
|
def add_rating(self, value):
|
|
if isinstance(value, float) or isinstance(value, int):
|
|
if config["api"]["session.id"] == "":
|
|
return "PROBLEM_AUTH"
|
|
sess_id = config["api"]["session.id"]
|
|
data = {"value": float(value)}
|
|
req = requests.post(config['urls']['movie.add.rating'] % (self.movie_id, sess_id), data=data)
|
|
res = simplejson.loads(bytes(req.content).decode())
|
|
if res['status_message'] == "Success":
|
|
return True
|
|
else:
|
|
return False
|
|
return "ERROR"
|
|
|
|
|
|
class People(Core):
|
|
def __init__(self, people_name, limit=False, language=None):
|
|
self.limit = limit
|
|
self.update_configuration()
|
|
people_name = self.escape(people_name)
|
|
self.people = self.getJSON(config['urls']['people.search'] % (people_name, str(1)), language=language)
|
|
pages = self.people["total_pages"]
|
|
if not self.limit:
|
|
if int(pages) > 1:
|
|
for i in range(2, int(pages) + 1):
|
|
self.people["results"].extend(
|
|
self.getJSON(config['urls']['people.search'] % (people_name, str(i)), language=language)[
|
|
"results"])
|
|
|
|
def __iter__(self):
|
|
for i in self.people["results"]:
|
|
yield Person(i["id"])
|
|
|
|
def total_results(self):
|
|
return self.people["total_results"]
|
|
|
|
def get_total_results(self):
|
|
if self.limit:
|
|
return len(self.movies["results"])
|
|
return self.movies["total_results"]
|
|
|
|
def iter_results(self):
|
|
for i in self.people["results"]:
|
|
yield i
|
|
|
|
|
|
class Person(Core):
|
|
def __init__(self, person_id, language=None):
|
|
self.person_id = person_id
|
|
self.update_configuration()
|
|
self.person = self.getJSON(config['urls']['person.info'] % self.person_id, language=language)
|
|
|
|
def get_id(self):
|
|
return self.person_id
|
|
|
|
def is_adult(self):
|
|
return self.person["adult"]
|
|
|
|
def get_name(self):
|
|
return self.person["name"]
|
|
|
|
# Sizes = s->w45 m->w185 l->w632 o->original(default)
|
|
def get_profile_image(self, img_size="o"):
|
|
img_path = self.person.get("profile_path")
|
|
if not img_path:
|
|
return
|
|
return config['api']['base.url'] + self.profile_sizes(img_size) + img_path
|
|
|
|
def get_biography(self):
|
|
return self.person['biography']
|
|
|
|
def get_birthday(self):
|
|
return self.person['birthday']
|
|
|
|
def get_deathday(self):
|
|
return self.person['deathday']
|
|
|
|
def get_place_of_birth(self):
|
|
return self.person['place_of_birth']
|
|
|
|
def get_homepage(self):
|
|
return self.person['homepage']
|
|
|
|
def get_also_known_as(self):
|
|
return self.person['also_known_as']
|
|
|
|
def get_image_aspect_ratio(self, image_index=0):
|
|
return self.person["images"]['profiles'][image_index]['aspect_ratio']
|
|
|
|
def get_image_height(self, image_index=0):
|
|
return self.person["images"]['profiles'][image_index]['height']
|
|
|
|
def get_image_width(self, image_index=0):
|
|
return self.person["images"]['profiles'][image_index]['width']
|
|
|
|
def get_image_iso_639_1(self, image_index=0):
|
|
return self.person["images"]['profiles'][image_index]['iso_639_1']
|
|
|
|
#Sizes = s->w92 m->w185 l->w500 o->original(default)
|
|
def get_image(self, img_size="o", image_index=0):
|
|
try:
|
|
img_path = self.person["images"]['profiles'][image_index]['file_path']
|
|
except KeyError:
|
|
return
|
|
return config['api']['base.url'] + self.poster_sizes(img_size) + img_path
|
|
|
|
def cast(self):
|
|
for c in self.person["credits"]["cast"]:
|
|
yield Cast(c)
|
|
|
|
def crew(self):
|
|
for c in self.person["credits"]["crew"]:
|
|
yield Crew(c)
|
|
|
|
|
|
class Cast:
|
|
def __init__(self, c):
|
|
self.cast = c
|
|
|
|
def get_id(self):
|
|
return self.cast["id"]
|
|
|
|
def get_character(self):
|
|
return self.cast["character"]
|
|
|
|
def get_original_title(self):
|
|
return self.cast["original_title"]
|
|
|
|
def get_title(self):
|
|
return self.cast["title"]
|
|
|
|
def get_release_date(self):
|
|
return self.cast["release_date"]
|
|
|
|
# Sizes = s->w92 m->w185 l->w500 o->original(default)
|
|
def get_poster(self, img_size="o", person_index=0):
|
|
img_path = self.cast.get("poster_path")
|
|
if not img_path:
|
|
return
|
|
return config['api']['base.url'] + Core().poster_sizes(img_size) + img_path
|
|
|
|
|
|
class Crew:
|
|
def __init__(self, c):
|
|
self.crew = c
|
|
|
|
def get_id(self):
|
|
return self.crew["id"]
|
|
|
|
def get_department(self):
|
|
return self.crew["department"]
|
|
|
|
def get_job(self):
|
|
return self.crew["job"]
|
|
|
|
def get_original_title(self):
|
|
return self.crew["original_title"]
|
|
|
|
def get_title(self):
|
|
return self.crew["title"]
|
|
|
|
def get_release_date(self):
|
|
return self.crew["release_date"]
|
|
|
|
# Sizes = s->w92 m->w185 l->w500 o->original(default)
|
|
def get_poster(self, img_size="o"):
|
|
img_path = self.crew.get("poster_path")
|
|
if not img_path:
|
|
return
|
|
return config['api']['base.url'] + Core().poster_sizes(img_size) + img_path
|