def write_dataframe_to_csv_on_s3()

in functions/source/loop/loop_lambda.py [0:0]

• 10 lines of code
• 3 McCabe index (conditional complexity)

def write_dataframe_to_csv_on_s3(dataframe, filename, bucket):
    """ Write a dataframe to a CSV on S3 """

    # Create buffer
    csv_buffer = StringIO()
    
    
    for c in dataframe.columns:
        if c[:3].lower() == 'unn':
            dataframe.drop(columns =c, inplace=True)
    

    # Write dataframe to buffer
    dataframe.to_csv(csv_buffer, index=False)

    # Create S3 object
    s3_resource = boto3.resource("s3")

    # Write buffer to S3 object
    s3_resource.Object(
        bucket, f'{filename}').put(
        Body=csv_buffer.getvalue(), ServerSideEncryption="aws:kms")