From ec18cbff26c76de58e40a04ba112fc2fb0b71b3d Mon Sep 17 00:00:00 2001
From: derAnfaenger <robhoenig@gmail.com>
Date: Fri, 1 Sep 2017 15:47:10 +0200
Subject: [PATCH] zulip package: Add provision option for scripts.

If enabled in a script, this option will try to install dependencies
in the requirements.txt in the script's directory.
---
 zulip/zulip/__init__.py | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/zulip/zulip/__init__.py b/zulip/zulip/__init__.py
index daf9ae6..06b6183 100644
--- a/zulip/zulip/__init__.py
+++ b/zulip/zulip/__init__.py
@@ -105,7 +105,7 @@ def _default_client():
     # type: () -> str
     return "ZulipPython/" + __version__
 
-def add_default_arguments(parser, patch_error_handling=True):
+def add_default_arguments(parser, patch_error_handling=True, allow_provisioning=False):
     # type: (argparse.ArgumentParser) ->  argparse.ArgumentParser
 
     if patch_error_handling:
@@ -114,6 +114,12 @@ def add_default_arguments(parser, patch_error_handling=True):
             self.exit(2, '{}: error: {}\n'.format(self.prog, message))
         parser.error = types.MethodType(custom_error_handling, parser)
 
+    if allow_provisioning:
+        parser.add_argument('--provision',
+                            action='store_true',
+                            dest="provision",
+                            help="install dependencies for this script (found in requirements.txt)")
+
     group = parser.add_argument_group('Zulip API configuration')
     group.add_argument('--site',
                        dest="zulip_site",
@@ -221,6 +227,20 @@ def generate_option_group(parser, prefix=''):
 
 def init_from_options(options, client=None):
     # type: (Any, Optional[str]) -> Client
+
+    if options.provision:
+        requirements_path = os.path.abspath(os.path.join(sys.path[0], 'requirements.txt'))
+        try:
+            import pip
+        except ImportError:
+            traceback.print_exc()
+            print("Module `pip` is not installed. To install `pip`, follow the instructions here: "
+                  "https://pip.pypa.io/en/stable/installing/")
+            sys.exit(1)
+        if not pip.main(['install', '--upgrade', '--requirement', requirements_path]):
+            print("{color_green}Provisioning successful.{end_color}".format(
+                color_green='\033[92m', end_color='\033[0m'))
+
     if options.zulip_client is not None:
         client = options.zulip_client
     elif client is None: