xenofem/python-zulip-api
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

zulip-bot-server: Add error handling and informative error messages.

Browse source
This commit is contained in:
vaibhav 2017-06-13 23:07:13 +05:30 committed by showell
parent c1f3544e33
commit b6ca6e5393
1 changed files with 12 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
flask_bot_server.py
View file

@ -1,4 +1,5 @@
from __future__ import absolute_import from __future__ import absolute_import
from __future__ import print_function
from flask import Flask, request, jsonify from flask import Flask, request, jsonify
import os import os
import sys import sys
@ -37,7 +38,12 @@ def load_lib_modules():
# type: () -> None # type: () -> None
for bot in available_bots: for bot in available_bots:
path = "bots/" + str(bot) + "/" + str(bot) + ".py" path = "bots/" + str(bot) + "/" + str(bot) + ".py"
try:
bots_lib_module[bot] = get_lib_module(path) bots_lib_module[bot] = get_lib_module(path)
except Exception:
print("\n ERROR: Bot \"{}\" doesn't exists. Please make sure you have set up the flaskbotrc "
"file correctly.\n".format(bot))
sys.exit(1)
app = Flask(__name__) app = Flask(__name__)
@ -50,7 +56,11 @@ def handle_bot(bot):
client = Client(email=bots_config[bot]["email"], client = Client(email=bots_config[bot]["email"],
api_key=bots_config[bot]["key"], api_key=bots_config[bot]["key"],
site=bots_config[bot]["site"]) site=bots_config[bot]["site"])
try:
restricted_client = ExternalBotHandler(client) restricted_client = ExternalBotHandler(client)
except SystemExit:
return BadRequest("Cannot fetch user profile for bot {}, make sure you have set up the flaskbotrc "
"file correctly.".format(bot))
message_handler = bots_lib_module[bot].handler_class() message_handler = bots_lib_module[bot].handler_class()
# TODO: Handle stateful bots properly. # TODO: Handle stateful bots properly.