Added an option to invoke() that allows you to specify an environment file to be loaded using python-dotenv. This will make it easier to support multiple configuration environments and parameterized environment variables in the config.yaml file

asaolabs · asaolabs · commit 4fd64e9a6ec7 · 2018-05-01T14:50:16.000-04:00
diff --git a/aws_lambda/aws_lambda.py b/aws_lambda/aws_lambda.py
@@ -18,6 +18,7 @@
 import botocore
 import pip
 import yaml
+from dotenv import load_dotenv
 
 from .helpers import archive
 from .helpers import get_environment_variable_value
@@ -183,16 +184,22 @@ def upload(
 
 def invoke(
     src, event_file='event.json',
-    config_file='config.yaml', profile_name=None,
+    config_file='config.yaml', profile_name=None, environment_file=None,
     verbose=False,
 ):
     """Simulates a call to your function.
 
     :param str src:
         The path to your Lambda ready project (folder must contain a valid
         config.yaml and handler module (e.g.: service.py).
-    :param str alt_event:
+    :param str event_file:
         An optional argument to override which event file to use.
+    :param str config_file:
+        An optional argument to override which config file to use
+    :param str profile_name:
+        An optional argument to specify which AWS profile name to use
+    :param str environment_file:
+        An optional argument to specify a .env style enviroment file to load
     :param bool verbose:
         Whether to print out verbose details.
     """
@@ -204,6 +211,10 @@ def invoke(
     if profile_name:
         os.environ['AWS_PROFILE'] = profile_name
 
+    # Load an env file if specified to initialize the environment
+    if environment_file:
+        load_dotenv(environment_file)
+
     # Load environment variables from the config file into the actual
     # environment.
     env_vars = cfg.get('environment_variables')
diff --git a/requirements.txt b/requirements.txt
@@ -5,5 +5,6 @@ docutils==0.12
 jmespath==0.9.0
 pyaml==15.8.2
 python-dateutil==2.5.3
+python-dotenv==0.8.2
 PyYAML==3.11
 six==1.10.0
diff --git a/scripts/lambda b/scripts/lambda
@@ -85,13 +85,19 @@ def build(requirements, local_package, config_file, profile):
     '--profile',
     help='AWS profile to use.',
 )
+@click.option(
+    '--environment-file',
+    default=None,
+    help='.env file used to load environment before invocation'
+)
 @click.option('--verbose', '-v', is_flag=True)
-def invoke(event_file, config_file, profile, verbose):
+def invoke(event_file, config_file, profile, environment_file, verbose):
     aws_lambda.invoke(
         CURRENT_DIR,
         event_file=event_file,
         config_file=config_file,
         profile_name=profile,
+        environment_file=environment_file,
         verbose=verbose,
     )
 
