def test_play_video_from_epg(self):
epg = EpgApi()
epg_programs = epg.get_epg('vier', date.today().strftime('%Y-%m-%d'))
epg_program = [program for program in epg_programs if program.video_url][0]
# Lookup the Episode data since we don't have an UUID
api = ContentApi(self._auth.get_token())
episode = api.get_episode(epg_program.channel, epg_program.video_url)
self.assertIsInstance(episode, Episode)
# Get stream based on the Episode's UUID
video = api.get_stream(episode.channel, episode.uuid)
self.assertTrue(video)
def send_epg(): # pylint: disable=no-method-argument
"""Return JSON-EPG formatted information to IPTV Manager"""
epg_api = EpgApi()
try: # Python 3
from urllib.parse import quote
except ImportError: # Python 2
from urllib import quote
today = datetime.today()
results = {}
for key, channel in CHANNELS.items():
iptv_id = channel.get('iptv_id')
if channel.get('iptv_id'):
results[iptv_id] = []
for i in range(-3, 7):
date = today + timedelta(days=i)
epg = epg_api.get_epg(key, date.strftime('%Y-%m-%d'))
results[iptv_id].extend([
dict(start=program.start.isoformat(),
stop=(program.start + timedelta(
seconds=program.duration)).isoformat(),
title=program.program_title,
subtitle=program.episode_title,
description=program.description,
episode='S%sE%s' %
(program.season, program.number)
if program.season and program.number else None,
genre=program.genre,
genre_id=program.genre_id,
image=program.thumb,
stream=kodiutils.url_for(
'play_from_page',
channel=key,
page=quote(program.video_url, safe=''))
if program.video_url else None) for program in epg
if program.duration
])
return dict(version=1, epg=results)
def __init__(self):
""" Initialise object """
self._epg = EpgApi()
class TvGuide:
""" Menu code related to the TV Guide """
def __init__(self):
""" Initialise object """
self._epg = EpgApi()
@staticmethod
def get_dates(date_format):
""" Return a dict of dates.
:rtype: list[dict]
"""
dates = []
today = datetime.today()
# The API provides 7 days in the past and 8 days in the future
for i in range(-7, 8):
day = today + timedelta(days=i)
if i == -1:
dates.append({
'title':
'%s, %s' % (kodiutils.localize(30301),
day.strftime(date_format)), # Yesterday
'key':
'yesterday',
'date':
day.strftime('%d.%m.%Y'),
'highlight':
False,
})
elif i == 0:
dates.append({
'title':
'%s, %s' % (kodiutils.localize(30302),
day.strftime(date_format)), # Today
'key':
'today',
'date':
day.strftime('%d.%m.%Y'),
'highlight':
True,
})
elif i == 1:
dates.append({
'title':
'%s, %s' % (kodiutils.localize(30303),
day.strftime(date_format)), # Tomorrow
'key':
'tomorrow',
'date':
day.strftime('%d.%m.%Y'),
'highlight':
False,
})
else:
dates.append({
'title': day.strftime(date_format),
'key': day.strftime('%Y-%m-%d'),
'date': day.strftime('%d.%m.%Y'),
'highlight': False,
})
return dates
def show_channel(self, channel):
""" Shows the dates in the tv guide
:type channel: str
"""
listing = []
for day in self.get_dates('%A %d %B %Y'):
if day.get('highlight'):
title = '[B]{title}[/B]'.format(title=day.get('title'))
else:
title = day.get('title')
listing.append(
TitleItem(title=title,
path=kodiutils.url_for('show_channel_tvguide_detail',
channel=channel,
date=day.get('key')),
art_dict={
'icon': 'DefaultYear.png',
'thumb': 'DefaultYear.png',
},
info_dict={
'plot': None,
'date': day.get('date'),
}))
kodiutils.show_listing(listing, 30013, content='files', sort=['date'])
def show_detail(self, channel=None, date=None):
""" Shows the programs of a specific date in the tv guide
:type channel: str
:type date: str
"""
try:
programs = self._epg.get_epg(channel=channel, date=date)
except UnavailableException as ex:
kodiutils.notification(message=str(ex))
kodiutils.end_of_directory()
return
listing = []
for program in programs:
if program.program_url:
context_menu = [(
kodiutils.localize(30102), # Go to Program
'Container.Update(%s)' %
kodiutils.url_for('show_catalog_program',
channel=channel,
program=program.program_url))]
else:
context_menu = None
title = '{time} - {title}'.format(
time=program.start.strftime('%H:%M'),
title=program.program_title)
if program.airing:
title = '[B]{title}[/B]'.format(title=title)
if program.video_url:
path = kodiutils.url_for('play_from_page',
channel=channel,
page=quote(program.video_url,
safe=''))
else:
path = kodiutils.url_for('play_catalog', uuid='')
title = '[COLOR gray]' + title + '[/COLOR]'
stream_dict = STREAM_DICT.copy()
stream_dict.update({
'duration': program.duration,
})
info_dict = {
'title': title,
'plot': program.description,
'studio': program.channel,
'duration': program.duration,
'tvshowtitle': program.program_title,
'season': program.season,
'episode': program.number,
'mediatype': 'episode',
}
listing.append(
TitleItem(title=title,
path=path,
art_dict={
'thumb': program.thumb,
},
info_dict=info_dict,
stream_dict=stream_dict,
context_menu=context_menu,
is_playable=True))
kodiutils.show_listing(listing,
30013,
content='episodes',
sort=['unsorted'])
def play_epg_datetime(self, channel, timestamp):
""" Play a program based on the channel and the timestamp when it was aired
:type channel: str
:type timestamp: str
"""
broadcast = self._epg.get_broadcast(channel, timestamp)
if not broadcast:
kodiutils.ok_dialog(message=kodiutils.localize(
30713)) # The requested video was not found in the guide.
kodiutils.end_of_directory()
return
if not broadcast.video_url:
kodiutils.ok_dialog(
message=kodiutils.localize(30712)
) # The video is unavailable and can't be played right now.
kodiutils.end_of_directory()
return
Player().play_from_page(broadcast.video_url)
def test_vier_out_of_range(self):
epg = EpgApi()
programs = epg.get_epg('vier', '2020-01-01')
self.assertEqual(programs, [])
def test_unknown_today(self):
epg = EpgApi()
with self.assertRaises(Exception):
epg.get_epg('vtm', date.today().strftime('%Y-%m-%d'))
def test_zes_today(self):
epg = EpgApi()
programs = epg.get_epg('zes', date.today().strftime('%Y-%m-%d'))
self.assertIsInstance(programs, list)
self.assertIsInstance(programs[0], EpgProgram)