Correctly handle case where video ID is too short

2018-12-20 01:25:36 -08:00 · 2018-12-20 01:25:36 -08:00 · cbdc404703
commit cbdc404703
parent a8d74ba082
1 changed files with 4 additions and 0 deletions
--- a/youtube/youtube.py
+++ b/youtube/youtube.py
@ -26,6 +26,10 @@ def youtube(env, start_response):
            return comments.get_comments_page(query_string).encode()

        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'),) )
            return watch.get_watch_page(query_string).encode()