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

zulip_bots/run: Add --provision command-line option.

Browse source
This commit is contained in:
Eeshan Garg 2017-07-23 16:16:42 -02:30 committed by Tim Abbott
parent e85be28119
commit bc8282b980
1 changed files with 14 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
zulip_bots/zulip_bots/run.py
View file

@ -5,6 +5,7 @@ from __future__ import absolute_import
import logging import logging
import argparse import argparse
import sys import sys
import os
from types import ModuleType from types import ModuleType
from importlib import import_module from importlib import import_module
from os.path import basename, splitext from os.path import basename, splitext
@ -12,6 +13,7 @@ from os.path import basename, splitext
import six import six
from zulip_bots.lib import run_message_handler_for_bot from zulip_bots.lib import run_message_handler_for_bot
from zulip_bots.provision import provision_bot
def import_module_from_source(path, name=None): def import_module_from_source(path, name=None):
@ -65,6 +67,10 @@ def parse_args():
parser.add_argument('--force', parser.add_argument('--force',
action='store_true', action='store_true',
help='Try running the bot even if dependencies install fails.') help='Try running the bot even if dependencies install fails.')
parser.add_argument('--provision',
action='store_true',
help='Install dependencies for the bot.')
options = parser.parse_args() options = parser.parse_args()
if not options.name and not options.path_to_bot: if not options.name and not options.path_to_bot:
@ -83,8 +89,15 @@ def main():
options = parse_args() options = parse_args()
if options.path_to_bot: if options.path_to_bot:
if options.provision:
bot_dir = os.path.dirname(os.path.abspath(options.path_to_bot))
provision_bot(bot_dir, options.force)
lib_module = import_module_from_source(options.path_to_bot, name=options.name) lib_module = import_module_from_source(options.path_to_bot, name=options.name)
else: elif options.name:
if options.provision:
bots_parent_dir = os.path.abspath("bots")
bot_dir = os.path.join(bots_parent_dir, options.name)
provision_bot(bot_dir, options.force)
lib_module = import_module('zulip_bots.bots.{bot}.{bot}'.format(bot=options.name)) lib_module = import_module('zulip_bots.bots.{bot}.{bot}'.format(bot=options.name))
if not options.quiet: if not options.quiet: