Add support for /embed urls

2019-12-20 19:23:15 -08:00 · 2019-12-20 19:23:15 -08:00 · 80de90b1bb
commit 80de90b1bb
parent 310585ae9e
1 changed files with 3 additions and 2 deletions
--- a/youtube/watch.py
+++ b/youtube/watch.py
@ -272,8 +272,9 @@ def format_bytes(bytes):


@yt_app.route('/watch')
-def get_watch_page():
-    video_id = request.args['v']
+@yt_app.route('/embed/<video_id>')
+def get_watch_page(video_id=None):
+    video_id = request.args.get('v') or video_id
    if len(video_id) < 11:
        flask.abort(404)
        flask.abort(flask.Response('Incomplete video id (too short): ' + video_id))