def parse_place(result, index, flatten):
"""Procesa un resultado de ElasticSearch para modificarlo.
Args:
result (dict): Diccionario con resultado.
index (str): Nombre del índice sobre el cual se realizó la búsqueda.
flatten (bool): Bandera para habilitar que el resultado sea aplanado.
Returns:
dict: Resultado modificado.
"""
result = result['_source']
result = dict(result)
if index == MUNICIPALITIES:
add = {'municipalidad': {'id': result[ID], 'nombre': result[NAME]}}
else:
add = {'departamento': {'id': result[ID], 'nombre': result[NAME]}}
result.update(add)
result.pop(ID)
result.pop(NAME)
if flatten:
get_flatten_result(result)
return result
def parse_entity(result, flatten):
"""Procesa un resultado de ElasticSearch para modificarlo.
Args:
result (dict): Diccionario con resultado.
flatten (bool): Bandera para habilitar que el resultado sea aplanado.
Returns:
dict: Resultado modificado.
"""
entity = result['_source']
if flatten:
get_flatten_result(entity)
return entity