diff --git a/sevabot/frontend/main.py b/sevabot/frontend/main.py index de2b53a..0a1f64c 100644 --- a/sevabot/frontend/main.py +++ b/sevabot/frontend/main.py @@ -15,6 +15,7 @@ from flask import Flask from flask import render_template from flask import request +from flask import jsonify import plac @@ -141,11 +142,18 @@ def chats_post(): settings = get_settings() shared_secret = request.form.get("secret") + format = request.form.get("format", "html") if shared_secret != settings.SHARED_SECRET: return "Bad shared secret", 403, {"Content-type": "text/plain"} - return render_template('chats.html', chats=chats, shared_secret=shared_secret) + if format == "json": + res = {} + for chat_id, chat in chats: + res.setdefault(chat_id, {}).update({'name': chat.FriendlyName}) + return jsonify(res) + + return render_template('chats.html' % format, chats=chats, shared_secret=shared_secret) @server.route("/chat_message///", methods=['GET'])