def lambda_handler()

in product-selector-lambda/handler.py [0:0]
59 lines of code
1 McCabe index (conditional complexity)

def lambda_handler(event, context):
    # Need to make sure we don't waste time if the request type is
    # update or delete.  Exit gracefully
    if event['RequestType'] == "Delete":
        logger.info(f'Request Type is Delete; unsupported')
        cfnsend(event, context, 'SUCCESS', responsedata)
        return event
    if event['RequestType'] == "Update":
        logger.info(f'Request Type is Update; unsupported')
        cfnsend(event, context, 'SUCCESS', responsedata)
        return event

    # Check if product provided
    if not 'ProductName' in event['ResourceProperties']:
        logger.info(f'Product Not Provided')
        responsedata['Id'] = 'Product not provided'
        cfnsend(event, context, 'FAILED', responsedata, 'Product name was not provided in lambda invocation')
        return event

    # get product name from CFN calling lambda
    productName = f'sc-{event["ResourceProperties"]["ProductName"]}-product'
    logger.info(f'Searching product: {productName}')
    # search product in AWS Service Catalog
    response = sc.search_products_as_admin(
        SortBy='Title',
        SortOrder='ASCENDING',
        ProductSource='ACCOUNT',
        Filters={
                'FullTextSearch': [
                    productName
                ]
            }
    )

    productList=[]
    #Iterate through all return products
    for p in response['ProductViewDetails']:
        # find the searching product
        if productName in p['ProductViewSummary']['Name']:
            productList.append({'ProductId' : p['ProductViewSummary']['ProductId'], 'CreatedTime' : p['CreatedTime'] })

    # if product found
    if productList:
        # if more the on product get the latest one based on the date/time
        last_product_Id = max(productList,key=lambda item:item['CreatedTime'])
        prodId = last_product_Id['ProductId']
        responsedata['ProductId'] = prodId

        # search artifacts (versions) of product
        prodArtif = sc.list_provisioning_artifacts(
            AcceptLanguage='en',
            ProductId=prodId
        )

        # if version provided in cfn
        if 'Version' in event['ResourceProperties']:
            logger.info(f'Searching version: {event["ResourceProperties"]["Version"]}')
            for a in prodArtif['ProvisioningArtifactDetails']:
                # find artifact that match provided version
                if a['Name'] == event['ResourceProperties']['Version']:
                    responsedata['ArtifactId'] = a['Id']

            if not 'ArtifactId' in responsedata:
                logger.info('Version not found')
                responsedata['ArtifactId'] = 'Version not found'
                cfnsend(event, context, 'FAILED', responsedata, 'Version not found in Service Catalog')
                return event
        # otherwise return last one based on the date/time
        else:
            last_artifact_Id = max(prodArtif['ProvisioningArtifactDetails'],key=lambda item:item['CreatedTime'])
            responsedata['ArtifactId'] = last_artifact_Id['Id']

        logger.info(f'Respond Product Id: {responsedata["ProductId"]}')
        logger.info(f'Respond Artifact Id: {responsedata["ArtifactId"]}')
        # Return product info back to CFN
        cfnsend(event, context, 'SUCCESS', responsedata)
    else:
        # if product not found failed CFN
        logger.info('Product Not Found')
        responsedata['Id']='Product not found'
        cfnsend(event, context, 'FAILED', responsedata, 'Product not found in Service Catalog')

    return event