def search_label()

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

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

def search_label(label, country = None, language = None):
    if language and language != 'en':
        translated_label = translate(language, label)
        logger.info("Translated label {} ({}) to {} (en).".format(label, language, translated_label))
        label = translated_label

    statement = "SELECT image_id FROM tags WHERE label=:label"
    parameters = [{'name':'label', 'value':{'stringValue': label.lower()}}]
    result = execute_statement(statement, parameters)

    logger.info(result)

    response = []
    
    for record in result["records"]:
        for item in record:
            response.append({
                "id": item["stringValue"]
            })

    return {
        'statusCode': 200,
        'headers': {
            'Content-Type': 'application/json; charset=UTF-8'
        },
        'body': json.dumps(response)
    }