Merge pull request #120 from brandond/support-environment

Add environment variable support

Author: jarosser06

lambda_uploader/config.py

@@ -29,7 +29,8 @@
 DEFAULT_PARAMS = {u'requirements': [], u'publish': False,
                   u'alias': None, u'alias_description': None,
                   u'ignore': [], u'extra_files': [], u'vpc': None,
-                  u's3_bucket': None, u's3_key': None, u'runtime': 'python2.7'}
+                  u's3_bucket': None, u's3_key': None, u'runtime': 'python2.7',
+                  u'variables': {}}
 
 
 class Config(object):

lambda_uploader/uploader.py

@@ -38,6 +38,7 @@ def __init__(self, config, profile_name):
     returns the package version
     '''
     def upload_existing(self, pkg):
+        environment = {'Variables': self._config.variables}
         self._validate_package_size(pkg.zip_file)
         with open(pkg.zip_file, "rb") as fil:
             zip_file = fil.read()
@@ -69,6 +70,7 @@ def upload_existing(self, pkg):
             Timeout=self._config.timeout,
             MemorySize=self._config.memory,
             VpcConfig=self._vpc_config,
+            Environment=environment,
         )
         LOG.debug("AWS update_function_configuration response: %s"
                   % response)
@@ -90,6 +92,7 @@ def upload_existing(self, pkg):
     returns the package version
     '''
     def upload_new(self, pkg):
+        environment = {'Variables': self._config.variables}
         code = {}
         if self._config.s3_bucket:
             code = {'S3Bucket': self._config.s3_bucket,
@@ -113,6 +116,7 @@ def upload_new(self, pkg):
             MemorySize=self._config.memory,
             Publish=self._config.publish,
             VpcConfig=self._vpc_config,
+            Environment=environment,
         )
         LOG.debug("AWS create_function response: %s" % response)
 

setup.py

@@ -5,7 +5,8 @@
 from setuptools import setup, find_packages
 
 INSTALL_REQUIRES = [
-    'boto3==1.4.0',
+    'boto3>=1.4.1',
+    'botocore>=1.4.76',
     'virtualenv',
 ]