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

bot_server: Add support for running botserver from bots registry.

Browse source
This commit is contained in:
PIG208 2021-07-22 13:04:54 +08:00 • committed by Tim Abbott
parent 745f2cd173
commit 285a946a1f
2 changed files with 43 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

20
zulip_botserver/zulip_botserver/server.py
View file

@ -16,7 +16,7 @@ from werkzeug.exceptions import BadRequest, Unauthorized
from zulip import Client
from zulip_bots import lib
from zulip_bots.finder import import_module_from_source
from zulip_bots.finder import import_module_from_source, import_module_from_zulip_bot_registry
from zulip_botserver.input_parameters import parse_args
@ -122,15 +122,17 @@ def load_lib_modules(available_bots: List[str]) -> Dict[str, ModuleType]:
lib_module = import_module(module_name)
bots_lib_module[bot] = lib_module
except ImportError:
error_message = (
'Error: Bot "{}" doesn\'t exist. Please make sure '
"you have set up the botserverrc file correctly.\n".format(bot)
)
if bot == "api":
error_message += (
"Did you forget to specify the bot you want to run with -b <botname> ?"
_, bots_lib_module[bot] = import_module_from_zulip_bot_registry(bot)
if bots_lib_module[bot] is None:
error_message = (
f'Error: Bot "{bot}" doesn\'t exist. Please make sure '
"you have set up the botserverrc file correctly.\n"
)
sys.exit(error_message)
if bot == "api":
error_message += (
"Did you forget to specify the bot you want to run with -b <botname> ?"
)
sys.exit(error_message)
return bots_lib_module