diff --git a/aws_lambda/aws_lambda.py b/aws_lambda/aws_lambda.py index 4d70115..339cf95 100755 --- a/aws_lambda/aws_lambda.py +++ b/aws_lambda/aws_lambda.py @@ -103,6 +103,12 @@ def invoke(src, alt_event=None, verbose=False): path_to_config_file = os.path.join(src, 'config.yaml') cfg = read(path_to_config_file, loader=yaml.load) + # Load environment variables + envvars = cfg.get('environment') + if envvars: + for k, v in envvars.items(): + os.environ[k] = str(v) + # Load and parse event file. if alt_event: path_to_event_file = os.path.join(src, alt_event) @@ -301,6 +307,7 @@ def create_function(cfg, path_to_zip_file): Handler=cfg.get('handler'), Code={'ZipFile': byte_stream}, Description=cfg.get('description'), + Environment={'Variables': cfg.get('environment')}, Timeout=cfg.get('timeout', 15), MemorySize=cfg.get('memory_size', 512), Publish=True @@ -332,6 +339,7 @@ def update_function(cfg, path_to_zip_file): Role=role, Handler=cfg.get('handler'), Description=cfg.get('description'), + Environment={'Variables': cfg.get('environment')}, Timeout=cfg.get('timeout', 15), MemorySize=cfg.get('memory_size', 512) ) diff --git a/requirements.txt b/requirements.txt index a57c496..c34efd8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ -boto3==1.3.1 -botocore==1.4.32 +boto3==1.4.1 +botocore==1.4.77 click==6.6 docutils==0.12 futures==3.0.5