Revision e8d47209

b/tools/burnin
323 323
    else:
324 324
      names = []
325 325
    try:
326
      op = opcodes.OpQueryNodes(output_fields=["name", "offline"],
326
      op = opcodes.OpQueryNodes(output_fields=["name", "offline", "drained"],
327 327
                                names=names, use_locking=True)
328 328
      result = self.ExecOp(op)
329 329
    except errors.GenericError, err:
330 330
      err_code, msg = cli.FormatError(err)
331 331
      Err(msg, exit_code=err_code)
332
    self.nodes = [data[0] for data in result if not data[1]]
332
    self.nodes = [data[0] for data in result if not (data[1] or data[2])]
333 333

  
334 334
    result = self.ExecOp(opcodes.OpDiagnoseOS(output_fields=["name", "valid"],
335 335
                                              names=[]))

Also available in: Unified diff