replace flask with sanic

This commit is contained in:
Yax 2017-07-08 20:21:39 +02:00
parent 17c302f999
commit f00db85541
6 changed files with 44 additions and 52 deletions

View file

@ -3,7 +3,7 @@
import logging
import config
from flask import request, jsonify, abort
from sanic import response
from app import app
from app import cache
from app.models.site import Site
@ -15,7 +15,7 @@ logger = logging.getLogger(__name__)
@app.route("/comments", methods=['GET'])
def query_comments():
def query_comments(request):
comments = []
try:
@ -37,18 +37,16 @@ def query_comments():
d['date'] = comment.published.strftime("%Y-%m-%d %H:%M:%S")
logger.debug(d)
comments.append(d)
r = jsonify({'data': comments})
r.status_code = 200
r = response.json({'data': comments})
except:
logger.warn('bad request')
r = jsonify({'data': []})
r.status_code = 400
r = response.json({'data': []}, status=400)
return r
@cache.cached(timeout=300)
#@cache.cached(timeout=300)
@app.route("/comments/count", methods=['GET'])
def get_comments_count():
def get_comments_count(request):
try:
token = request.args.get('token', '')
url = request.args.get('url', '')
@ -56,16 +54,14 @@ def get_comments_count():
(Comment.url == url) &
(Comment.published.is_null(False)) &
(Site.token == token)).count()
r = jsonify({'count': count})
r.status_code = 200
r = response.json({'count': count})
except:
r = jsonify({'count': 0})
r.status_code = 200
r = response.json({'count': 0})
return r
@app.route("/comments", methods=['POST'])
def new_comment():
def new_comment(request):
try:
data = request.get_json()
@ -76,25 +72,25 @@ def new_comment():
site = Site.select().where(Site.token == token).get()
if site is None:
logger.warn('Unknown site %s' % token)
abort(400)
return response.text('BAD_REQUEST', status=400)
# honeypot for spammers
captcha = data.get('captcha', '')
if captcha:
logger.warn('discard spam: data %s' % data)
abort(400)
return response.text('BAD_REQUEST', status=400)
processor.enqueue({'request': 'new_comment', 'data': data})
except:
logger.exception("new comment failure")
abort(400)
return response.text('BAD_REQUEST', status=400)
return "OK"
return response.text('OK')
@app.route("/report", methods=['GET'])
def report():
def report(request):
try:
token = request.args.get('token', '')
@ -102,25 +98,25 @@ def report():
if secret != config.SECRET:
logger.warn('Unauthorized request')
abort(401)
return response.text('UNAUTHORIZED', status=401)
site = Site.select().where(Site.token == token).get()
if site is None:
logger.warn('Unknown site %s' % token)
abort(404)
return response.text('', status=404)
processor.enqueue({'request': 'report', 'data': token})
except:
logger.exception("report failure")
abort(500)
return response.text('ERROR', status=500)
return "OK"
return response.text('OK')
@app.route("/accept", methods=['GET'])
def accept_comment():
def accept_comment(request):
try:
id = request.args.get('comment', '')
@ -128,19 +124,19 @@ def accept_comment():
if secret != config.SECRET:
logger.warn('Unauthorized request')
abort(401)
return response.text('UNAUTHORIZED', status=401)
processor.enqueue({'request': 'late_accept', 'data': id})
except:
logger.exception("accept failure")
abort(500)
return response.text('', status=500)
return "PUBLISHED"
return response.text('PUBLISHED')
@app.route("/reject", methods=['GET'])
def reject_comment():
def reject_comment(request):
try:
id = request.args.get('comment', '')
@ -148,12 +144,12 @@ def reject_comment():
if secret != config.SECRET:
logger.warn('Unauthorized request')
abort(401)
return response.text('UNAUTHORIZED', status=401)
processor.enqueue({'request': 'late_reject', 'data': id})
except:
logger.exception("reject failure")
abort(500)
return response.text('ERROR', status=500)
return "REJECTED"
return response.text('REJECTED')

View file

@ -2,7 +2,7 @@
# -*- coding: utf-8 -*-
import logging
from flask import request, abort
from sanic import response
from app import app
from app.services import processor
@ -10,10 +10,10 @@ logger = logging.getLogger(__name__)
@app.route("/inbox", methods=['POST'])
def new_mail():
def new_mail(request):
try:
data = request.get_json()
data = request.json
logger.debug(data)
processor.enqueue({'request': 'new_mail', 'data': data})

View file

@ -2,7 +2,7 @@
# -*- coding: utf-8 -*-
import logging
from flask import request, abort
from sanic import response
from app import app
from app.services import processor
@ -10,7 +10,7 @@ logger = logging.getLogger(__name__)
@app.route("/unsubscribe", methods=['GET'])
def unsubscribe():
def unsubscribe(request):
try:
data = {
@ -24,6 +24,6 @@ def unsubscribe():
except:
logger.exception("unsubscribe failure")
abort(400)
return response.text('BAD_REQUEST', status=400)
return "OK"
return response.text('OK')