From 67a33353460bcd9f5e0850e20d2c255b9f600cda Mon Sep 17 00:00:00 2001
From: fatshotty <fat@fatshotty.net>
Date: Mon, 19 Apr 2021 16:20:55 +0200
Subject: [PATCH] get tmdb from imdb or title/year

---
 addon.xml         |  2 +-
 externalsearch.py | 34 ++++++++++++++++++++--------------
 2 files changed, 21 insertions(+), 15 deletions(-)

diff --git a/addon.xml b/addon.xml
index ea492595..8262d7a5 100644
--- a/addon.xml
+++ b/addon.xml
@@ -18,7 +18,7 @@
       </item>
       <item library="externalsearch.py">
         <label>90001</label>
-        <visible>!String.IsEmpty(ListItem.Property(tmdb_id))</visible>
+        <visible>String.IsEqual(ListItem.dbtype,tvshow) | String.IsEqual(ListItem.dbtype,movie)</visible>
       </item>
     </menu>
   </extension>
diff --git a/externalsearch.py b/externalsearch.py
index d6312ca8..154102ff 100644
--- a/externalsearch.py
+++ b/externalsearch.py
@@ -6,6 +6,7 @@ librerias = xbmc.translatePath(os.path.join(config.get_runtime_path(), 'lib'))
 sys.path.insert(0, librerias)
 
 
+from core import tmdb
 from core.item import Item
 
 
@@ -21,30 +22,34 @@ def execute_search():
   # These following lines are commented and keep in the code just as reminder.
   # In future, they could be used to filter the search outcome
 
-  # IMDB: the imdb of the selected ListItem
-  # this field is the Kodi's core field (not TMDB)
-  #
-  # imdb = xbmc.getInfoLabel('ListItem.IMDBNumber')
-
   # ADDON: maybe can we know if the current windows is related to a specific addon?
   # we could skip the ContextMenu if we already are in KOD's window
   #
   # addon = xbmc.getInfoLabel('ListItem.Property(Addon.ID)')
 
-  # YEAR: the year of the selected ListItem
-  # this field is the Kodi's core field
-  #
-  # year = xbmc.getInfoLabel('ListItem.Year')
 
-
-  tmdb = xbmc.getInfoLabel('ListItem.Property(tmdb_id)')
+  tmdbid = xbmc.getInfoLabel('ListItem.Property(tmdb_id)')
   mediatype = xbmc.getInfoLabel('ListItem.DBTYPE')
   title = xbmc.getInfoLabel('ListItem.Title')
+  year = xbmc.getInfoLabel('ListItem.Year')
+  imdb = xbmc.getInfoLabel('ListItem.IMDBNumber')
 
-
-  logstr = "Selected ListItem is: 'TMDB: {}' - 'Title: {}' - 'Type: {}'".format(tmdb, title, mediatype)
+  logstr = "Selected ListItem is: 'IMDB: {}' - TMDB: {}' - 'Title: {}' - 'Year: {}'' - 'Type: {}'".format(imdb, tmdbid, title, year, mediatype)
   logger.info(logstr)
 
+  if not tmdbid and imdb:
+    logger.info('No TMDBid found. Try to get by IMDB')
+    it = Item(contentType= mediatype, infoLabels={'imdb_id' : imdb})
+    tmdb.set_infoLabels(it)
+    tmdbid = it.infoLabels.get('tmdb_id', '')
+
+  if not tmdbid:
+    logger.info('No TMDBid found. Try to get by Title/Year')
+    it = Item(contentTitle= title, contentType= mediatype, infoLabels={'year' : year})
+    tmdb.set_infoLabels(it)
+    tmdbid = it.infoLabels.get('tmdb_id', '')
+
+
   item = Item(
     action="Search",
     channel="globalsearch",
@@ -53,7 +58,8 @@ def execute_search():
     text= title,
     type= mediatype,
     infoLabels= {
-      'tmdb_id': tmdb
+      'tmdb_id': tmdbid,
+      'year': year
     },
     folder= False
   )