diff options
| -rw-r--r-- | youtube/watch.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/youtube/watch.py b/youtube/watch.py index 75ee7e9..7a57873 100644 --- a/youtube/watch.py +++ b/youtube/watch.py @@ -272,12 +272,14 @@ def format_bytes(bytes): @yt_app.route('/watch') +@yt_app.route('/embed') @yt_app.route('/embed/<video_id>') def get_watch_page(video_id=None): video_id = request.args.get('v') or video_id + if not video_id: + return flask.render_template('error.html', error_message='Missing video id'), 404 if len(video_id) < 11: - flask.abort(404) - flask.abort(flask.Response('Incomplete video id (too short): ' + video_id)) + return flask.render_template('error.html', error_message='Incomplete video id (too short): ' + video_id), 404 lc = request.args.get('lc', '') tasks = ( |