from . import ComputeNodeShutdownActor as ShutdownActorBase
class ComputeNodeShutdownActor(ShutdownActorBase):
from . import ComputeNodeShutdownActor as ShutdownActorBase
class ComputeNodeShutdownActor(ShutdownActorBase):
return super(ComputeNodeShutdownActor, self).on_start()
else:
self._nodename = arv_node['hostname']
return super(ComputeNodeShutdownActor, self).on_start()
else:
self._nodename = arv_node['hostname']
@ShutdownActorBase._retry((subprocess.CalledProcessError,))
def cancel_shutdown(self):
@ShutdownActorBase._retry((subprocess.CalledProcessError,))
def cancel_shutdown(self):
return super(ComputeNodeShutdownActor, self).cancel_shutdown()
@ShutdownActorBase._stop_if_window_closed
return super(ComputeNodeShutdownActor, self).cancel_shutdown()
@ShutdownActorBase._stop_if_window_closed
def await_slurm_drain(self):
output = subprocess.check_output(
['sinfo', '--noheader', '-o', '%t', '-n', self._nodename])
def await_slurm_drain(self):
output = subprocess.check_output(
['sinfo', '--noheader', '-o', '%t', '-n', self._nodename])
self._later.shutdown_node()
else:
self._timer.schedule(time.time() + 10,
self._later.shutdown_node()
else:
self._timer.schedule(time.time() + 10,