Source Code for Module ganeti.cmdlib.query

  1  # 
  2  # 
  3   
  4  # Copyright (C) 2006, 2007, 2008, 2009, 2010, 2011, 2012, 2013 Google Inc. 
  5  # All rights reserved. 
  6  # 
  7  # Redistribution and use in source and binary forms, with or without 
  8  # modification, are permitted provided that the following conditions are 
  9  # met: 
 10  # 
 11  # 1. Redistributions of source code must retain the above copyright notice, 
 12  # this list of conditions and the following disclaimer. 
 13  # 
 14  # 2. Redistributions in binary form must reproduce the above copyright 
 15  # notice, this list of conditions and the following disclaimer in the 
 16  # documentation and/or other materials provided with the distribution. 
 17  # 
 18  # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS 
 19  # IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 
 20  # TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 
 21  # PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR 
 22  # CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 
 23  # EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 
 24  # PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 
 25  # PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 
 26  # LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 
 27  # NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 
 28  # SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
 29   
 30   
 31  """Logical units for queries.""" 
 32   
 33  from ganeti import constants 
 34  from ganeti import errors 
 35  from ganeti import query 
 36  from ganeti.cmdlib.base import NoHooksLU 
 37  from ganeti.cmdlib.cluster import ClusterQuery 
 38  from ganeti.cmdlib.misc import ExtStorageQuery 
 39  from ganeti.cmdlib.operating_system import OsQuery 
 40   
 41   
 42  #: Query type implementations 
 43  _QUERY_IMPL = { 
 44    constants.QR_CLUSTER: ClusterQuery, 
 45    constants.QR_OS: OsQuery, 
 46    constants.QR_EXTSTORAGE: ExtStorageQuery, 
 47    } 
 48   
 49  assert set(_QUERY_IMPL.keys()) == constants.QR_VIA_OP 
 50   
 51   
 52 -def _GetQueryImplementation(name): 
 53    """Returns the implementation for a query type. 
 54   
 55    @param name: Query type, must be one of L{constants.QR_VIA_OP} 
 56   
 57    """ 
 58    try: 
 59      return _QUERY_IMPL[name] 
 60    except KeyError: 
 61      raise errors.OpPrereqError("Unknown query resource '%s'" % name, 
 62                                 errors.ECODE_INVAL) 
 63   
 64   
 65 -class LUQuery(NoHooksLU): 
 66    """Query for resources/items of a certain kind. 
 67   
 68    """ 
 69    # pylint: disable=W0142 
 70    REQ_BGL = False 
 71   
 72 -  def CheckArguments(self): 
 73      qcls = _GetQueryImplementation(self.op.what) 
 74   
 75      self.impl = qcls(self.op.qfilter, self.op.fields, self.op.use_locking) 
 76   
 77 -  def ExpandNames(self): 
 78      self.impl.ExpandNames(self) 
 79   
 80 -  def DeclareLocks(self, level): 
 81      self.impl.DeclareLocks(self, level) 
 82   
 83 -  def Exec(self, feedback_fn): 
 84      return self.impl.NewStyleQuery(self) 
 85   
 86   
 87 -class LUQueryFields(NoHooksLU): 
 88    """Query for resources/items of a certain kind. 
 89   
 90    """ 
 91    # pylint: disable=W0142 
 92    REQ_BGL = False 
 93   
 94 -  def CheckArguments(self): 
 95      self.qcls = _GetQueryImplementation(self.op.what) 
 96   
 97 -  def ExpandNames(self): 
 98      self.needed_locks = {} 
 99   
100 -  def Exec(self, feedback_fn): 
101      return query.QueryFields(self.qcls.FIELDS, self.op.fields) 
102