def format_dynamodb_payload()

in security_hub_correlation_cdk/lambdas/create_ddb_sh_entry/create_ddb_sh_entry.py [0:0]

• 27 lines of code
• 2 McCabe index (conditional complexity)

def format_dynamodb_payload (event):
    logger.info('Formatting Security Hub entry for DynamoDB...')
    SourceUrl = ""
    if "SourceUrl" in event['detail']['findings'][0]:
        # Inspector does not have SourceUrl field
        SourceUrl = event['detail']['findings'][0]['SourceUrl']        
    ddbpayload = {
        "SchemaVersion": event['detail']['findings'][0]['SchemaVersion'],
        "Title": event['detail']['findings'][0]['Title'],
        "AwsAccountId": event['detail']['findings'][0]['AwsAccountId'],
        "CreatedAt": event['detail']['findings'][0]['CreatedAt'],
        "UpdatedAt": event['detail']['findings'][0]['UpdatedAt'],
        "Description": event['detail']['findings'][0]['Description'],
        "Types": event['detail']['findings'][0]['FindingProviderFields']['Types'][0],
        "SourceUrl": SourceUrl,
        "ProductArn": event['detail']['findings'][0]['ProductArn'],
        "ProductName": event['detail']['findings'][0]['ProductName'],
        "GeneratorId": event['detail']['findings'][0]['GeneratorId'],
        "Id": event['detail']['findings'][0]['Id'],
        "Region": event['detail']['findings'][0]['Region'],
        "CompanyName": event['detail']['findings'][0]['CompanyName'],
        "ResourceType": event['detail']['findings'][0]['Resources'][0]['Type'],
        "ResourceRegion": event['detail']['findings'][0]['Resources'][0]['Region'],
        "ResourceId": event['detail']['findings'][0]['Resources'][0]['Id'],
        "Severity": event['detail']['findings'][0]['Severity']['Label'],
        "ExpDate": int((datetime.now()+timedelta(days=int(DYNAMODB_TTL))).timestamp())
    }
    return ddbpayload