heckyel/yt-local
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add subtitles support

Browse Source
This commit is contained in:
James Taylor 2018-07-10 01:49:55 -07:00
parent 1134efed49
commit c7edcff93a
2 changed files with 42 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
youtube/watch.py
View File

@ -114,8 +114,7 @@ _formats = {
source_tag_template = Template('''
<source src="$src" type="$type">''')
with open("yt_watch_template.html", "r") as file: with open("yt_watch_template.html", "r") as file:
yt_watch_template = Template(file.read()) yt_watch_template = Template(file.read())
@ -245,7 +244,43 @@ def choose_format(info):
current_best = (format, video_priority_index) current_best = (format, video_priority_index)
return current_best[0] return current_best[0]
subtitles_tag_template = Template('''
<track label="$label" src="$src" kind="subtitles" srclang="$srclang" $default>''')
def subtitles_html(info):
result = ''
default_found = False
for language, formats in info['subtitles'].items():
for format in formats:
if format['ext'] == 'vtt':
if language == "en":
default_found = True
result += subtitles_tag_template.substitute(
src = html.escape('/' + format['url']),
label = html.escape(language),
srclang = html.escape(language),
default = 'default' if language == 'en' else '',
)
break
try:
formats = info['automatic_captions']['en']
except KeyError:
pass
else:
for format in formats:
if format['ext'] == 'vtt':
result += subtitles_tag_template.substitute(
src = html.escape('/' + format['url']),
label = 'en' + ' - Automatic',
srclang = 'en',
default = '' if default_found else 'default',
)
return result
more_comments_template = Template('''<a class="page-button more-comments" href="$url">More comments</a>''') more_comments_template = Template('''<a class="page-button more-comments" href="$url">More comments</a>''')
source_tag_template = Template('''
<source src="$src" type="$type">''')
def get_watch_page(query_string): def get_watch_page(query_string):
id = urllib.parse.parse_qs(query_string)['v'][0] id = urllib.parse.parse_qs(query_string)['v'][0]
tasks = ( tasks = (
@ -292,7 +327,7 @@ def get_watch_page(query_string):
dislikes = (lambda x: '{:,}'.format(x) if x is not None else "")(info["dislike_count"]), dislikes = (lambda x: '{:,}'.format(x) if x is not None else "")(info["dislike_count"]),
video_info = html.escape(json.dumps(video_info)), video_info = html.escape(json.dumps(video_info)),
description = html.escape(info["description"]), description = html.escape(info["description"]),
video_sources = formats_html(info), video_sources = formats_html(info) + subtitles_html(info),
related = related_videos_html, related = related_videos_html,
comments = comments_html, comments = comments_html,
more_comments_button = more_comments_button, more_comments_button = more_comments_button,

6
youtube/youtube.py
View File

@ -1,6 +1,6 @@
import mimetypes import mimetypes
import urllib.parse import urllib.parse
from youtube import local_playlist, watch, search, playlist, channel, comments from youtube import local_playlist, watch, search, playlist, channel, comments, common
YOUTUBE_FILES = ( YOUTUBE_FILES = (
"/shared.css", "/shared.css",
"/opensearch.xml", "/opensearch.xml",
@ -43,7 +43,9 @@ def youtube(env, start_response):
elif path.startswith("/playlists"): elif path.startswith("/playlists"):
start_response('200 OK', (('Content-type','text/html'),) ) start_response('200 OK', (('Content-type','text/html'),) )
return local_playlist.get_playlist_page(path[10:], query_string=query_string).encode() return local_playlist.get_playlist_page(path[10:], query_string=query_string).encode()
elif path.startswith("/api/"):
start_response('200 OK', () )
return common.fetch_url('https://www.youtube.com' + path + ('?' + query_string if query_string else ''))
else: else:
start_response('404 Not Found', () ) start_response('404 Not Found', () )
return b'404 Not Found' return b'404 Not Found'