Move make_json_response to voussoirkit.flasktools.
This commit is contained in:
parent
d2af56c3f0
commit
f2a81720bd
3 changed files with 16 additions and 20 deletions
|
@ -2,6 +2,8 @@ import flask; from flask import request
|
|||
import itertools
|
||||
import traceback
|
||||
|
||||
from voussoirkit import flasktools
|
||||
|
||||
import ycdl
|
||||
|
||||
from .. import common
|
||||
|
@ -13,7 +15,7 @@ site = common.site
|
|||
def get_all_channel_names():
|
||||
all_channels = {channel.id: channel.name for channel in common.ycdldb.get_channels()}
|
||||
response = {'channels': all_channels}
|
||||
return jsonify.make_json_response(response)
|
||||
return flasktools.make_json_response(response)
|
||||
|
||||
@site.route('/channels')
|
||||
def get_channels():
|
||||
|
@ -84,20 +86,20 @@ def post_add_channel():
|
|||
try:
|
||||
channel_id = common.ycdldb.youtube.get_user_id(username=channel_id)
|
||||
except ycdl.ytapi.ChannelNotFound:
|
||||
return jsonify.make_json_response({}, status=404)
|
||||
return flasktools.make_json_response({}, status=404)
|
||||
|
||||
channel = common.ycdldb.add_channel(channel_id, get_videos=True)
|
||||
return jsonify.make_json_response(channel.jsonify())
|
||||
return flasktools.make_json_response(channel.jsonify())
|
||||
|
||||
@site.route('/channel/<channel_id>/delete', methods=['POST'])
|
||||
def post_delete_channel(channel_id):
|
||||
try:
|
||||
channel = common.ycdldb.get_channel(channel_id)
|
||||
except ycdl.exceptions.NoSuchChannel as exc:
|
||||
return jsonify.make_json_response(exc.jsonify(), status=404)
|
||||
return flasktools.make_json_response(exc.jsonify(), status=404)
|
||||
|
||||
channel.delete()
|
||||
return jsonify.make_json_response({})
|
||||
return flasktools.make_json_response({})
|
||||
|
||||
@site.route('/channel/<channel_id>/refresh', methods=['POST'])
|
||||
def post_refresh_channel(channel_id):
|
||||
|
@ -106,17 +108,17 @@ def post_refresh_channel(channel_id):
|
|||
try:
|
||||
channel = common.ycdldb.get_channel(channel_id)
|
||||
except ycdl.exceptions.NoSuchChannel as exc:
|
||||
return jsonify.make_json_response(exc.jsonify(), status=404)
|
||||
return flasktools.make_json_response(exc.jsonify(), status=404)
|
||||
|
||||
channel.refresh(force=force)
|
||||
return jsonify.make_json_response(channel.jsonify())
|
||||
return flasktools.make_json_response(channel.jsonify())
|
||||
|
||||
@site.route('/refresh_all_channels', methods=['POST'])
|
||||
def post_refresh_all_channels():
|
||||
force = request.form.get('force', False)
|
||||
force = ycdl.helpers.truthystring(force)
|
||||
common.ycdldb.refresh_all_channels(force=force, skip_failures=True)
|
||||
return jsonify.make_json_response({})
|
||||
return flasktools.make_json_response({})
|
||||
|
||||
@site.route('/channel/<channel_id>/set_automark', methods=['POST'])
|
||||
def post_set_automark(channel_id):
|
||||
|
@ -128,7 +130,7 @@ def post_set_automark(channel_id):
|
|||
except ycdl.exceptions.InvalidVideoState:
|
||||
flask.abort(400)
|
||||
|
||||
return jsonify.make_json_response({})
|
||||
return flasktools.make_json_response({})
|
||||
|
||||
@site.route('/channel/<channel_id>/set_queuefile_extension', methods=['POST'])
|
||||
def post_set_queuefile_extension(channel_id):
|
||||
|
@ -137,4 +139,4 @@ def post_set_queuefile_extension(channel_id):
|
|||
|
||||
channel.set_queuefile_extension(extension)
|
||||
|
||||
return jsonify.make_json_response({})
|
||||
return flasktools.make_json_response({})
|
||||
|
|
|
@ -1,6 +1,8 @@
|
|||
import flask; from flask import request
|
||||
import traceback
|
||||
|
||||
from voussoirkit import flasktools
|
||||
|
||||
import ycdl
|
||||
|
||||
from .. import common
|
||||
|
@ -29,7 +31,7 @@ def post_mark_video_state():
|
|||
common.ycdldb.rollback()
|
||||
flask.abort(400)
|
||||
|
||||
return jsonify.make_json_response({'video_ids': video_ids, 'state': state})
|
||||
return flasktools.make_json_response({'video_ids': video_ids, 'state': state})
|
||||
|
||||
@site.route('/start_download', methods=['POST'])
|
||||
def post_start_download():
|
||||
|
@ -46,4 +48,4 @@ def post_start_download():
|
|||
common.ycdldb.rollback()
|
||||
flask.abort(404)
|
||||
|
||||
return jsonify.make_json_response({'video_ids': video_ids, 'state': 'downloaded'})
|
||||
return flasktools.make_json_response({'video_ids': video_ids, 'state': 'downloaded'})
|
||||
|
|
|
@ -1,8 +0,0 @@
|
|||
import flask
|
||||
import json
|
||||
|
||||
def make_json_response(j, *args, **kwargs):
|
||||
dumped = json.dumps(j)
|
||||
response = flask.Response(dumped, *args, **kwargs)
|
||||
response.headers['Content-Type'] = 'application/json;charset=utf-8'
|
||||
return response
|
Loading…
Reference in a new issue