270 lines
9.6 KiB
Python
270 lines
9.6 KiB
Python
from youtube.template import Template
|
|
from youtube import local_playlist, yt_data_extract, util
|
|
|
|
import json
|
|
import html
|
|
|
|
|
|
with open('yt_basic_template.html', 'r', encoding='utf-8') as file:
|
|
yt_basic_template = Template(file.read())
|
|
|
|
|
|
|
|
|
|
page_button_template = Template('''<a class="page-button" href="$href">$page</a>''')
|
|
current_page_button_template = Template('''<div class="current-page-button">$page</a>''')
|
|
|
|
medium_playlist_item_template = Template('''
|
|
<div class="medium-item-box">
|
|
<div class="medium-item">
|
|
<a class="playlist-thumbnail-box" href="$url" title="$title">
|
|
<img class="playlist-thumbnail-img" src="$thumbnail">
|
|
<div class="playlist-thumbnail-info">
|
|
<span>$size</span>
|
|
</div>
|
|
</a>
|
|
|
|
<a class="title" href="$url" title="$title">$title</a>
|
|
|
|
<div class="stats">$stats</div>
|
|
</div>
|
|
</div>
|
|
''')
|
|
medium_video_item_template = Template('''
|
|
<div class="medium-item-box">
|
|
<div class="medium-item">
|
|
<a class="video-thumbnail-box" href="$url" title="$title">
|
|
<img class="video-thumbnail-img" src="$thumbnail">
|
|
<span class="video-duration">$duration</span>
|
|
</a>
|
|
|
|
<a class="title" href="$url" title="$title">$title</a>
|
|
|
|
<div class="stats">$stats</div>
|
|
|
|
<span class="description">$description</span>
|
|
<span class="badges">$badges</span>
|
|
</div>
|
|
<input class="item-checkbox" type="checkbox" name="video_info_list" value="$video_info" form="playlist-edit">
|
|
</div>
|
|
''')
|
|
|
|
small_video_item_template = Template('''
|
|
<div class="small-item-box">
|
|
<div class="small-item">
|
|
<a class="video-thumbnail-box" href="$url" title="$title">
|
|
<img class="video-thumbnail-img" src="$thumbnail">
|
|
<span class="video-duration">$duration</span>
|
|
</a>
|
|
<a class="title" href="$url" title="$title">$title</a>
|
|
|
|
<address>$author</address>
|
|
<span class="views">$views</span>
|
|
|
|
</div>
|
|
<input class="item-checkbox" type="checkbox" name="video_info_list" value="$video_info" form="playlist-edit">
|
|
</div>
|
|
''')
|
|
|
|
small_playlist_item_template = Template('''
|
|
<div class="small-item-box">
|
|
<div class="small-item">
|
|
<a class="playlist-thumbnail-box" href="$url" title="$title">
|
|
<img class="playlist-thumbnail-img" src="$thumbnail">
|
|
<div class="playlist-thumbnail-info">
|
|
<span>$size</span>
|
|
</div>
|
|
</a>
|
|
<a class="title" href="$url" title="$title">$title</a>
|
|
|
|
<address>$author</address>
|
|
</div>
|
|
</div>
|
|
''')
|
|
|
|
medium_channel_item_template = Template('''
|
|
<div class="medium-item-box">
|
|
<div class="medium-item">
|
|
<a class="video-thumbnail-box" href="$url" title="$title">
|
|
<img class="video-thumbnail-img" src="$thumbnail">
|
|
<span class="video-duration">$duration</span>
|
|
</a>
|
|
|
|
<a class="title" href="$url">$title</a>
|
|
|
|
<span>$subscriber_count</span>
|
|
<span>$size</span>
|
|
|
|
<span class="description">$description</span>
|
|
</div>
|
|
</div>
|
|
''')
|
|
|
|
|
|
|
|
|
|
|
|
def badges_html(badges):
|
|
return ' | '.join(map(html.escape, badges))
|
|
|
|
|
|
html_transform_dispatch = {
|
|
'title': html.escape,
|
|
'published': html.escape,
|
|
'id': html.escape,
|
|
'description': yt_data_extract.format_text_runs,
|
|
'duration': html.escape,
|
|
'thumbnail': lambda url: html.escape('/' + url.lstrip('/')),
|
|
'size': html.escape,
|
|
'author': html.escape,
|
|
'author_url': lambda url: html.escape(util.URL_ORIGIN + url),
|
|
'views': html.escape,
|
|
'subscriber_count': html.escape,
|
|
'badges': badges_html,
|
|
'playlist_index': html.escape,
|
|
}
|
|
|
|
def get_html_ready(item):
|
|
html_ready = {}
|
|
for key, value in item.items():
|
|
try:
|
|
function = html_transform_dispatch[key]
|
|
except KeyError:
|
|
continue
|
|
html_ready[key] = function(value)
|
|
return html_ready
|
|
|
|
|
|
author_template_url = Template('''<address>By <a href="$author_url">$author</a></address>''')
|
|
author_template = Template('''<address><b>$author</b></address>''')
|
|
stat_templates = (
|
|
Template('''<span class="views">$views</span>'''),
|
|
Template('''<time datetime="$datetime">$published</time>'''),
|
|
)
|
|
def get_stats(html_ready):
|
|
stats = []
|
|
if 'author' in html_ready:
|
|
if 'author_url' in html_ready:
|
|
stats.append(author_template_url.substitute(html_ready))
|
|
else:
|
|
stats.append(author_template.substitute(html_ready))
|
|
for stat in stat_templates:
|
|
try:
|
|
stats.append(stat.strict_substitute(html_ready))
|
|
except KeyError:
|
|
pass
|
|
return ' | '.join(stats)
|
|
|
|
def video_item_html(item, template, html_exclude=set()):
|
|
|
|
video_info = {}
|
|
for key in ('id', 'title', 'author'):
|
|
try:
|
|
video_info[key] = item[key]
|
|
except KeyError:
|
|
video_info[key] = ''
|
|
try:
|
|
video_info['duration'] = item['duration']
|
|
except KeyError:
|
|
video_info['duration'] = 'Live' # livestreams don't have a duration
|
|
|
|
html_ready = get_html_ready(item)
|
|
|
|
html_ready['video_info'] = html.escape(json.dumps(video_info) )
|
|
html_ready['url'] = util.URL_ORIGIN + "/watch?v=" + html_ready['id']
|
|
html_ready['datetime'] = '' #TODO
|
|
|
|
for key in html_exclude:
|
|
del html_ready[key]
|
|
html_ready['stats'] = get_stats(html_ready)
|
|
|
|
return template.substitute(html_ready)
|
|
|
|
|
|
def playlist_item_html(item, template, html_exclude=set()):
|
|
html_ready = get_html_ready(item)
|
|
|
|
html_ready['url'] = util.URL_ORIGIN + "/playlist?list=" + html_ready['id']
|
|
html_ready['datetime'] = '' #TODO
|
|
|
|
for key in html_exclude:
|
|
del html_ready[key]
|
|
html_ready['stats'] = get_stats(html_ready)
|
|
|
|
return template.substitute(html_ready)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
page_button_template = Template('''<a class="page-button" href="$href">$page</a>''')
|
|
current_page_button_template = Template('''<div class="page-button">$page</div>''')
|
|
|
|
def page_buttons_html(current_page, estimated_pages, url, current_query_string):
|
|
if current_page <= 5:
|
|
page_start = 1
|
|
page_end = min(9, estimated_pages)
|
|
else:
|
|
page_start = current_page - 4
|
|
page_end = min(current_page + 4, estimated_pages)
|
|
|
|
result = ""
|
|
for page in range(page_start, page_end+1):
|
|
if page == current_page:
|
|
template = current_page_button_template
|
|
else:
|
|
template = page_button_template
|
|
result += template.substitute(page=page, href = url + "?" + util.update_query_string(current_query_string, {'page': [str(page)]}) )
|
|
return result
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
showing_results_for = Template('''
|
|
<div class="showing-results-for">
|
|
<div>Showing results for <a>$corrected_query</a></div>
|
|
<div>Search instead for <a href="$original_query_url">$original_query</a></div>
|
|
</div>
|
|
''')
|
|
|
|
did_you_mean = Template('''
|
|
<div class="did-you-mean">
|
|
<div>Did you mean <a href="$corrected_query_url">$corrected_query</a></div>
|
|
</div>
|
|
''')
|
|
|
|
def renderer_html(renderer, additional_info={}, current_query_string=''):
|
|
type = list(renderer.keys())[0]
|
|
renderer = renderer[type]
|
|
if type == 'itemSectionRenderer':
|
|
return renderer_html(renderer['contents'][0], additional_info, current_query_string)
|
|
|
|
if type == 'channelRenderer':
|
|
info = yt_data_extract.renderer_info(renderer)
|
|
html_ready = get_html_ready(info)
|
|
html_ready['url'] = util.URL_ORIGIN + "/channel/" + html_ready['id']
|
|
return medium_channel_item_template.substitute(html_ready)
|
|
|
|
if type in ('movieRenderer', 'clarificationRenderer'):
|
|
return ''
|
|
|
|
info = yt_data_extract.renderer_info(renderer)
|
|
info.update(additional_info)
|
|
html_exclude = set(additional_info.keys())
|
|
if type == 'compactVideoRenderer':
|
|
return video_item_html(info, small_video_item_template, html_exclude=html_exclude)
|
|
if type in ('compactPlaylistRenderer', 'compactRadioRenderer', 'compactShowRenderer'):
|
|
return playlist_item_html(info, small_playlist_item_template, html_exclude=html_exclude)
|
|
if type in ('videoRenderer', 'gridVideoRenderer'):
|
|
return video_item_html(info, medium_video_item_template, html_exclude=html_exclude)
|
|
if type in ('playlistRenderer', 'gridPlaylistRenderer', 'radioRenderer', 'gridRadioRenderer', 'gridShowRenderer', 'showRenderer'):
|
|
return playlist_item_html(info, medium_playlist_item_template, html_exclude=html_exclude)
|
|
|
|
#print(renderer)
|
|
#raise NotImplementedError('Unknown renderer type: ' + type)
|
|
return '' |