From b852662836055d681ef645deee0d4cdbfbbe36b9 Mon Sep 17 00:00:00 2001
From: Anton Soroko <anton.soroko@gmail.com>
Date: Sun, 14 Mar 2021 19:35:44 +0300
Subject: [PATCH] add season handling

---
 addon.xml |  4 ++--
 plugin.py | 18 ++++++++++++++++--
 2 files changed, 18 insertions(+), 4 deletions(-)

diff --git a/addon.xml b/addon.xml
index 2e54536..9e83b29 100644
--- a/addon.xml
+++ b/addon.xml
@@ -9,11 +9,11 @@
           <label>My submenu</label>
           <item library="context_play.py">
             <label>32000</label>
-            <visible>StringCompare(ListItem.dbtype,movie) | StringCompare(ListItem.dbtype,episode) | String.IsEqual(ListItem.dbtype,movie) | String.IsEqual(ListItem.dbtype,episode)</visible>
+            <visible>StringCompare(ListItem.dbtype,movie) | StringCompare(ListItem.dbtype,episode) | StringCompare(ListItem.dbtype,season) | String.IsEqual(ListItem.dbtype,movie) | String.IsEqual(ListItem.dbtype,episode) | String.IsEqual(ListItem.dbtype,season)</visible>
           </item>
           <item library="context_download.py">
             <label>32012</label>
-            <visible>StringCompare(ListItem.dbtype,movie) | StringCompare(ListItem.dbtype,episode) | String.IsEqual(ListItem.dbtype,movie) | String.IsEqual(ListItem.dbtype,episode)</visible>
+            <visible>StringCompare(ListItem.dbtype,movie) | StringCompare(ListItem.dbtype,episode) | StringCompare(ListItem.dbtype,season) | String.IsEqual(ListItem.dbtype,movie) | String.IsEqual(ListItem.dbtype,episode) | String.IsEqual(ListItem.dbtype,season)</visible>
           </item>
           <!-- <item library="context_menu.py">
             <label>32001</label>
diff --git a/plugin.py b/plugin.py
index 55664a0..d721143 100644
--- a/plugin.py
+++ b/plugin.py
@@ -66,10 +66,9 @@ def doDownload():
 def getDbId():
     infolabel = xbmc.getInfoLabel('ListItem.Label')
     truelabel = sys.listitem.getLabel()
+
     if infolabel == truelabel and xbmc.getInfoLabel('ListItem.DBID'):
         dbid = xbmc.getInfoLabel('ListItem.DBID')
-    elif 'elementum' in sys.listitem.getfilename():
-        dbid = sys.listitem.getfilename().split('?')[0].rstrip('/').split('/')[-1]
     else:
         if xbmc.getInfoLabel('ListItem.Episode') and xbmc.getInfoLabel('ListItem.TVSHowTitle') and xbmc.getInfoLabel('ListItem.Season'):
             season = int(xbmc.getInfoLabel('ListItem.Season'))
@@ -77,6 +76,17 @@ def getDbId():
 
             dbid = '{} s{:02d}e{:02d}'.format(xbmc.getInfoLabel('ListItem.TVSHowTitle'), season, episode)
             dbid = requests.utils.quote(dbid)
+        elif xbmc.getInfoLabel('ListItem.TVSHowTitle') and xbmc.getInfoLabel('ListItem.Season'):
+            season = int(xbmc.getInfoLabel('ListItem.Season'))
+
+            dbid = '{} s{:02d}'.format(xbmc.getInfoLabel('ListItem.TVSHowTitle'), season)
+            dbid = requests.utils.quote(dbid)
+        elif xbmc.getInfoLabel('ListItem.Title') and xbmc.getInfoLabel('ListItem.Year'):
+            title = xbmc.getInfoLabel('ListItem.Title')
+            year = xbmc.getInfoLabel('ListItem.Year')
+
+            dbid = '{} ({})'.format(title, year)
+            dbid = requests.utils.quote(dbid)
         else:
             dbid = requests.utils.quote(infolabel)
     return dbid
@@ -108,6 +118,10 @@ def getMediaType():
             if 'error' not in response:
                 return 'episode'
 
+            response = getJSONResponse('{"jsonrpc": "2.0", "method": "VideoLibrary.GetSeasonDetails", "params": { "seasonid": %s }, "id": "0"}' % dbid)
+            if 'error' not in response:
+                return 'season'
+
         return None