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() traceback.print_exc() 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'})