50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
|
import flask; from flask import request
|
||
|
|
import traceback
|
||
|
|
|
||
|
|
import ycdl
|
||
|
|
|
||
|
|
from .. import common
|
||
|
|
from .. import jsonify
|
||
|
|
|
||
|
|
site = common.site
|
||
|
|
|
||
|
|
@site.route('/mark_video_state', methods=['POST'])
|
||
|
|
def post_mark_video_state():
|
||
|
|
if 'video_ids' not in request.form or 'state' not in request.form:
|
||
|
|
flask.abort(400)
|
||
|
|
video_ids = request.form['video_ids']
|
||
|
|
state = request.form['state']
|
||
|
|
try:
|
||
|
|
video_ids = video_ids.split(',')
|
||
|
|
for video_id in video_ids:
|
||
|
|
video = common.ycdldb.get_video(video_id)
|
||
|
|
video.mark_state(state, commit=False)
|
||
|
|
common.ycdldb.sql.commit()
|
||
|
|
|
||
|
|
except ycdl.exceptions.NoSuchVideo:
|
||
|
|
common.ycdldb.rollback()
|
||
|
|
flask.abort(404)
|
||
|
|
|
||
|
|
except ycdl.exceptions.InvalidVideoState:
|
||
|
|
common.ycdldb.rollback()
|
||
|
|
flask.abort(400)
|
||
|
|
|
||
|
|
return jsonify.make_json_response({'video_ids': video_ids, 'state': state})
|
||
|
|
|
||
|
|
@site.route('/start_download', methods=['POST'])
|
||
|
|
def post_start_download():
|
||
|
|
if 'video_ids' not in request.form:
|
||
|
|
flask.abort(400)
|
||
|
|
video_ids = request.form['video_ids']
|
||
|
|
try:
|
||
|
|
video_ids = video_ids.split(',')
|
||
|
|
for video_id in video_ids:
|
||
|
|
common.ycdldb.download_video(video_id, commit=False)
|
||
|
|
common.ycdldb.sql.commit()
|
||
|
|
|
||
|
|
except ycdl.ytapi.VideoNotFound:
|
||
|
|
common.ycdldb.rollback()
|
||
|
|
flask.abort(404)
|
||
|
|
|
||
|
|
return jsonify.make_json_response({'video_ids': video_ids, 'state': 'downloaded'})
|