Set the VPC during lambda creation time (nficano#101)

Scot Kronenfeld · Scot Kronenfeld · commit 5e6331d9898a · 2018-02-28T09:57:22.000-06:00
diff --git a/aws_lambda/aws_lambda.py b/aws_lambda/aws_lambda.py
@@ -523,6 +523,10 @@ def create_function(cfg, path_to_zip_file, use_s3=False, s3_file=None):
             'Description': cfg.get('description'),
             'Timeout': cfg.get('timeout', 15),
             'MemorySize': cfg.get('memory_size', 512),
+            'VpcConfig': {
+                'SubnetIds': cfg.get('subnet_ids', []),
+                'SecurityGroupIds': cfg.get('security_group_ids', []),
+            },
             'Publish': True,
         }
     else:
@@ -535,6 +539,10 @@ def create_function(cfg, path_to_zip_file, use_s3=False, s3_file=None):
             'Description': cfg.get('description'),
             'Timeout': cfg.get('timeout', 15),
             'MemorySize': cfg.get('memory_size', 512),
+            'VpcConfig': {
+                'SubnetIds': cfg.get('subnet_ids', []),
+                'SecurityGroupIds': cfg.get('security_group_ids', []),
+            },
             'Publish': True,
         }
 
