def get_entity_types()

in source/lambda_handlers/04-NewEntityCheck.py [0:0]

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

def get_entity_types(hrw_updated_custom_entities):
    list_of_hrw_updated_entities = []

    line_count = 0
    for entity in hrw_updated_custom_entities:
        row = str(entity).replace("b'", "").replace("'", "").split(',')
        if line_count != 0 and len(row) == 2:
            list_of_hrw_updated_entities.append(row[1][:-2].upper())
        line_count += 1

    response_object = []
    for i in list(set(list_of_hrw_updated_entities)):
        response_object.append({"Type": i})

    return response_object