From 5a067c78c7a1251d25c1e2c22903a417f9f984cc Mon Sep 17 00:00:00 2001 From: Rizki Wicaksono Date: Fri, 22 Sep 2017 23:24:14 +0000 Subject: [PATCH] fix exception on invoke when no env vars found in config --- aws_lambda/aws_lambda.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/aws_lambda/aws_lambda.py b/aws_lambda/aws_lambda.py index 836b633..3bf7e57 100755 --- a/aws_lambda/aws_lambda.py +++ b/aws_lambda/aws_lambda.py @@ -144,8 +144,10 @@ def invoke(src, alt_event=None, verbose=False): # Load environment variables from the config file into the actual # environment. - for key, value in cfg.get('environment_variables').items(): - os.environ[key] = value + env_vars = cfg.get('environment_variables') + if env_vars: + for key, value in env_vars.items(): + os.environ[key] = value # Load and parse event file. if alt_event: