class Watchdog: _create = make_parent_action("create_watchdog") _delete = make_parent_action("delete_watchdog") def __init__(self, t): self.t = t def __enter__(self): self.wid = Watchdog._create(self.t) def __exit__(self, type, value, traceback): Watchdog._delete(self.wid) set_watchdog_factory(Watchdog) class Scheduler: pause_noexc = staticmethod(make_parent_action("pause")) @host_only def pause(self): if self.pause_noexc(): raise TerminationRequested submit = staticmethod(make_parent_action("scheduler_submit")) delete = staticmethod(make_parent_action("scheduler_delete")) request_termination = staticmethod( make_parent_action("scheduler_request_termination")) get_status = staticmethod(make_parent_action("scheduler_get_status"))
class Watchdog: _create = make_parent_action("create_watchdog", "t") _delete = make_parent_action("delete_watchdog", "wid") def __init__(self, t): self.t = t def __enter__(self): self.wid = Watchdog._create(self.t) def __exit__(self, type, value, traceback): Watchdog._delete(self.wid) set_watchdog_factory(Watchdog) class Scheduler: pause = staticmethod(make_parent_action("pause", "")) submit = staticmethod( make_parent_action("scheduler_submit", "pipeline_name expid priority due_date flush")) cancel = staticmethod(make_parent_action("scheduler_cancel", "rid")) def set_run_info(self, pipeline_name, expid, priority): self.pipeline_name = pipeline_name self.expid = expid self.priority = priority