152 lines
5.7 KiB
Python
152 lines
5.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
""" Catalog module """
|
|
|
|
from __future__ import absolute_import, division, unicode_literals
|
|
|
|
import logging
|
|
|
|
from resources.lib import kodiutils
|
|
from resources.lib.kodiutils import TitleItem
|
|
from resources.lib.modules.menu import Menu
|
|
from resources.lib.viervijfzes import CHANNELS
|
|
from resources.lib.viervijfzes.content import ContentApi, UnavailableException
|
|
|
|
_LOGGER = logging.getLogger('catalog')
|
|
|
|
|
|
class Catalog:
|
|
""" Menu code related to the catalog """
|
|
|
|
def __init__(self):
|
|
""" Initialise object """
|
|
self._api = ContentApi()
|
|
self._menu = Menu()
|
|
|
|
def show_catalog(self):
|
|
""" Show all the programs of all channels """
|
|
try:
|
|
items = []
|
|
for channel in list(CHANNELS):
|
|
items.extend(self._api.get_programs(channel))
|
|
except Exception as ex:
|
|
kodiutils.notification(message=str(ex))
|
|
raise
|
|
|
|
listing = [self._menu.generate_titleitem(item) for item in items]
|
|
|
|
# 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_catalog_channel(self, channel):
|
|
""" Show the programs of a specific channel
|
|
:type channel: str
|
|
"""
|
|
try:
|
|
items = self._api.get_programs(channel)
|
|
except Exception as ex:
|
|
kodiutils.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.
|
|
kodiutils.show_listing(listing, 30003, content='tvshows', sort='label')
|
|
|
|
def show_program(self, channel, program_id):
|
|
""" Show a program from the catalog
|
|
:type channel: str
|
|
:type program_id: str
|
|
"""
|
|
try:
|
|
program = self._api.get_program(channel, program_id)
|
|
except UnavailableException:
|
|
kodiutils.ok_dialog(message=kodiutils.localize(30717)) # This program is not available in the catalogue.
|
|
kodiutils.end_of_directory()
|
|
return
|
|
|
|
if not program.episodes:
|
|
kodiutils.ok_dialog(message=kodiutils.localize(30717)) # This program is not available in the catalogue.
|
|
kodiutils.end_of_directory()
|
|
return
|
|
|
|
# Go directly to the season when we have only one season
|
|
if len(program.seasons) == 1:
|
|
self.show_program_season(channel, program_id, program.seasons.values()[0].number)
|
|
return
|
|
|
|
studio = CHANNELS.get(program.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(
|
|
TitleItem(
|
|
title='* %s' % kodiutils.localize(30204), # * All seasons
|
|
path=kodiutils.url_for('show_catalog_program_season', channel=channel, program=program_id, season=-1),
|
|
art_dict={
|
|
'thumb': program.cover,
|
|
'fanart': program.background,
|
|
},
|
|
info_dict={
|
|
'tvshowtitle': program.title,
|
|
'title': kodiutils.localize(30204), # All seasons
|
|
'plot': program.description,
|
|
'set': program.title,
|
|
'studio': studio,
|
|
}
|
|
)
|
|
)
|
|
|
|
# Add the seasons
|
|
for s in list(program.seasons.values()):
|
|
listing.append(
|
|
TitleItem(
|
|
title=s.title, # kodiutils.localize(30205, season=s.number), # Season {season}
|
|
path=kodiutils.url_for('show_catalog_program_season', channel=channel, program=program_id, season=s.number),
|
|
art_dict={
|
|
'thumb': s.cover,
|
|
'fanart': program.background,
|
|
},
|
|
info_dict={
|
|
'tvshowtitle': program.title,
|
|
'title': kodiutils.localize(30205, season=s.number), # Season {season}
|
|
'plot': s.description,
|
|
'set': program.title,
|
|
'studio': studio,
|
|
}
|
|
)
|
|
)
|
|
|
|
# Sort by label. Some programs return seasons unordered.
|
|
kodiutils.show_listing(listing, 30003, content='tvshows', sort=['label'])
|
|
|
|
def show_program_season(self, channel, program_id, season):
|
|
""" Show the episodes of a program from the catalog
|
|
:type channel: str
|
|
:type program_id: str
|
|
:type season: int
|
|
"""
|
|
try:
|
|
program = self._api.get_program(channel, program_id)
|
|
except UnavailableException:
|
|
kodiutils.ok_dialog(message=kodiutils.localize(30717)) # This program is not available in the catalogue.
|
|
kodiutils.end_of_directory()
|
|
return
|
|
|
|
if season == -1:
|
|
# Show all episodes
|
|
episodes = program.episodes
|
|
else:
|
|
# Show the episodes of the season that was selected
|
|
episodes = [e for e in program.episodes if e.season == season]
|
|
|
|
listing = [self._menu.generate_titleitem(episode) for episode in episodes]
|
|
|
|
# Sort by episode number by default. Takes seasons into account.
|
|
kodiutils.show_listing(listing, 30003, content='episodes', sort=['episode', 'duration'])
|