def getJobResults()

in rekognition-pipeline/lambda/jobresultprocessor/lambda_function.py [0:0]

• 39 lines of code
• 14 McCabe index (conditional complexity)

def getJobResults(api, jobId):

    pages = []

    time.sleep(5)

    client = AwsHelper().getClient('rekognition')
    if(api == "StartLabelDetection"):
        response = client.get_label_detection(JobId=jobId)
    elif (api == "StartTextDetection"):
        response = client.get_text_detection(JobId=jobId)
    elif (api == "StartFaceDetection"):
        response = client.get_face_detection(JobId=jobId)
    elif (api == "StartContentModeration"):
        response = client.get_content_moderation(JobId=jobId)
    elif (api == "StartCelebrityRecognition"):
        response = client.get_celebrity_recognition(JobId=jobId)
    
    pages.append(response)
    print("Resultset received: {}".format(len(pages)))
    nextToken = None
    if('NextToken' in response):
        nextToken = response['NextToken']
        print("Next token: {}".format(nextToken))

    while(nextToken):
        time.sleep(5)
        
        if(api == "StartLabelDetection"):
            response = client.get_label_detection(JobId=jobId, NextToken=nextToken)
        elif (api == "StartTextDetection"):
            response = client.get_text_detection(JobId=jobId, NextToken=nextToken)
        elif (api == "StartFaceDetection"):
            response = client.get_face_detection(JobId=jobId, NextToken=nextToken)
        elif (api == "StartContentModeration"):
            response = client.get_content_moderation(JobId=jobId, NextToken=nextToken)
        elif (api == "StartCelebrityRecognition"):
            response = client.get_celebrity_recognition(JobId=jobId, NextToken=nextToken)

        pages.append(response)
        print("Resultset received: {}".format(len(pages)))
        nextToken = None
        if('NextToken' in response):
            nextToken = response['NextToken']
            print("Next token: {}".format(nextToken))

    return pages