adding function to upload to s3

slapula · slapula · commit 9ff2c3eb741e · 2017-08-17T21:44:55.000-05:00
diff --git a/aws_lambda/aws_lambda.py b/aws_lambda/aws_lambda.py
@@ -94,6 +94,27 @@ def deploy(src, requirements=False, local_package=None):
     else:
         create_function(cfg, path_to_zip_file)
 
+def upload(src, requirements=False, local_package=None):
+    """Uploads a new function to AWS S3.
+
+    :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 local_package:
+        The path to a local package with should be included in the deploy as
+        well (and/or is not available on PyPi)
+    """
+    # Load and parse the config file.
+    path_to_config_file = os.path.join(src, 'config.yaml')
+    cfg = read(path_to_config_file, loader=yaml.load)
+
+    # Copy all the pip dependencies required to run your code into a temporary
+    # folder then add the handler file in the root of this directory.
+    # Zip the contents of this folder into a single file and output to the dist
+    # directory.
+    path_to_zip_file = build(src, requirements, local_package)
+
+    upload_s3(cfg, path_to_zip_file)
 
 def invoke(src, alt_event=None, verbose=False):
     """Simulates a call to your function.
@@ -432,6 +453,36 @@ def update_function(cfg, path_to_zip_file):
 
     client.update_function_configuration(**kwargs)
 
+def upload_s3(cfg, path_to_zip_file):
+    """Upload a function to AWS S3."""
+
+    print('Uploading your new Lambda function')
+    byte_stream = read(path_to_zip_file, binary_file=True)
+    aws_access_key_id = cfg.get('aws_access_key_id')
+    aws_secret_access_key = cfg.get('aws_secret_access_key')
+
+    account_id = get_account_id(aws_access_key_id, aws_secret_access_key)
+    role = get_role_name(account_id, cfg.get('role', 'lambda_basic_execution'))
+
+    client = get_client('s3', aws_access_key_id, aws_secret_access_key,
+                        cfg.get('region'))
+
+    # Do we prefer development variable over config?
+    buck_name = (
+        os.environ.get('S3_BUCKET_NAME') or cfg.get('bucket_name')
+    )
+    func_name = (
+        os.environ.get('LAMBDA_FUNCTION_NAME') or cfg.get('function_name')
+    )
+    kwargs = {
+        'Bucket': cfg.get('bucket_name'),
+        'Key': cfg.get('s3_key', '{}'.format(func_name)),
+        'Body': {'ZipFile': byte_stream}
+    }
+
+    client.put_object(**kwargs)
+    print('Finished uploading {} to S3 bucket {}'.format(func_name, buck_name))
+
 
 def function_exists(cfg, function_name):
     """Check whether a function exists or not"""
diff --git a/aws_lambda/project_templates/config.yaml b/aws_lambda/project_templates/config.yaml
@@ -3,6 +3,8 @@ region: us-east-1
 function_name: my_lambda_function
 handler: service.handler
 # role: lambda_basic_execution
+# bucket_name: my_s3_bucket
+# s3_key: 'path/to/file.zip'
 description: My first lambda function
 runtime: python2.7
 
