def lambda_handler()

in functions/stop_model/app.py [0:0]

• 25 lines of code
• 5 McCabe index (conditional complexity)

def lambda_handler(event, context):
    
    rekog_client = boto3.client('rekognition')
    projectversionarn = os.environ['rekog_model_project_version_arn']
    projectarn = os.environ['rekog_model_project_arn']
    running_states = ['STARTING', 'RUNNING']
    projectversionname = projectversionarn.split("/")[3]
    # Check if already running
    # Call Custom Rekog
    try:
        isrunning_response = rekog_client.describe_project_versions(
            ProjectArn=projectarn,
            VersionNames=[projectversionname]
        )
    except Exception as e:
        print(e)
    running_status = isrunning_response['ProjectVersionDescriptions'][0]['Status']
    if running_status in running_states:
        # Stop Model
        try:
            running_status = rekog_client.stop_project_version(
                ProjectVersionArn=projectversionarn
            )
        except Exception as e:
            print(e)
        print('Model Start Status: %s' % running_status)
    else:
        # If not running - Do Nothing
        print('Model Start Status: %s' % running_status)
    return running_status