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

Better error handling for incorrect watch page urls

Browse Source 
- Correctly handle /embed, /watch with no video ids
- Correctly report error for this and for too short video ids
This commit is contained in:
James Taylor 2019-12-20 20:35:05 -08:00
parent 98fbdf77cb
commit d2ba9be7a7
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
youtube/watch.py
View File

@ -272,12 +272,14 @@ def format_bytes(bytes):
@yt_app.route('/watch') @yt_app.route('/watch')
@yt_app.route('/embed')
@yt_app.route('/embed/<video_id>') @yt_app.route('/embed/<video_id>')
def get_watch_page(video_id=None): def get_watch_page(video_id=None):
video_id = request.args.get('v') or video_id 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: if len(video_id) < 11:
flask.abort(404) return flask.render_template('error.html', error_message='Incomplete video id (too short): ' + video_id), 404
flask.abort(flask.Response('Incomplete video id (too short): ' + video_id))
lc = request.args.get('lc', '') lc = request.args.get('lc', '')
tasks = ( tasks = (