88 lines
2.7 KiB
Python
Executable file
88 lines
2.7 KiB
Python
Executable file
#!/usr/bin/env python
|
|
|
|
from __future__ import absolute_import
|
|
|
|
import argparse
|
|
import logging
|
|
import os
|
|
import sys
|
|
import glob
|
|
import pip
|
|
|
|
def provision_bot(path_to_bot, force):
|
|
# type: (str, bool) -> None
|
|
req_path = os.path.join(path_to_bot, 'requirements.txt')
|
|
install_path = os.path.join(path_to_bot, 'bot_dependencies')
|
|
|
|
if os.path.isfile(req_path):
|
|
bot_name = os.path.basename(path_to_bot)
|
|
logging.info('Installing dependencies for {}...'.format(bot_name))
|
|
if not os.path.isdir(install_path):
|
|
os.makedirs(install_path)
|
|
|
|
# pip install -r $BASEDIR/requirements.txt -t $BASEDIR/bot_dependencies --quiet
|
|
rcode = pip.main(['install', '-r', req_path, '-t', install_path, '--quiet'])
|
|
|
|
if rcode != 0:
|
|
logging.error('Error. Check output of `pip install` above for details.')
|
|
if not force:
|
|
logging.error('Use --force to try running anyway.')
|
|
sys.exit(rcode) # Use pip's exit code
|
|
else:
|
|
logging.info('Installed dependencies successfully.')
|
|
|
|
sys.path.insert(0, install_path)
|
|
|
|
|
|
def parse_args(available_bots):
|
|
usage = """
|
|
Installs dependencies of bots in the bots/<bot_name>
|
|
directories. Add a requirements.txt file in a bot's folder
|
|
before provisioning.
|
|
|
|
To provision all bots, use:
|
|
./provision.py
|
|
|
|
To provision specific bots, use:
|
|
./provision.py [names of bots]
|
|
Example: ./provision.py helloworld xkcd wikipedia
|
|
"""
|
|
parser = argparse.ArgumentParser(usage=usage)
|
|
|
|
parser.add_argument('bots_to_provision',
|
|
metavar='bots',
|
|
nargs='*',
|
|
default=available_bots,
|
|
help='specific bots to provision (default is all)')
|
|
|
|
parser.add_argument('--force',
|
|
default=False,
|
|
action="store_true",
|
|
help='Continue installation despite pip errors.')
|
|
|
|
parser.add_argument('--quiet', '-q',
|
|
action='store_true',
|
|
default=False,
|
|
help='Turn off logging output.')
|
|
|
|
return parser.parse_args()
|
|
|
|
|
|
def main():
|
|
# type: () -> None
|
|
current_dir = os.path.dirname(os.path.abspath(__file__))
|
|
bots_dir = os.path.join(current_dir, "bots")
|
|
bots_subdirs = map(os.path.abspath, glob.glob(bots_dir + '/*'))
|
|
available_bots = filter(lambda d: os.path.isdir(d), bots_subdirs)
|
|
|
|
options = parse_args(available_bots)
|
|
|
|
if not options.quiet:
|
|
logging.basicConfig(stream=sys.stdout, level=logging.INFO)
|
|
|
|
for bot in options.bots_to_provision:
|
|
provision_bot(os.path.join(bots_dir, bot), options.force)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|