def search_task_by_order(exp_id):
doc = es.search_one_document(
t_idx,
es.dict_query(flatten({
'experiment_id': exp_id,
'state': "runnable"
}), {'order_in_exp': {
'order': 'asc'
}}))
if doc:
return doc['_id'], Task.from_es_data(doc['_source'])
return None, None
def get_task_by_id(task_id):
doc = es.get_document_by_id(t_idx, task_id)
if doc:
return Task.from_es_data(doc['_source'])
return None
def search_task_by_dict(dict):
doc = es.search_one_document(t_idx, es.dict_query(flatten(dict)))
if doc:
return doc['_id'], Task.from_es_data(doc['_source'])
return None, None
