TheMovieDB.py

import xbmcaddon
import os
import xbmc
import xbmcvfs
import time
import simplejson as json
from Utils import log, prettyprint, read_from_file, save_to_file, CompareWithLibrary
import urllib
from urllib2 import Request, urlopen

moviedb_key = '34142515d9d23817496eeb4ff1d223d0'
__addon__ = xbmcaddon.Addon()
__addonid__ = __addon__.getAddonInfo('id')
__language__ = __addon__.getLocalizedString
Addon_Data_Path = os.path.join(xbmc.translatePath("special://profile/addon_data/%s" % __addonid__).decode("utf-8"))
base_url = ""
poster_size = ""
fanart_size = ""


def HandleTheMovieDBMovieResult(results):
    movies = []
    log("starting HandleTheMovieDBMovieResult")
    for movie in results:
        try:
            newmovie = {'Art(fanart)': base_url + fanart_size + str(movie.get('backdrop_path', "")),
                        'Art(poster)': base_url + poster_size + str(movie.get('poster_path', "")),
                        'Title': movie.get('title', ""),
                        'OriginalTitle': movie.get('original_title', ""),
                        'ID': movie.get('id', ""),
                        'Path': "",
                        'Play': "",
                        'DBID': "",
                        'Rating': movie.get('vote_average', ""),
                        'Votes': movie.get('vote_count', ""),
                        'Year': movie.get('release_date', "")[:4],
                        'Premiered': movie.get('release_date', "")}
            if not str(movie['id']) in str(movies):  # too dirty
                movies.append(newmovie)
        except Exception as e:
            log("Exception:" + e)
            prettyprint(movie)
    movies = CompareWithLibrary(movies)
    return movies


def HandleTheMovieDBTVShowResult(results):
    tvshows = []
    log("starting HandleTheMovieDBTVShowResult")
    for tv in results:
        newtv = {'Art(fanart)': base_url + fanart_size + str(tv.get('backdrop_path', "")),
                 'Art(poster)': base_url + poster_size + str(tv.get('poster_path', "")),
                 'Title': tv.get('name', ""),
                 'OriginalTitle': tv.get('original_name', ""),
                 'ID': tv.get('id', ""),
                 'Path': "",
                 'Play': "",
                 'DBID': "",
                 'Rating': tv.get('vote_average', ""),
                 'Votes': tv.get('vote_count', ""),
                 'Premiered': tv.get('first_air_date', "")}
        if not str(tv['id']) in str(tvshows):  # too dirty
            tvshows.append(newtv)
    tvshows = CompareWithLibrary(tvshows)
    return tvshows


def HandleTheMovieDBListResult(results):
    lists = []
    for list in results["lists"]["results"]:
        newlist = {'Art(poster)': base_url + poster_size + str(list.get('poster_path', "")),
                   'Title': list['name'],
                   'ID': list['id'],
                   'Description': list['description']}
        lists.append(newlist)
    return lists


def HandleTheMovieDBPeopleResult(results):
    people = []
    for person in results:
        newperson = {'adult': person['adult'],
                     'name': person['name'],
                     'also_known_as': person['also_known_as'],
                     'biography': person['biography'],
                     'birthday': person['birthday'],
                     'id': person['id'],
                     'deathday': person['deathday'],
                     'place_of_birth': person['place_of_birth'],
                     'thumb': person['profile_path']}
        people.append(newperson)
    return people


def HandleTheMovieDBCompanyResult(results):
    companies = []
    log("starting HandleLastFMCompanyResult")
    for company in results:
        newcompany = {'parent_company': company['parent_company'],
                      'name': company['name'],
                      'description': company['description'],
                      'headquarters': company['headquarters'],
                      'homepage': company['homepage'],
                      'id': company['id'],
                      'logo_path': company['logo_path']}
        companies.append(newcompany)
    return companies


def SearchforCompany(Company):
  #  Companies = Company.split(" / ")
   # Company = Companies[0]
    import re
    regex = re.compile('\(.+?\)')
    Company = regex.sub('', Company)
    log(Company)
    response = GetMovieDBData("search/company?query=%s&" % urllib.quote_plus(Company), 30)
    try:
        return response["results"][0]["id"]
    except:
        log("could not find Company ID")
        return ""


def GetPersonID(person):
    Persons = person.split(" / ")
    person = Persons[0]
    response = GetMovieDBData("search/person?query=%s&" % urllib.quote_plus(person), 30)
    try:
        return response["results"][0]["id"]
    except:
        log("could not find Person ID")
        return ""


