Source Code for Module ganeti.rapi.client_utils

  1  # 
  2  # 
  3   
  4  # Copyright (C) 2010 Google Inc. 
  5  # 
  6  # This program is free software; you can redistribute it and/or modify 
  7  # it under the terms of the GNU General Public License as published by 
  8  # the Free Software Foundation; either version 2 of the License, or 
  9  # (at your option) any later version. 
 10  # 
 11  # This program is distributed in the hope that it will be useful, but 
 12  # WITHOUT ANY WARRANTY; without even the implied warranty of 
 13  # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU 
 14  # General Public License for more details. 
 15  # 
 16  # You should have received a copy of the GNU General Public License 
 17  # along with this program; if not, write to the Free Software 
 18  # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 
 19  # 02110-1301, USA. 
 20   
 21   
 22  """RAPI client utilities. 
 23   
 24  """ 
 25   
 26  from ganeti import constants 
 27  from ganeti import cli 
 28   
 29  from ganeti.rapi import client 
 30   
 31  # Local constant to avoid importing ganeti.http 
 32  HTTP_NOT_FOUND = 404 
 33   
 34   
 35 -class RapiJobPollCb(cli.JobPollCbBase): 
 36 -  def __init__(self, cl): 
 37      """Initializes this class. 
 38   
 39      @param cl: RAPI client instance 
 40   
 41      """ 
 42      cli.JobPollCbBase.__init__(self) 
 43   
 44      self.cl = cl 
 45   
 46 -  def WaitForJobChangeOnce(self, job_id, fields, 
 47                             prev_job_info, prev_log_serial): 
 48      """Waits for changes on a job. 
 49   
 50      """ 
 51      try: 
 52        result = self.cl.WaitForJobChange(job_id, fields, 
 53                                          prev_job_info, prev_log_serial) 
 54      except client.GanetiApiError, err: 
 55        if err.code == HTTP_NOT_FOUND: 
 56          return None 
 57   
 58        raise 
 59   
 60      if result is None: 
 61        return constants.JOB_NOTCHANGED 
 62   
 63      return (result["job_info"], result["log_entries"]) 
 64   
 65 -  def QueryJobs(self, job_ids, fields): 
 66      """Returns the given fields for the selected job IDs. 
 67   
 68      @type job_ids: list of numbers 
 69      @param job_ids: Job IDs 
 70      @type fields: list of strings 
 71      @param fields: Fields 
 72   
 73      """ 
 74      if len(job_ids) != 1: 
 75        raise NotImplementedError("Only one job supported at this time") 
 76   
 77      try: 
 78        result = self.cl.GetJobStatus(job_ids[0]) 
 79      except client.GanetiApiError, err: 
 80        if err.code == HTTP_NOT_FOUND: 
 81          return [None] 
 82   
 83        raise 
 84   
 85      return [[result[name] for name in fields], ] 
 86   
 87   
 88 -def PollJob(rapi_client, job_id, reporter): 
 89    """Function to poll for the result of a job. 
 90   
 91    @param rapi_client: RAPI client instance 
 92    @type job_id: number 
 93    @param job_id: Job ID 
 94    @type reporter: L{cli.JobPollReportCbBase} 
 95    @param reporter: PollJob reporter instance 
 96   
 97    """ 
 98    return cli.GenericPollJob(job_id, RapiJobPollCb(rapi_client), reporter) 
 99