class Catalog:
""" Menu code related to the catalog """
def __init__(self):
""" Initialise object """
self._auth = VtmGoAuth(kodiutils.get_setting('username'),
kodiutils.get_setting('password'), 'VTM',
kodiutils.get_setting('profile'),
kodiutils.get_tokens_path())
self._vtm_go = VtmGo(self._auth)
self._menu = Menu()
def show_catalog(self):
""" Show the catalog """
try:
categories = self._vtm_go.get_categories()
except ApiUpdateRequired:
kodiutils.ok_dialog(message=kodiutils.localize(
30705)) # The VTM GO Service has been updated...
return
except Exception as ex: # pylint: disable=broad-except
_LOGGER.error("%s", ex)
kodiutils.ok_dialog(message="%s" % ex)
return
listing = []
for cat in categories:
listing.append(
kodiutils.TitleItem(
title=cat.title,
path=kodiutils.url_for('show_catalog_category',
category=cat.category_id),
info_dict=dict(
plot='[B]{category}[/B]'.format(category=cat.title), ),
))
# Sort categories by default like in VTM GO.
kodiutils.show_listing(listing, 30003, content='files')
def show_catalog_category(self, category=None):
""" Show a category in the catalog
:type category: str
"""
try:
items = self._vtm_go.get_items(category)
except ApiUpdateRequired:
kodiutils.ok_dialog(message=kodiutils.localize(
30705)) # The VTM GO Service has been updated...
return
except Exception as ex: # pylint: disable=broad-except
_LOGGER.error("%s", ex)
kodiutils.ok_dialog(message="%s" % ex)
return
listing = []
for item in items:
listing.append(self._menu.generate_titleitem(item))
# Sort items by label, but don't put folders at the top.
# Used for A-Z listing or when movies and episodes are mixed.
kodiutils.show_listing(
listing,
30003,
content='movies' if category == 'films' else 'tvshows',
sort=['label', 'year', 'duration'])
def show_catalog_channel(self, channel):
""" Show a category in the catalog
:type channel: str
"""
try:
items = self._vtm_go.get_items()
except ApiUpdateRequired:
kodiutils.ok_dialog(message=kodiutils.localize(
30705)) # The VTM GO Service has been updated...
return
except Exception as ex: # pylint: disable=broad-except
_LOGGER.error("%s", ex)
kodiutils.ok_dialog(message="%s" % ex)
return
listing = []
for item in items:
if item.channel == channel:
listing.append(self._menu.generate_titleitem(item))
# Sort items by label, but don't put folders at the top.
# Used for A-Z listing or when movies and episodes are mixed.
kodiutils.show_listing(listing, 30003, content='tvshows', sort='label')
def show_program(self, program):
""" Show a program from the catalog
:type program: str
"""
try:
program_obj = self._vtm_go.get_program(
program, cache=CACHE_PREVENT
) # Use CACHE_PREVENT since we want fresh data
except UnavailableException:
kodiutils.ok_dialog(
message=kodiutils.localize(30717)
) # This program is not available in the VTM GO catalogue.
kodiutils.end_of_directory()
return
# Go directly to the season when we have only one season
if len(program_obj.seasons) == 1:
self.show_program_season(
program,
list(program_obj.seasons.values())[0].number)
return
studio = CHANNELS.get(program_obj.channel, {}).get('studio_icon')
listing = []
# Add an '* All seasons' entry when configured in Kodi
if kodiutils.get_global_setting('videolibrary.showallitems') is True:
listing.append(
kodiutils.TitleItem(
title='* %s' % kodiutils.localize(30204), # * All seasons
path=kodiutils.url_for('show_catalog_program_season',
program=program,
season=-1),
art_dict=dict(
thumb=program_obj.cover,
fanart=program_obj.cover,
),
info_dict=dict(
tvshowtitle=program_obj.name,
title=kodiutils.localize(30204), # All seasons
tagline=program_obj.description,
set=program_obj.name,
studio=studio,
mpaa=', '.join(program_obj.legal)
if hasattr(program_obj, 'legal') and program_obj.legal
else kodiutils.localize(30216), # All ages
),
))
# Add the seasons
for season in list(program_obj.seasons.values()):
listing.append(
kodiutils.TitleItem(
title=kodiutils.localize(
30205, season=season.number), # Season {season}
path=kodiutils.url_for('show_catalog_program_season',
program=program,
season=season.number),
art_dict=dict(
thumb=season.cover,
fanart=program_obj.cover,
),
info_dict=dict(
tvshowtitle=program_obj.name,
title=kodiutils.localize(
30205, season=season.number), # Season {season}
tagline=program_obj.description,
set=program_obj.name,
studio=studio,
mpaa=', '.join(program_obj.legal)
if hasattr(program_obj, 'legal') and program_obj.legal
else kodiutils.localize(30216), # All ages
),
))
# Sort by label. Some programs return seasons unordered.
kodiutils.show_listing(listing,
30003,
content='tvshows',
sort=['label'])
def show_program_season(self, program, season):
""" Show the episodes of a program from the catalog
:type program: str
:type season: int
"""
try:
program_obj = self._vtm_go.get_program(
program
) # Use CACHE_AUTO since the data is just refreshed in show_program
except UnavailableException:
kodiutils.ok_dialog(
message=kodiutils.localize(30717)
) # This program is not available in the VTM GO catalogue.
kodiutils.end_of_directory()
return
if season == -1:
# Show all seasons
seasons = list(program_obj.seasons.values())
else:
# Show the season that was selected
seasons = [program_obj.seasons[season]]
listing = [
self._menu.generate_titleitem(e) for s in seasons
for e in list(s.episodes.values())
]
# Sort by episode number by default. Takes seasons into account.
kodiutils.show_listing(listing,
30003,
content='episodes',
sort=['episode', 'duration'])
def show_recommendations(self, storefront):
""" Show the recommendations.
:type storefront: str
"""
try:
recommendations = self._vtm_go.get_recommendations(storefront)
except ApiUpdateRequired:
kodiutils.ok_dialog(message=kodiutils.localize(
30705)) # The VTM GO Service has been updated...
return
except Exception as ex: # pylint: disable=broad-except
_LOGGER.error("%s", ex)
kodiutils.ok_dialog(message="%s" % ex)
return
listing = []
for cat in recommendations:
listing.append(
kodiutils.TitleItem(
title=cat.title,
path=kodiutils.url_for('show_recommendations_category',
storefront=storefront,
category=cat.category_id),
info_dict=dict(
plot='[B]{category}[/B]'.format(category=cat.title), ),
))
# Sort categories by default like in VTM GO.
kodiutils.show_listing(listing, 30015, content='files')
def show_recommendations_category(self, storefront, category):
""" Show the items in a recommendations category.
:type storefront: str
:type category: str
"""
try:
recommendations = self._vtm_go.get_recommendations(storefront)
except ApiUpdateRequired:
kodiutils.ok_dialog(message=kodiutils.localize(
30705)) # The VTM GO Service has been updated...
return
except Exception as ex: # pylint: disable=broad-except
_LOGGER.error("%s", ex)
kodiutils.ok_dialog(message="%s" % ex)
return
listing = []
for cat in recommendations:
# Only show the requested category
if cat.category_id != category:
continue
for item in cat.content:
listing.append(self._menu.generate_titleitem(item))
# Sort categories by default like in VTM GO.
kodiutils.show_listing(listing, 30015, content='tvshows')
def show_mylist(self):
""" Show the items in "My List" """
try:
mylist = self._vtm_go.get_swimlane('my-list')
except ApiUpdateRequired:
kodiutils.ok_dialog(message=kodiutils.localize(
30705)) # The VTM GO Service has been updated...
return
except Exception as ex: # pylint: disable=broad-except
_LOGGER.error("%s", ex)
kodiutils.ok_dialog(message="%s" % ex)
return
listing = []
for item in mylist:
item.my_list = True
listing.append(self._menu.generate_titleitem(item))
# Sort categories by default like in VTM GO.
kodiutils.show_listing(listing, 30017, content='tvshows')
def mylist_add(self, video_type, content_id):
""" Add an item to "My List"
:type video_type: str
:type content_id: str
"""
self._vtm_go.add_mylist(video_type, content_id)
kodiutils.end_of_directory()
def mylist_del(self, video_type, content_id):
""" Remove an item from "My List"
:type video_type: str
:type content_id: str
"""
self._vtm_go.del_mylist(video_type, content_id)
kodiutils.end_of_directory()
kodiutils.container_refresh()
def show_continuewatching(self):
""" Show the items in "Continue Watching" """
try:
mylist = self._vtm_go.get_swimlane('continue-watching')
except ApiUpdateRequired:
kodiutils.ok_dialog(message=kodiutils.localize(
30705)) # The VTM GO Service has been updated...
return
except Exception as ex: # pylint: disable=broad-except
_LOGGER.error("%s", ex)
kodiutils.ok_dialog(message="%s" % ex)
return
listing = []
for item in mylist:
titleitem = self._menu.generate_titleitem(item, progress=True)
# Add Program Name to title since this list contains episodes from multiple programs
title = '%s - %s' % (titleitem.info_dict.get('tvshowtitle'),
titleitem.info_dict.get('title'))
titleitem.info_dict['title'] = title
listing.append(titleitem)
# Sort categories by default like in VTM GO.
kodiutils.show_listing(listing,
30019,
content='episodes',
sort='label')
class Catalog:
""" Menu code related to the catalog """
def __init__(self, kodi):
""" Initialise object
:type kodi: KodiWrapper
"""
self._kodi = kodi
self._vtm_go = VtmGo(self._kodi)
self._menu = Menu(self._kodi)
def show_catalog(self):
""" Show the catalog """
try:
categories = self._vtm_go.get_categories()
except Exception as ex:
self._kodi.show_notification(message=str(ex))
raise
listing = []
for cat in categories:
listing.append(
TitleItem(title=cat.title,
path=self._kodi.url_for('show_catalog_category',
kids=self._kodi.kids_mode(),
category=cat.category_id),
info_dict={
'plot':
'[B]{category}[/B]'.format(category=cat.title),
}))
# Sort categories by default like in VTM GO.
self._kodi.show_listing(listing, 30003, content='files')
def show_catalog_category(self, category=None):
""" Show a category in the catalog
:type category: str
"""
try:
items = self._vtm_go.get_items(category)
except Exception as ex:
self._kodi.show_notification(message=str(ex))
raise
listing = []
for item in items:
listing.append(self._menu.generate_titleitem(item))
# Sort items by label, but don't put folders at the top.
# Used for A-Z listing or when movies and episodes are mixed.
self._kodi.show_listing(
listing,
30003,
content='movies' if category == 'films' else 'tvshows',
sort='label')
def show_catalog_channel(self, channel):
""" Show a category in the catalog
:type channel: str
"""
try:
items = self._vtm_go.get_items()
except Exception as ex:
self._kodi.show_notification(message=str(ex))
raise
listing = []
for item in items:
if item.channel == channel:
listing.append(self._menu.generate_titleitem(item))
# Sort items by label, but don't put folders at the top.
# Used for A-Z listing or when movies and episodes are mixed.
self._kodi.show_listing(listing,
30003,
content='tvshows',
sort='label')
def show_program(self, program):
""" Show a program from the catalog
:type program: str
"""
try:
program_obj = self._vtm_go.get_program(program)
except UnavailableException:
self._kodi.show_ok_dialog(
message=self._kodi.localize(30717)
) # This program is not available in the VTM GO catalogue.
self._kodi.end_of_directory()
return
studio = CHANNELS.get(program_obj.channel, {}).get('studio_icon')
listing = []
# Add an '* All seasons' entry when configured in Kodi
if self._kodi.get_global_setting('videolibrary.showallitems') is True:
listing.append(
TitleItem(
title='* %s' % self._kodi.localize(30204), # * All seasons
path=self._kodi.url_for('show_catalog_program_season',
program=program,
season=-1),
art_dict={
'thumb': program_obj.cover,
'fanart': program_obj.cover,
},
info_dict={
'tvshowtitle':
program_obj.name,
'title':
self._kodi.localize(30204), # All seasons
'tagline':
program_obj.description,
'set':
program_obj.name,
'studio':
studio,
'mpaa':
', '.join(program_obj.legal)
if hasattr(program_obj, 'legal') and program_obj.legal
else self._kodi.localize(30216),
}))
# Add the seasons
for s in program_obj.seasons.values():
listing.append(
TitleItem(
title=self._kodi.localize(30205,
season=s.number), # Season X
path=self._kodi.url_for('show_catalog_program_season',
program=program,
season=s.number),
art_dict={
'thumb': s.cover,
'fanart': program_obj.cover,
},
info_dict={
'tvshowtitle':
program_obj.name,
'title':
self._kodi.localize(30205, season=s.number),
'tagline':
program_obj.description,
'set':
program_obj.name,
'studio':
studio,
'mpaa':
', '.join(program_obj.legal)
if hasattr(program_obj, 'legal') and program_obj.legal
else self._kodi.localize(30216),
}))
# Sort by label. Some programs return seasons unordered.
self._kodi.show_listing(listing,
30003,
content='tvshows',
sort='label')
def show_program_season(self, program, season):
""" Show a program from the catalog
:type program: str
:type season: int
"""
try:
program_obj = self._vtm_go.get_program(program)
except UnavailableException:
self._kodi.show_ok_dialog(
message=self._kodi.localize(30717)
) # This program is not available in the VTM GO catalogue.
self._kodi.end_of_directory()
return
if season == -1:
# Show all seasons
seasons = program_obj.seasons.values()
else:
# Show the season that was selected
seasons = [program_obj.seasons[season]]
listing = []
for s in seasons:
for episode in s.episodes.values():
listing.append(self._menu.generate_titleitem(episode))
# Sort by episode number by default. Takes seasons into account.
self._kodi.show_listing(listing,
30003,
content='episodes',
sort='episode')
def show_recommendations(self):
""" Show the recommendations """
try:
recommendations = self._vtm_go.get_recommendations()
except Exception as ex:
self._kodi.show_notification(message=str(ex))
raise
listing = []
for cat in recommendations:
listing.append(
TitleItem(title=cat.title,
path=self._kodi.url_for(
'show_recommendations_category',
kids=self._kodi.kids_mode(),
category=cat.category_id),
info_dict={
'plot':
'[B]{category}[/B]'.format(category=cat.title),
}))
# Sort categories by default like in VTM GO.
self._kodi.show_listing(listing, 30015, content='files')
def show_recommendations_category(self, category):
""" Show the items in a recommendations category
:type category: str
"""
try:
recommendations = self._vtm_go.get_recommendations()
except Exception as ex:
self._kodi.show_notification(message=str(ex))
raise
listing = []
for cat in recommendations:
# Only show the requested category
if cat.category_id != category:
continue
for item in cat.content:
listing.append(self._menu.generate_titleitem(item))
# Sort categories by default like in VTM GO.
self._kodi.show_listing(listing, 30015, content='tvshows')
def show_mylist(self):
""" Show the items in "My List" """
try:
mylist = self._vtm_go.get_swimlane('my-list')
except Exception as ex:
self._kodi.show_notification(message=str(ex))
raise
listing = []
for item in mylist:
item.my_list = True
listing.append(self._menu.generate_titleitem(item))
# Sort categories by default like in VTM GO.
self._kodi.show_listing(listing, 30017, content='tvshows')
def mylist_add(self, video_type, content_id):
""" Add an item to "My List"
:type video_type: str
:type content_id: str
"""
self._vtm_go.add_mylist(video_type, content_id)
self._kodi.end_of_directory()
def mylist_del(self, video_type, content_id):
""" Remove an item from "My List"
:type video_type: str
:type content_id: str
"""
self._vtm_go.del_mylist(video_type, content_id)
self._kodi.end_of_directory()
self._kodi.container_refresh()
def show_continuewatching(self):
""" Show the items in "Continue Watching" """
try:
mylist = self._vtm_go.get_swimlane('continue-watching')
except Exception as ex:
self._kodi.show_notification(message=str(ex))
raise
listing = []
for item in mylist:
titleitem = self._menu.generate_titleitem(item, progress=True)
# Add Program Name to title since this list contains episodes from multiple programs
title = '%s - %s' % (titleitem.info_dict.get('tvshowtitle'),
titleitem.info_dict.get('title'))
titleitem.title = title
titleitem.info_dict['title'] = title
listing.append(titleitem)
# Sort categories by default like in VTM GO.
self._kodi.show_listing(listing,
30019,
content='episodes',
sort='label')