def SearchForSet(setname):
    setname = setname.replace("[", "").replace("]", "").replace("Kollektion", "Collection")
    response = GetMovieDBData("search/collection?query=%s&language=%s&" % (urllib.quote_plus(setname.encode("utf-8")), __addon__.getSetting("LanguageID")), 14)
    try:
        return response["results"][0]["id"]
    except:
        return ""


def GetMovieDBData(url="", cache_days=14):
    from base64 import b64encode
    global base_url
    global poster_size
    global fanart_size
    if not base_url:
        log("fetching base_url and size (MovieDB config)")
        base_url = True
        base_url, poster_size, fanart_size = GetMovieDBConfig()
    filename = b64encode(url).replace("/", "XXXX")
    path = Addon_Data_Path + "/" + filename + ".txt"
    log("trying to load " + path)
    if xbmcvfs.exists(path) and ((time.time() - os.path.getmtime(path)) < (cache_days * 86400)):
        return read_from_file(path)
    else:
        url = "http://api.themoviedb.org/3/" + url + "api_key=%s" % moviedb_key
        log("Downloading MovieDB Data: " + url)
        headers = {"Accept": "application/json"}
        succeed = 0
        while succeed < 3:
            try:
                request = Request(url, headers=headers)
                response = urlopen(request).read()
                log("saved file " + filename)
                response = json.loads(response)
                save_to_file(response, filename, Addon_Data_Path)
                return response
            except:
                log("could not get data from %s" % url)
                xbmc.sleep(1000)
                succeed += 1
        return []


def GetMovieDBConfig():
    response = GetMovieDBData("configuration?", 60)
    if response:
        return (response["images"]["base_url"], response["images"]["poster_sizes"][-2], response["images"]["backdrop_sizes"][-2])
    else:
        return ("", "")


def GetCompanyInfo(Id):
    response = GetMovieDBData("company/%s/movies?append_to_response=movies&" % (Id), 30)
    if response and "results" in response:
        return HandleTheMovieDBMovieResult(response["results"])
    else:
        return []


def millify(n):
    import math
    millnames = ['', 'Thousand', 'Million', 'Billion', 'Trillion']
    millidx = max(0, min(len(millnames) - 1, int(math.floor(math.log10(abs(n)) / 3.0))))
    return '%.0f %s' % (n / 10 ** (3 * millidx), millnames[millidx])


def GetExtendedMovieInfo(Id):
    response = GetMovieDBData("movie/%s?append_to_response=trailers,casts,releases,keywords,similar_movies,lists&language=%s&" % (Id, __addon__.getSetting("LanguageID")), 30)
#    prettyprint(response)
    authors = []
    directors = []
    genres = []
    for item in response['genres']:
        genres.append(item["name"])
    for item in response['casts']['crew']:
        if item["job"] == "Author":
            authors.append(item["name"])
        if item["job"] == "Director":
            directors.append(item["name"])
    Writer = " / ".join(authors)
    Director = " / ".join(directors)
    Genre = " / ".join(genres)
    if len(response['trailers']['youtube']) > 0:
        Trailer = response['trailers']['youtube'][0]['source']
    else:
        Trailer = ""
    if len(response['releases']['countries']) > 0:
        mpaa = response['releases']['countries'][0]['certification']
    else:
        mpaa = ""
    if len(response['production_countries']) > 0:
        Country = response['production_countries'][0]["name"]
    else:
        Country = ""
    if len(response['production_companies']) > 0:
        Studio = response['production_companies'][0]["name"]
    else:
        Studio = ""
    Set = response.get("belongs_to_collection", "")
    if Set:
        SetName = Set.get("name", "")
        SetID = Set.get("id", "")
    else:
        SetName = ""
        SetID = ""
    if 'release_date' in response and response.get('release_date') is not None:
        year = response.get('release_date', "")[:4]
    else:
        year = ""
    BudgetValue = response.get('budget', "")
    if not BudgetValue in [0, ""]:
        Budget = millify(float(BudgetValue))
    else:
        Budget = ""
    newmovie = {'Art(fanart)': base_url + fanart_size + str(response.get('backdrop_path', "")),
                'Fanart': base_url + fanart_size + str(response.get('backdrop_path', "")),
                'Art(poster)': base_url + poster_size + str(response.get('poster_path', "")),
                'Poster': base_url + poster_size + str(response.get('poster_path', "")),
                'Title': response.get('title', ""),
                'Label': response.get('title', ""),
                'Tagline': response.get('tagline', ""),
                'RunningTime': response.get('runtime', ""),
                'Budget': response.get('budget', ""),
                'mpaa': mpaa,
                'Director': Director,
                'Writer': Writer,
                'Budget': Budget,
                'Homepage': response.get('homepage', ""),
                'Set': SetName,
                'SetId': SetID,
                'ID': response.get('id', ""),
                'Plot': response.get('overview', ""),
                'OriginalTitle': response.get('original_title', ""),
                'Genre': Genre,
                'Rating': response.get('vote_average', ""),
                'Play': '',
                'Trailer': 'plugin://plugin.video.youtube/?action=play_video&videoid=%s' % Trailer,
                'ReleaseDate': response.get('release_date', ""),
                'Premiered': response.get('release_date', ""),
                'Country': Country,
                'Studio': Studio,
                'DiscArt': "",
                'VideoResolution': "",
                'AudioChannels': "",
                'VideoCodec': "",
                'VideoAspect': "",
                'Logo': "",
                'DBID': "",
                'Studio': Studio,
                'Year': year}
    newmovie = CompareWithLibrary([newmovie])
    return newmovie[0]


