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.backup import ExportQuery 
 37  from ganeti.cmdlib.base import NoHooksLU 
 38  from ganeti.cmdlib.cluster import ClusterQuery 
 39  from ganeti.cmdlib.group import GroupQuery 
 40  from ganeti.cmdlib.instance_query import InstanceQuery 
 41  from ganeti.cmdlib.misc import ExtStorageQuery 
 42  from ganeti.cmdlib.network import NetworkQuery 
 43  from ganeti.cmdlib.node import NodeQuery 
 44  from ganeti.cmdlib.operating_system import OsQuery 
 45   
 46   
 47  #: Query type implementations 
 48  _QUERY_IMPL = { 
 49    constants.QR_CLUSTER: ClusterQuery, 
 50    constants.QR_INSTANCE: InstanceQuery, 
 51    constants.QR_NODE: NodeQuery, 
 52    constants.QR_GROUP: GroupQuery, 
 53    constants.QR_NETWORK: NetworkQuery, 
 54    constants.QR_OS: OsQuery, 
 55    constants.QR_EXTSTORAGE: ExtStorageQuery, 
 56    constants.QR_EXPORT: ExportQuery, 
 57    } 
 58   
 59  assert set(_QUERY_IMPL.keys()) == constants.QR_VIA_OP 
 60   
 61   
 62 -def _GetQueryImplementation(name): 
 63    """Returns the implemtnation for a query type. 
 64   
 65    @param name: Query type, must be one of L{constants.QR_VIA_OP} 
 66   
 67    """ 
 68    try: 
 69      return _QUERY_IMPL[name] 
 70    except KeyError: 
 71      raise errors.OpPrereqError("Unknown query resource '%s'" % name, 
 72                                 errors.ECODE_INVAL) 
 73   
 74   
 75 -class LUQuery(NoHooksLU): 
 76    """Query for resources/items of a certain kind. 
 77   
 78    """ 
 79    # pylint: disable=W0142 
 80    REQ_BGL = False 
 81   
 82 -  def CheckArguments(self): 
 83      qcls = _GetQueryImplementation(self.op.what) 
 84   
 85      self.impl = qcls(self.op.qfilter, self.op.fields, self.op.use_locking) 
 86   
 87 -  def ExpandNames(self): 
 88      self.impl.ExpandNames(self) 
 89   
 90 -  def DeclareLocks(self, level): 
 91      self.impl.DeclareLocks(self, level) 
 92   
 93 -  def Exec(self, feedback_fn): 
 94      return self.impl.NewStyleQuery(self) 
 95   
 96   
 97 -class LUQueryFields(NoHooksLU): 
 98    """Query for resources/items of a certain kind. 
 99   
100    """ 
101    # pylint: disable=W0142 
102    REQ_BGL = False 
103   
104 -  def CheckArguments(self): 
105      self.qcls = _GetQueryImplementation(self.op.what) 
106   
107 -  def ExpandNames(self): 
108      self.needed_locks = {} 
109   
110 -  def Exec(self, feedback_fn): 
111      return query.QueryFields(self.qcls.FIELDS, self.op.fields) 
112