def lambda_handler()

in assets/functions/batch_anomaly_detection/app.py [0:0]

• 11 lines of code
• 1 McCabe index (conditional complexity)

def lambda_handler(event, context):
    ATHENA_OUTPUT_BUCKET = os.environ['Athena_bucket']
    S3_BUCKET = os.environ['Working_bucket']
    DB_SCHEMA = os.environ['Db_schema']

    BATCH_START = event['Batch_start']
    BATCH_END = event['Batch_end']
    DATA_END = event['Data_end']

    connection = connect(s3_staging_dir='s3://{}/'.format(ATHENA_OUTPUT_BUCKET), region_name=REGION)
    result = process_batch(BATCH_START, BATCH_END, DATA_END, DB_SCHEMA, connection)

    result.to_csv('/tmp/anomaly.csv', index=False)
    boto3.Session().resource('s3').Bucket(S3_BUCKET).Object(os.path.join('meteranalytics', 'anomaly/{}/batch_{}_{}.csv'.format(DATA_END,BATCH_START, BATCH_END))).upload_file('/tmp/anomaly.csv')