def dag_stop(): data = request_data() provider = DagProvider(_write_session) id = int(data['id']) dag = provider.by_id(id, joined_load=['tasks']) for t in dag.tasks: celery_tasks.stop(logger, _write_session, t, dag) return {'dag': provider.get({'id': id})['data'][0]}
def dag_stop(): data = request_data() provider = TaskProvider(_write_session) id = int(data['id']) tasks = provider.by_dag(id) supervisor.stop_tasks(tasks) dag_provider = DagProvider(_write_session) return {'dag': dag_provider.get({'id': id})['data'][0]}
def dags(): data = request_data() options = PaginatorOptions(**data['paginator']) provider = DagProvider(_read_session) res = provider.get(data, options) return res