def GetMovieLists(Id):
    response = GetMovieDBData("movie/%s?append_to_response=trailers,casts,releases,keywords,similar_movies,lists&language=%s&" % (Id, __addon__.getSetting("LanguageID")), 30)
    return HandleTheMovieDBListResult(response)


def GetMovieKeywords(Id):
    response = GetMovieDBData("movie/%s?append_to_response=trailers,casts,releases,keywords,similar_movies,lists&language=%s&" % (Id, __addon__.getSetting("LanguageID")), 30)
    keywords = []
    if "keywords" in response:
        for keyword in response["keywords"]["keywords"]:
            newkeyword = {'id': keyword.get('id', ""),
                          'name': keyword['name']}
            keywords.append(newkeyword)
        return keywords
    else:
        log("No Keywords in JSON answer")
        return []


def GetSimilarMovies(Id):
    response = GetMovieDBData("movie/%s?append_to_response=trailers,casts,releases,keywords,similar_movies,lists&language=%s&" % (Id, __addon__.getSetting("LanguageID")), 30)
    if "similar_movies" in response:
        return HandleTheMovieDBMovieResult(response["similar_movies"]["results"])
    else:
        log("No JSON Data available")

def GetMovieDBTVShows(type):
    response = GetMovieDBData("tv/%s?language=%s&" % (type, __addon__.getSetting("LanguageID")), 2)
    if "results" in response:
        return HandleTheMovieDBTVShowResult(response["results"])
    else:
        log("No JSON Data available")

def GetMovieDBMovies(type):
    response = GetMovieDBData("movie/%s?language=%s&" % (type, __addon__.getSetting("LanguageID")), 2)
    if "results" in response:
      return HandleTheMovieDBMovieResult(response["results"])
    else:
        log("No JSON Data available")


def GetSetMovies(Id):
    response = GetMovieDBData("collection/%s?language=%s&" % (Id, __addon__.getSetting("LanguageID")), 14)
    if "parts" in response:
        return HandleTheMovieDBMovieResult(response["parts"])
    else:
        log("No JSON Data available")

def GetDirectorMovies(Id):
    response = GetMovieDBData("person/%s/credits?language=%s&" % (Id, __addon__.getSetting("LanguageID")), 14)
    # return HandleTheMovieDBMovieResult(response["crew"]) + HandleTheMovieDBMovieResult(response["cast"])
    if "crew" in response:
        return HandleTheMovieDBMovieResult(response["crew"])
    else:
        log("No JSON Data available")

def search_movie(medianame, year=''):
    log('TMDB API search criteria: Title[''%s''] | Year[''%s'']' % (medianame, year))
    medianame = urllib.quote_plus(medianame.encode('utf8', 'ignore'))
    response = GetMovieDBData("search/movie?query=%s+%s&language=%s&" % (medianame, year, __addon__.getSetting("LanguageID")), 1)
    tmdb_id = ''
    try:
        if response == "Empty":
            tmdb_id = ''
        else:
            for item in response['results']:
                if item['id']:
                    tmdb_id = item['id']
                    log(tmdb_id)
                    break
    except Exception as e:
        log(e)
    if tmdb_id == '':
        log('TMDB API search found no ID')
    else:
        log('TMDB API search found ID: %s' % tmdb_id)
    return tmdb_id