def update_channel_infos(provider,
context,
channel_id_dict,
subscription_id_dict={},
channel_items_dict=None):
channel_ids = list(channel_id_dict.keys())
if len(channel_ids) == 0:
return
resource_manager = provider.get_resource_manager(context)
channel_data = resource_manager.get_channels(channel_ids)
for channel_id in channel_data.keys():
yt_item = channel_data[channel_id]
channel_item = channel_id_dict[channel_id]
snippet = yt_item['snippet']
# title
title = snippet['title']
channel_item.set_name(title)
# image
image = snippet.get('thumbnails', {}).get('medium', {}).get('url', '')
channel_item.set_image(image)
# - update context menu
context_menu = []
# -- unsubscribe from channel
subscription_id = subscription_id_dict.get(channel_id, '')
if subscription_id:
yt_context_menu.append_unsubscribe_from_channel(
context_menu, provider, context, subscription_id)
pass
# -- subscribe to the channel
if provider.is_logged_in(
) and context.get_path() != '/subscriptions/list/':
yt_context_menu.append_subscribe_to_channel(
context_menu, provider, context, channel_id)
pass
channel_item.set_context_menu(context_menu)
# update channel mapping
if channel_items_dict is not None:
if not channel_id in channel_items_dict:
channel_items_dict[channel_id] = []
channel_items_dict[channel_id].append(channel_item)
pass
pass
pass
def update_channel_infos(provider, context, channel_id_dict, subscription_id_dict={}, channel_items_dict=None):
channel_ids = list(channel_id_dict.keys())
if len(channel_ids) == 0:
return
resource_manager = provider.get_resource_manager(context)
channel_data = resource_manager.get_channels(channel_ids)
for channel_id in channel_data.keys():
yt_item = channel_data[channel_id]
channel_item = channel_id_dict[channel_id]
snippet = yt_item['snippet']
# title
title = snippet['title']
channel_item.set_name(title)
# image
image = snippet.get('thumbnails', {}).get('medium', {}).get('url', '')
channel_item.set_image(image)
# - update context menu
context_menu = []
# -- unsubscribe from channel
subscription_id = subscription_id_dict.get(channel_id, '')
if subscription_id:
yt_context_menu.append_unsubscribe_from_channel(context_menu, provider, context, subscription_id)
pass
# -- subscribe to the channel
if provider.is_logged_in() and context.get_path() != '/subscriptions/list/':
yt_context_menu.append_subscribe_to_channel(context_menu, provider, context, channel_id)
pass
channel_item.set_context_menu(context_menu)
# update channel mapping
if channel_items_dict is not None:
if not channel_id in channel_items_dict:
channel_items_dict[channel_id] = []
channel_items_dict[channel_id].append(channel_item)
pass
pass
pass
def _process_list_response(provider, context, json_data):
video_id_dict = {}
channel_item_dict = {}
playlist_item_id_dict = {}
result = []
yt_items = json_data.get('items', [])
if len(yt_items) == 0:
context.log_warning('List of search result is empty')
return result
for yt_item in yt_items:
yt_kind = yt_item.get('kind', '')
if yt_kind == u'youtube#video':
video_id = yt_item['id']
snippet = yt_item['snippet']
title = snippet['title']
image = snippet.get('thumbnails', {}).get('medium', {}).get('url', '')
video_item = items.VideoItem(title,
context.create_uri(['play'], {'video_id': video_id}),
image=image)
video_item.set_fanart(provider.get_fanart(context))
result.append(video_item)
video_id_dict[video_id] = video_item
# map video to channel
channel_id = snippet['channelId']
if not channel_id in channel_item_dict:
channel_item_dict[channel_id] = []
channel_item_dict[channel_id].append(video_item)
pass
elif yt_kind == u'youtube#channel':
channel_id = yt_item['id']
snippet = yt_item['snippet']
title = snippet['title']
image = snippet.get('thumbnails', {}).get('medium', {}).get('url', '')
channel_item = items.DirectoryItem(title,
context.create_uri(['channel', channel_id]),
image=image)
channel_item.set_fanart(provider.get_fanart(context))
# if logged in => provide subscribing to the channel
if provider.is_logged_in():
context_menu = []
yt_context_menu.append_subscribe_to_channel(context_menu, provider, context, channel_id)
channel_item.set_context_menu(context_menu)
pass
result.append(channel_item)
# map channel
if not channel_id in channel_item_dict:
channel_item_dict[channel_id] = []
channel_item_dict[channel_id].append(channel_item)
pass
elif yt_kind == u'youtube#guideCategory':
guide_id = yt_item['id']
snippet = yt_item['snippet']
title = snippet['title']
guide_item = items.DirectoryItem(title,
context.create_uri(['special', 'browse_channels'], {'guide_id': guide_id}))
guide_item.set_fanart(provider.get_fanart(context))
result.append(guide_item)
pass
elif yt_kind == u'youtube#subscription':
snippet = yt_item['snippet']
image = snippet.get('thumbnails', {}).get('high', {}).get('url', '')
channel_id = snippet['resourceId']['channelId']
playlist_item = items.DirectoryItem(snippet['title'],
context.create_uri(['channel', channel_id]),
image=image)
playlist_item.set_fanart(provider.get_fanart(context))
# unsubscribe from a channel
subscription_id = yt_item['id']
context_menu = []
yt_context_menu.append_unsubscribe_from_channel(context_menu, provider, context, subscription_id)
playlist_item.set_context_menu(context_menu)
result.append(playlist_item)
# map playlist to channel
if not channel_id in channel_item_dict:
channel_item_dict[channel_id] = []
channel_item_dict[channel_id].append(playlist_item)
pass
elif yt_kind == u'youtube#playlist':
playlist_id = yt_item['id']
snippet = yt_item['snippet']
title = snippet['title']
image = snippet.get('thumbnails', {}).get('medium', {}).get('url', '')
channel_id = snippet['channelId']
# if the path directs to a playlist of our own, we correct the channel id to 'mine'
if context.get_path() == '/channel/mine/playlists/':
channel_id = 'mine'
pass
playlist_item = items.DirectoryItem(title,
context.create_uri(['channel', channel_id, 'playlist', playlist_id]),
image=image)
playlist_item.set_fanart(provider.get_fanart(context))
# subscribe to the channel via the playlist item
channel_name = snippet.get('channelTitle', '')
if channel_id != 'mine' and provider.is_logged_in():
context_menu = []
yt_context_menu.append_subscribe_to_channel(context_menu, provider, context, channel_id, channel_name)
playlist_item.set_context_menu(context_menu)
pass
result.append(playlist_item)
# map playlist to channel
if not channel_id in channel_item_dict:
channel_item_dict[channel_id] = []
channel_item_dict[channel_id].append(playlist_item)
pass
elif yt_kind == u'youtube#playlistItem':
snippet = yt_item['snippet']
video_id = snippet['resourceId']['videoId']
# store the id of the playlistItem - for deleting this item we need this item
playlist_item_id_dict[video_id] = yt_item['id']
title = snippet['title']
image = snippet.get('thumbnails', {}).get('medium', {}).get('url', '')
video_item = items.VideoItem(title,
context.create_uri(['play'], {'video_id': video_id}),
image=image)
video_item.set_fanart(provider.get_fanart(context))
result.append(video_item)
video_id_dict[video_id] = video_item
# map video to channel
channel_id = snippet['channelId']
if not channel_id in channel_item_dict:
channel_item_dict[channel_id] = []
channel_item_dict[channel_id].append(video_item)
pass
elif yt_kind == 'youtube#searchResult':
yt_kind = yt_item.get('id', {}).get('kind', '')
# video
if yt_kind == 'youtube#video':
video_id = yt_item['id']['videoId']
snippet = yt_item['snippet']
title = snippet['title']
image = snippet.get('thumbnails', {}).get('medium', {}).get('url', '')
video_item = items.VideoItem(title,
context.create_uri(['play'], {'video_id': video_id}),
image=image)
video_item.set_fanart(provider.get_fanart(context))
result.append(video_item)
video_id_dict[video_id] = video_item
# map video to channel
channel_id = snippet['channelId']
if not channel_id in channel_item_dict:
channel_item_dict[channel_id] = []
channel_item_dict[channel_id].append(video_item)
pass
# playlist
elif yt_kind == 'youtube#playlist':
playlist_id = yt_item['id']['playlistId']
snippet = yt_item['snippet']
title = snippet['title']
image = snippet.get('thumbnails', {}).get('medium', {}).get('url', '')
channel_id = snippet['channelId']
channel_name = snippet.get('channelTitle', '')
playlist_item = items.DirectoryItem(title,
context.create_uri(
['channel', channel_id, 'playlist', playlist_id]),
image=image)
playlist_item.set_fanart(provider.get_fanart(context))
if provider.is_logged_in():
context_menu = []
yt_context_menu.append_subscribe_to_channel(context_menu, provider, context, channel_id,
channel_name)
playlist_item.set_context_menu(context_menu)
pass
result.append(playlist_item)
# map playlist to channel
if not channel_id in channel_item_dict:
channel_item_dict[channel_id] = []
channel_item_dict[channel_id].append(playlist_item)
pass
elif yt_kind == 'youtube#channel':
channel_id = yt_item['id']['channelId']
snippet = yt_item['snippet']
title = snippet['title']
image = snippet.get('thumbnails', {}).get('medium', {}).get('url', '')
channel_item = items.DirectoryItem(title,
context.create_uri(['channel', channel_id]),
image=image)
channel_item.set_fanart(provider.get_fanart(context))
# subscribe to the channel
if provider.is_logged_in():
context_menu = []
yt_context_menu.append_subscribe_to_channel(context_menu, provider, context, channel_id)
channel_item.set_context_menu(context_menu)
pass
result.append(channel_item)
# map channel
if not channel_id in channel_item_dict:
channel_item_dict[channel_id] = []
channel_item_dict[channel_id].append(channel_item)
pass
else:
raise kodion.KodimonException("Unknown kind '%s'" % yt_kind)
pass
else:
raise kodion.KodimonException("Unknown kind '%s'" % yt_kind)
pass
utils.update_video_infos(provider, context, video_id_dict, playlist_item_id_dict)
utils.update_channel_infos(provider, context, channel_item_dict)
return result
def _process_list_response(provider, context, json_data):
video_id_dict = {}
channel_id_dict = {}
playlist_item_id_dict = {}
result = []
yt_items = json_data.get('items', [])
if len(yt_items) == 0:
context.log_warning('List of search result is empty')
return result
for yt_item in yt_items:
yt_kind = yt_item.get('kind', '')
if yt_kind == u'youtube#video':
video_id = yt_item['id']
snippet = yt_item['snippet']
title = snippet['title']
image = snippet.get('thumbnails', {}).get('medium',
{}).get('url', '')
video_item = items.VideoItem(title,
context.create_uri(
['play'], {'video_id': video_id}),
image=image)
video_item.set_fanart(provider.get_fanart(context))
result.append(video_item)
video_id_dict[video_id] = video_item
pass
elif yt_kind == u'youtube#channel':
channel_id = yt_item['id']
snippet = yt_item['snippet']
title = snippet['title']
image = snippet.get('thumbnails', {}).get('medium',
{}).get('url', '')
channel_item = items.DirectoryItem(title,
context.create_uri(
['channel', channel_id]),
image=image)
channel_item.set_fanart(provider.get_fanart(context))
# if logged in => provide subscribing to the channel
if provider.is_logged_in():
context_menu = []
yt_context_menu.append_subscribe_to_channel(
context_menu, provider, context, channel_id)
channel_item.set_context_menu(context_menu)
pass
result.append(channel_item)
# map channel
if not channel_id in channel_id_dict:
channel_id_dict[channel_id] = []
channel_id_dict[channel_id].append(channel_item)
pass
elif yt_kind == u'youtube#guideCategory':
guide_id = yt_item['id']
snippet = yt_item['snippet']
title = snippet['title']
guide_item = items.DirectoryItem(
title,
context.create_uri(['special', 'browse_channels'],
{'guide_id': guide_id}))
guide_item.set_fanart(provider.get_fanart(context))
result.append(guide_item)
pass
elif yt_kind == u'youtube#subscription':
snippet = yt_item['snippet']
image = snippet.get('thumbnails', {}).get('high',
{}).get('url', '')
channel_id = snippet['resourceId']['channelId']
playlist_item = items.DirectoryItem(snippet['title'],
context.create_uri(
['channel', channel_id]),
image=image)
playlist_item.set_fanart(provider.get_fanart(context))
# unsubscribe from a channel
subscription_id = yt_item['id']
context_menu = []
yt_context_menu.append_unsubscribe_from_channel(
context_menu, provider, context, subscription_id)
playlist_item.set_context_menu(context_menu)
result.append(playlist_item)
# map playlist to channel
if not channel_id in channel_id_dict:
channel_id_dict[channel_id] = []
channel_id_dict[channel_id].append(playlist_item)
pass
elif yt_kind == u'youtube#playlist':
playlist_id = yt_item['id']
snippet = yt_item['snippet']
title = snippet['title']
image = snippet.get('thumbnails', {}).get('medium',
{}).get('url', '')
channel_id = snippet['channelId']
# if the path directs to a playlist of our own, we correct the channel id to 'mine'
if context.get_path() == '/channel/mine/playlists/':
channel_id = 'mine'
pass
playlist_item = items.DirectoryItem(title,
context.create_uri([
'channel', channel_id,
'playlist', playlist_id
]),
image=image)
playlist_item.set_fanart(provider.get_fanart(context))
channel_name = snippet.get('channelTitle', '')
if provider.is_logged_in():
context_menu = []
if channel_id != 'mine':
# subscribe to the channel via the playlist item
yt_context_menu.append_subscribe_to_channel(
context_menu, provider, context, channel_id,
channel_name)
pass
else:
# remove my playlist
yt_context_menu.append_remove_playlist(
context_menu, provider, context, playlist_id)
# rename playlist
yt_context_menu.append_rename_playlist(
context_menu, provider, context, playlist_id, title)
pass
playlist_item.set_context_menu(context_menu)
pass
result.append(playlist_item)
# map playlist to channel
if not channel_id in channel_id_dict:
channel_id_dict[channel_id] = []
channel_id_dict[channel_id].append(playlist_item)
pass
elif yt_kind == u'youtube#playlistItem':
snippet = yt_item['snippet']
video_id = snippet['resourceId']['videoId']
# store the id of the playlistItem - for deleting this item we need this item
playlist_item_id_dict[video_id] = yt_item['id']
title = snippet['title']
image = snippet.get('thumbnails', {}).get('medium',
{}).get('url', '')
video_item = items.VideoItem(title,
context.create_uri(
['play'], {'video_id': video_id}),
image=image)
video_item.set_fanart(provider.get_fanart(context))
result.append(video_item)
video_id_dict[video_id] = video_item
pass
elif yt_kind == 'youtube#searchResult':
yt_kind = yt_item.get('id', {}).get('kind', '')
# video
if yt_kind == 'youtube#video':
video_id = yt_item['id']['videoId']
snippet = yt_item['snippet']
title = snippet['title']
image = snippet.get('thumbnails', {}).get('medium',
{}).get('url', '')
video_item = items.VideoItem(
title,
context.create_uri(['play'], {'video_id': video_id}),
image=image)
video_item.set_fanart(provider.get_fanart(context))
result.append(video_item)
video_id_dict[video_id] = video_item
pass
# playlist
elif yt_kind == 'youtube#playlist':
playlist_id = yt_item['id']['playlistId']
snippet = yt_item['snippet']
title = snippet['title']
image = snippet.get('thumbnails', {}).get('medium',
{}).get('url', '')
channel_id = snippet['channelId']
# if the path directs to a playlist of our own, we correct the channel id to 'mine'
if context.get_path() == '/channel/mine/playlists/':
channel_id = 'mine'
pass
channel_name = snippet.get('channelTitle', '')
playlist_item = items.DirectoryItem(title,
context.create_uri([
'channel', channel_id,
'playlist', playlist_id
]),
image=image)
playlist_item.set_fanart(provider.get_fanart(context))
if provider.is_logged_in():
context_menu = []
# subscribe to the channel of the playlist
yt_context_menu.append_subscribe_to_channel(
context_menu, provider, context, channel_id,
channel_name)
playlist_item.set_context_menu(context_menu)
pass
result.append(playlist_item)
# map playlist to channel
if not channel_id in channel_id_dict:
channel_id_dict[channel_id] = []
channel_id_dict[channel_id].append(playlist_item)
pass
elif yt_kind == 'youtube#channel':
channel_id = yt_item['id']['channelId']
snippet = yt_item['snippet']
title = snippet['title']
image = snippet.get('thumbnails', {}).get('medium',
{}).get('url', '')
channel_item = items.DirectoryItem(
title,
context.create_uri(['channel', channel_id]),
image=image)
channel_item.set_fanart(provider.get_fanart(context))
# subscribe to the channel
if provider.is_logged_in():
context_menu = []
yt_context_menu.append_subscribe_to_channel(
context_menu, provider, context, channel_id)
channel_item.set_context_menu(context_menu)
pass
result.append(channel_item)
# map channel
if not channel_id in channel_id_dict:
channel_id_dict[channel_id] = []
channel_id_dict[channel_id].append(channel_item)
pass
else:
raise kodion.KodimonException("Unknown kind '%s'" % yt_kind)
pass
else:
raise kodion.KodimonException("Unknown kind '%s'" % yt_kind)
pass
# this will also update the channel_id_dict with the correct channel id for each video.
utils.update_video_infos(provider, context, video_id_dict,
playlist_item_id_dict, channel_id_dict)
utils.update_channel_infos(provider, context, channel_id_dict)
return result