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

Correctly handle case where video ID is too short

Browse Source
This commit is contained in:
James Taylor 2018-12-20 01:25:36 -08:00
parent a8d74ba082
commit cbdc404703
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
youtube/youtube.py
View File

@ -26,6 +26,10 @@ def youtube(env, start_response):
return comments.get_comments_page(query_string).encode() return comments.get_comments_page(query_string).encode()
elif path == "/watch": elif path == "/watch":
video_id = urllib.parse.parse_qs(query_string)['v'][0]
if len(video_id) < 11:
start_response('404 Not Found', ())
return b'Incomplete video id (too short): ' + video_id.encode('ascii')
start_response('200 OK', (('Content-type','text/html'),) ) start_response('200 OK', (('Content-type','text/html'),) )
return watch.get_watch_page(query_string).encode() return watch.get_watch_page(query_string).encode()