def handler()

in python/image-content-search/src/imageData/main.py [0:0]


def handler(event, context):
    # logger.info(event)

    if "ResourceProperties" in event:
        body = event["ResourceProperties"]
        for k in body: event[k] = body[k]

    if "body" in event:
        body = get_http_params(event["body"])
        for k in body: event[k] = body[k]

    source = event["source"]

    if source == "Cloudformation": # Cloudformation => create schema
        return create_schema()
    elif source == "EventBridge": # Event Bridge => image labels
        image_id = event["detail"]["image_id"]
        labels = event["detail"]["labels"]
        response = insert_new_image(image_id, labels)
        return response
    elif source == "API": #API Gateway => search 
        if "language" in event:
            response = search_label(event["label"], event["country"], event["language"])
        else:
            response = search_label(event["label"])

        return response