From 1dc369bc3b02350ac2cee26c9cb1db8eaf428008 Mon Sep 17 00:00:00 2001
From: Ronald Schaten <ronald@schatenseite.de>
Date: Thu, 18 Mar 2021 22:11:37 +0100
Subject: [PATCH] find tmdbid by imdbid if necessary

---
 addon.py | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/addon.py b/addon.py
index 1e86a7b..fe6e3b2 100644
--- a/addon.py
+++ b/addon.py
@@ -37,6 +37,22 @@ def files_from_dir(count, location):
         files[i] = location + files[i]
     return files
 
+def get_tmdbid(apikey, imdbid):
+    print("getting tmdbid for imdbid %s" % imdbid)
+    baseurl = 'https://api.themoviedb.org/3/'
+    url = baseurl + 'find/%s?api_key=%s&external_source=imdb_id'
+    url = url % (imdbid, apikey)
+    json_url = urllib.urlopen(url)
+    data = json.loads(json_url.read())
+    print(data)
+    try:
+        tmdbid = data['movie_results'][0]['id']
+        print("tmdbid is %d" % tmdbid)
+    except:
+        tmdbid = 0
+        print("tmdbid could not be found")
+    return tmdbid
+
 def get_recommendations(apikey, movieid, language, choice):
     print("getting %s for %d" % (choice, movieid))
     baseurl = 'https://api.themoviedb.org/3/'
@@ -99,10 +115,13 @@ def conduct_program(program_file, feature):
         elif settings['source'] == 'tmdbtrailer':
             apikey = settings['apikey']
             tmdbid = int(feature['tmdbid'])
+            imdbid = feature['imdbid']
             language = settings['language']
             choice = settings['choice']
             trailertype = settings['type']
             count = settings['count']
+            if not tmdbid:
+                tmdbid = get_tmdbid(apikey, imdbid)
             if tmdbid:
                 movies = get_recommendations(apikey, tmdbid, language, choice)
                 random.shuffle(movies)
@@ -138,11 +157,14 @@ def get_feature(movieid):
     response = json.loads(json_response)
     if not 'tmdb' in response['result']['moviedetails']['uniqueid']:
         response['result']['moviedetails']['uniqueid']['tmdb'] = 0
+    if not 'imdb' in response['result']['moviedetails']['uniqueid']:
+        response['result']['moviedetails']['uniqueid']['tmdb'] = ''
     feature = {
        'label': response['result']['moviedetails']['label'],
        'file': response['result']['moviedetails']['file'],
        'mpaa': response['result']['moviedetails']['mpaa'],
-       'tmdbid': response['result']['moviedetails']['uniqueid']['tmdb']
+       'tmdbid': response['result']['moviedetails']['uniqueid']['tmdb'],
+       'imdbid': response['result']['moviedetails']['uniqueid']['imdb']
     }
     return feature