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