yannic/stacosys
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

remove umbrella

Browse source
This commit is contained in:
Yax 2021-02-08 18:20:13 +01:00
parent 087d2f9442
commit 359294e166
1 changed files with 35 additions and 53 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
stacosys/interface/api.py
View file

@ -3,13 +3,13 @@
import logging import logging
from flask import abort, jsonify, request from flask import abort, jsonify, request
from stacosys.interface import app from stacosys.interface import app
from stacosys.model.comment import Comment from stacosys.model.comment import Comment
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@app.route("/ping", methods=["GET"]) @app.route("/ping", methods=["GET"])
def ping(): def ping():
return "OK" return "OK"
@ -17,13 +17,10 @@ def ping():
@app.route("/comments", methods=["GET"]) @app.route("/comments", methods=["GET"])
def query_comments(): def query_comments():
comments = [] comments = []
try:
token = request.args.get("token", "") token = request.args.get("token", "")
if token != app.config.get("SITE_TOKEN"): if token != app.config.get("SITE_TOKEN"):
abort(401) abort(401)
url = request.args.get("url", "") url = request.args.get("url", "")
logger.info("retrieve comments for url %s" % (url)) logger.info("retrieve comments for url %s" % (url))
@ -42,22 +39,14 @@ def query_comments():
d["site"] = comment.author_site d["site"] = comment.author_site
logger.debug(d) logger.debug(d)
comments.append(d) comments.append(d)
r = jsonify({"data": comments}) return jsonify({"data": comments})
r.status_code = 200
except Exception:
logger.warn("bad request")
r = jsonify({"data": []})
r.status_code = 400
return r
@app.route("/comments/count", methods=["GET"]) @app.route("/comments/count", methods=["GET"])
def get_comments_count(): def get_comments_count():
try:
token = request.args.get("token", "") token = request.args.get("token", "")
if token != app.config.get("SITE_TOKEN"): if token != app.config.get("SITE_TOKEN"):
abort(401) abort(401)
url = request.args.get("url", "") url = request.args.get("url", "")
if url: if url:
count = ( count = (
@ -66,12 +55,5 @@ def get_comments_count():
.count() .count()
) )
else: else:
count = ( count = Comment.select(Comment).where(Comment.publishd.is_null(False)).count()
Comment.select(Comment).where(Comment.published.is_null(False)).count() return jsonify({"count": count})
)
r = jsonify({"count": count})
r.status_code = 200
except Exception:
r = jsonify({"count": 0})
r.status_code = 200
return r