def get_experiment_by_name(exp_name):
doc = es.search_one_document(e_idx,
es.dict_query(flatten({'name': exp_name})))
if doc:
return doc['_id']
return None
def flatten(self):
return flatten({
'assemble_config': self.assemble_config,
'assemble_task_id': self.assemble_task_id,
'learn_config': self.learn_config,
'up_to_date': self.up_to_date,
})
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 flatten(self):
return flatten({
'assemble_config': self.assemble_config,
'up_to_date': self.up_to_date,
})
def flatten(self):
return flatten({
'experiment_id': self.experiment_id,
'assemble_task_id': self.assemble_task_id,
'learn_task_id': self.learn_task_id,
})
def get_all_worker():
return es.search_documents(w_idx, es.dict_query(flatten({'deleted':
False})),
Worker.from_es_data)
def flatten(self):
return flatten(self.to_dict())
def delete_exp_tasks(exp_id):
return es.delete_documents(
t_idx, es.dict_query(flatten({'experiment_id': exp_id})))
def get_all_task_by_dict(dict):
return es.search_documents(t_idx, es.dict_query(flatten(dict)),
Task.from_es_data)
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