Statistics
| Branch: | Tag: | Revision:

root / lib / client / gnt_instance.py @ 0f8810df

History | View | Annotate | Download (53.1 kB)

1 e792102d Michael Hanselmann
#
2 a8083063 Iustin Pop
#
3 a8083063 Iustin Pop
4 60472d29 Iustin Pop
# Copyright (C) 2006, 2007, 2008, 2009, 2010, 2011 Google Inc.
5 a8083063 Iustin Pop
#
6 a8083063 Iustin Pop
# This program is free software; you can redistribute it and/or modify
7 a8083063 Iustin Pop
# it under the terms of the GNU General Public License as published by
8 a8083063 Iustin Pop
# the Free Software Foundation; either version 2 of the License, or
9 a8083063 Iustin Pop
# (at your option) any later version.
10 a8083063 Iustin Pop
#
11 a8083063 Iustin Pop
# This program is distributed in the hope that it will be useful, but
12 a8083063 Iustin Pop
# WITHOUT ANY WARRANTY; without even the implied warranty of
13 a8083063 Iustin Pop
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 a8083063 Iustin Pop
# General Public License for more details.
15 a8083063 Iustin Pop
#
16 a8083063 Iustin Pop
# You should have received a copy of the GNU General Public License
17 a8083063 Iustin Pop
# along with this program; if not, write to the Free Software
18 a8083063 Iustin Pop
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
19 a8083063 Iustin Pop
# 02110-1301, USA.
20 a8083063 Iustin Pop
21 7260cfbe Iustin Pop
"""Instance related commands"""
22 a8083063 Iustin Pop
23 7260cfbe Iustin Pop
# pylint: disable-msg=W0401,W0614,C0103
24 2f79bd34 Iustin Pop
# W0401: Wildcard import ganeti.cli
25 2f79bd34 Iustin Pop
# W0614: Unused import %s from wildcard import (since we need cli)
26 7260cfbe Iustin Pop
# C0103: Invalid name gnt-instance
27 2f79bd34 Iustin Pop
28 312ac745 Iustin Pop
import itertools
29 0d0e9090 René Nussbaumer
import simplejson
30 25ce3ec4 Michael Hanselmann
import logging
31 a8083063 Iustin Pop
from cStringIO import StringIO
32 a8083063 Iustin Pop
33 a8083063 Iustin Pop
from ganeti.cli import *
34 a8083063 Iustin Pop
from ganeti import opcodes
35 a8083063 Iustin Pop
from ganeti import constants
36 e2736e40 Guido Trotter
from ganeti import compat
37 a8083063 Iustin Pop
from ganeti import utils
38 312ac745 Iustin Pop
from ganeti import errors
39 a744b676 Manuel Franceschini
from ganeti import netutils
40 25ce3ec4 Michael Hanselmann
from ganeti import ssh
41 25ce3ec4 Michael Hanselmann
from ganeti import objects
42 312ac745 Iustin Pop
43 312ac745 Iustin Pop
44 312ac745 Iustin Pop
_SHUTDOWN_CLUSTER = "cluster"
45 312ac745 Iustin Pop
_SHUTDOWN_NODES_BOTH = "nodes"
46 312ac745 Iustin Pop
_SHUTDOWN_NODES_PRI = "nodes-pri"
47 312ac745 Iustin Pop
_SHUTDOWN_NODES_SEC = "nodes-sec"
48 39dfd93e René Nussbaumer
_SHUTDOWN_NODES_BOTH_BY_TAGS = "nodes-by-tags"
49 39dfd93e René Nussbaumer
_SHUTDOWN_NODES_PRI_BY_TAGS = "nodes-pri-by-tags"
50 39dfd93e René Nussbaumer
_SHUTDOWN_NODES_SEC_BY_TAGS = "nodes-sec-by-tags"
51 312ac745 Iustin Pop
_SHUTDOWN_INSTANCES = "instances"
52 39dfd93e René Nussbaumer
_SHUTDOWN_INSTANCES_BY_TAGS = "instances-by-tags"
53 39dfd93e René Nussbaumer
54 39dfd93e René Nussbaumer
_SHUTDOWN_NODES_TAGS_MODES = (
55 39dfd93e René Nussbaumer
    _SHUTDOWN_NODES_BOTH_BY_TAGS,
56 39dfd93e René Nussbaumer
    _SHUTDOWN_NODES_PRI_BY_TAGS,
57 39dfd93e René Nussbaumer
    _SHUTDOWN_NODES_SEC_BY_TAGS)
58 312ac745 Iustin Pop
59 7c0d6283 Michael Hanselmann
60 7232c04c Iustin Pop
#: default list of options for L{ListInstances}
61 48c4dfa8 Iustin Pop
_LIST_DEF_FIELDS = [
62 e69d05fd Iustin Pop
  "name", "hypervisor", "os", "pnode", "status", "oper_ram",
63 48c4dfa8 Iustin Pop
  ]
64 48c4dfa8 Iustin Pop
65 bdb7d4e8 Michael Hanselmann
66 479636a3 Iustin Pop
def _ExpandMultiNames(mode, names, client=None):
67 312ac745 Iustin Pop
  """Expand the given names using the passed mode.
68 312ac745 Iustin Pop

69 312ac745 Iustin Pop
  For _SHUTDOWN_CLUSTER, all instances will be returned. For
70 312ac745 Iustin Pop
  _SHUTDOWN_NODES_PRI/SEC, all instances having those nodes as
71 7232c04c Iustin Pop
  primary/secondary will be returned. For _SHUTDOWN_NODES_BOTH, all
72 312ac745 Iustin Pop
  instances having those nodes as either primary or secondary will be
73 312ac745 Iustin Pop
  returned. For _SHUTDOWN_INSTANCES, the given instances will be
74 312ac745 Iustin Pop
  returned.
75 312ac745 Iustin Pop

76 7232c04c Iustin Pop
  @param mode: one of L{_SHUTDOWN_CLUSTER}, L{_SHUTDOWN_NODES_BOTH},
77 7232c04c Iustin Pop
      L{_SHUTDOWN_NODES_PRI}, L{_SHUTDOWN_NODES_SEC} or
78 7232c04c Iustin Pop
      L{_SHUTDOWN_INSTANCES}
79 7232c04c Iustin Pop
  @param names: a list of names; for cluster, it must be empty,
80 7232c04c Iustin Pop
      and for node and instance it must be a list of valid item
81 7232c04c Iustin Pop
      names (short names are valid as usual, e.g. node1 instead of
82 7232c04c Iustin Pop
      node1.example.com)
83 7232c04c Iustin Pop
  @rtype: list
84 7232c04c Iustin Pop
  @return: the list of names after the expansion
85 7232c04c Iustin Pop
  @raise errors.ProgrammerError: for unknown selection type
86 7232c04c Iustin Pop
  @raise errors.OpPrereqError: for invalid input parameters
87 7232c04c Iustin Pop

88 312ac745 Iustin Pop
  """
89 7260cfbe Iustin Pop
  # pylint: disable-msg=W0142
90 39dfd93e René Nussbaumer
91 479636a3 Iustin Pop
  if client is None:
92 479636a3 Iustin Pop
    client = GetClient()
93 312ac745 Iustin Pop
  if mode == _SHUTDOWN_CLUSTER:
94 312ac745 Iustin Pop
    if names:
95 debac808 Iustin Pop
      raise errors.OpPrereqError("Cluster filter mode takes no arguments",
96 debac808 Iustin Pop
                                 errors.ECODE_INVAL)
97 ec79568d Iustin Pop
    idata = client.QueryInstances([], ["name"], False)
98 312ac745 Iustin Pop
    inames = [row[0] for row in idata]
99 312ac745 Iustin Pop
100 312ac745 Iustin Pop
  elif mode in (_SHUTDOWN_NODES_BOTH,
101 312ac745 Iustin Pop
                _SHUTDOWN_NODES_PRI,
102 39dfd93e René Nussbaumer
                _SHUTDOWN_NODES_SEC) + _SHUTDOWN_NODES_TAGS_MODES:
103 39dfd93e René Nussbaumer
    if mode in _SHUTDOWN_NODES_TAGS_MODES:
104 39dfd93e René Nussbaumer
      if not names:
105 39dfd93e René Nussbaumer
        raise errors.OpPrereqError("No node tags passed", errors.ECODE_INVAL)
106 39dfd93e René Nussbaumer
      ndata = client.QueryNodes([], ["name", "pinst_list",
107 39dfd93e René Nussbaumer
                                     "sinst_list", "tags"], False)
108 39dfd93e René Nussbaumer
      ndata = [row for row in ndata if set(row[3]).intersection(names)]
109 39dfd93e René Nussbaumer
    else:
110 39dfd93e René Nussbaumer
      if not names:
111 39dfd93e René Nussbaumer
        raise errors.OpPrereqError("No node names passed", errors.ECODE_INVAL)
112 39dfd93e René Nussbaumer
      ndata = client.QueryNodes(names, ["name", "pinst_list", "sinst_list"],
113 77921a95 Iustin Pop
                              False)
114 39dfd93e René Nussbaumer
115 312ac745 Iustin Pop
    ipri = [row[1] for row in ndata]
116 312ac745 Iustin Pop
    pri_names = list(itertools.chain(*ipri))
117 312ac745 Iustin Pop
    isec = [row[2] for row in ndata]
118 312ac745 Iustin Pop
    sec_names = list(itertools.chain(*isec))
119 39dfd93e René Nussbaumer
    if mode in (_SHUTDOWN_NODES_BOTH, _SHUTDOWN_NODES_BOTH_BY_TAGS):
120 312ac745 Iustin Pop
      inames = pri_names + sec_names
121 39dfd93e René Nussbaumer
    elif mode in (_SHUTDOWN_NODES_PRI, _SHUTDOWN_NODES_PRI_BY_TAGS):
122 312ac745 Iustin Pop
      inames = pri_names
123 39dfd93e René Nussbaumer
    elif mode in (_SHUTDOWN_NODES_SEC, _SHUTDOWN_NODES_SEC_BY_TAGS):
124 312ac745 Iustin Pop
      inames = sec_names
125 312ac745 Iustin Pop
    else:
126 312ac745 Iustin Pop
      raise errors.ProgrammerError("Unhandled shutdown type")
127 312ac745 Iustin Pop
  elif mode == _SHUTDOWN_INSTANCES:
128 312ac745 Iustin Pop
    if not names:
129 debac808 Iustin Pop
      raise errors.OpPrereqError("No instance names passed",
130 debac808 Iustin Pop
                                 errors.ECODE_INVAL)
131 ec79568d Iustin Pop
    idata = client.QueryInstances(names, ["name"], False)
132 312ac745 Iustin Pop
    inames = [row[0] for row in idata]
133 39dfd93e René Nussbaumer
  elif mode == _SHUTDOWN_INSTANCES_BY_TAGS:
134 39dfd93e René Nussbaumer
    if not names:
135 39dfd93e René Nussbaumer
      raise errors.OpPrereqError("No instance tags passed",
136 39dfd93e René Nussbaumer
                                 errors.ECODE_INVAL)
137 39dfd93e René Nussbaumer
    idata = client.QueryInstances([], ["name", "tags"], False)
138 39dfd93e René Nussbaumer
    inames = [row[0] for row in idata if set(row[1]).intersection(names)]
139 312ac745 Iustin Pop
  else:
140 debac808 Iustin Pop
    raise errors.OpPrereqError("Unknown mode '%s'" % mode, errors.ECODE_INVAL)
141 312ac745 Iustin Pop
142 312ac745 Iustin Pop
  return inames
143 a8083063 Iustin Pop
144 a8083063 Iustin Pop
145 a76f0c4a Iustin Pop
def _EnsureInstancesExist(client, names):
146 a76f0c4a Iustin Pop
  """Check for and ensure the given instance names exist.
147 a76f0c4a Iustin Pop

148 a76f0c4a Iustin Pop
  This function will raise an OpPrereqError in case they don't
149 a76f0c4a Iustin Pop
  exist. Otherwise it will exit cleanly.
150 a76f0c4a Iustin Pop

151 f2fd87d7 Iustin Pop
  @type client: L{ganeti.luxi.Client}
152 a76f0c4a Iustin Pop
  @param client: the client to use for the query
153 a76f0c4a Iustin Pop
  @type names: list
154 a76f0c4a Iustin Pop
  @param names: the list of instance names to query
155 a76f0c4a Iustin Pop
  @raise errors.OpPrereqError: in case any instance is missing
156 a76f0c4a Iustin Pop

157 a76f0c4a Iustin Pop
  """
158 f2af0bec Iustin Pop
  # TODO: change LUInstanceQuery to that it actually returns None
159 a76f0c4a Iustin Pop
  # instead of raising an exception, or devise a better mechanism
160 ec79568d Iustin Pop
  result = client.QueryInstances(names, ["name"], False)
161 a76f0c4a Iustin Pop
  for orig_name, row in zip(names, result):
162 a76f0c4a Iustin Pop
    if row[0] is None:
163 debac808 Iustin Pop
      raise errors.OpPrereqError("Instance '%s' does not exist" % orig_name,
164 debac808 Iustin Pop
                                 errors.ECODE_NOENT)
165 a76f0c4a Iustin Pop
166 a76f0c4a Iustin Pop
167 1c5945b6 Iustin Pop
def GenericManyOps(operation, fn):
168 1c5945b6 Iustin Pop
  """Generic multi-instance operations.
169 1c5945b6 Iustin Pop

170 1c5945b6 Iustin Pop
  The will return a wrapper that processes the options and arguments
171 1c5945b6 Iustin Pop
  given, and uses the passed function to build the opcode needed for
172 1c5945b6 Iustin Pop
  the specific operation. Thus all the generic loop/confirmation code
173 1c5945b6 Iustin Pop
  is abstracted into this function.
174 1c5945b6 Iustin Pop

175 1c5945b6 Iustin Pop
  """
176 1c5945b6 Iustin Pop
  def realfn(opts, args):
177 1c5945b6 Iustin Pop
    if opts.multi_mode is None:
178 1c5945b6 Iustin Pop
      opts.multi_mode = _SHUTDOWN_INSTANCES
179 1c5945b6 Iustin Pop
    cl = GetClient()
180 1c5945b6 Iustin Pop
    inames = _ExpandMultiNames(opts.multi_mode, args, client=cl)
181 1c5945b6 Iustin Pop
    if not inames:
182 c37bb2c6 Stephen Shirley
      if opts.multi_mode == _SHUTDOWN_CLUSTER:
183 c37bb2c6 Stephen Shirley
        ToStdout("Cluster is empty, no instances to shutdown")
184 c37bb2c6 Stephen Shirley
        return 0
185 1c5945b6 Iustin Pop
      raise errors.OpPrereqError("Selection filter does not match"
186 debac808 Iustin Pop
                                 " any instances", errors.ECODE_INVAL)
187 1c5945b6 Iustin Pop
    multi_on = opts.multi_mode != _SHUTDOWN_INSTANCES or len(inames) > 1
188 1c5945b6 Iustin Pop
    if not (opts.force_multi or not multi_on
189 25bd815c René Nussbaumer
            or ConfirmOperation(inames, "instances", operation)):
190 1c5945b6 Iustin Pop
      return 1
191 cb573a31 Iustin Pop
    jex = JobExecutor(verbose=multi_on, cl=cl, opts=opts)
192 1c5945b6 Iustin Pop
    for name in inames:
193 1c5945b6 Iustin Pop
      op = fn(name, opts)
194 1c5945b6 Iustin Pop
      jex.QueueJob(name, op)
195 b4e68848 Iustin Pop
    results = jex.WaitOrShow(not opts.submit_only)
196 b4e68848 Iustin Pop
    rcode = compat.all(row[0] for row in results)
197 b4e68848 Iustin Pop
    return int(not rcode)
198 1c5945b6 Iustin Pop
  return realfn
199 1c5945b6 Iustin Pop
200 1c5945b6 Iustin Pop
201 a8083063 Iustin Pop
def ListInstances(opts, args):
202 f5abe9bd Oleksiy Mishchenko
  """List instances and their properties.
203 a8083063 Iustin Pop

204 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
205 7232c04c Iustin Pop
  @type args: list
206 7232c04c Iustin Pop
  @param args: should be an empty list
207 7232c04c Iustin Pop
  @rtype: int
208 7232c04c Iustin Pop
  @return: the desired exit code
209 7232c04c Iustin Pop

210 a8083063 Iustin Pop
  """
211 a4ebd726 Michael Hanselmann
  selected_fields = ParseFields(opts.output, _LIST_DEF_FIELDS)
212 a8083063 Iustin Pop
213 b82c5ff5 Michael Hanselmann
  fmtoverride = dict.fromkeys(["tags", "disk.sizes", "nic.macs", "nic.ips",
214 b82c5ff5 Michael Hanselmann
                               "nic.modes", "nic.links", "nic.bridges",
215 b82c5ff5 Michael Hanselmann
                               "snodes"],
216 b82c5ff5 Michael Hanselmann
                              (lambda value: ",".join(str(item)
217 b82c5ff5 Michael Hanselmann
                                                      for item in value),
218 b82c5ff5 Michael Hanselmann
                               False))
219 a8083063 Iustin Pop
220 b82c5ff5 Michael Hanselmann
  return GenericList(constants.QR_INSTANCE, selected_fields, args, opts.units,
221 b82c5ff5 Michael Hanselmann
                     opts.separator, not opts.no_headers,
222 87e87959 Michael Hanselmann
                     format_override=fmtoverride, verbose=opts.verbose,
223 87e87959 Michael Hanselmann
                     force_filter=opts.force_filter)
224 b82c5ff5 Michael Hanselmann
225 b82c5ff5 Michael Hanselmann
226 b82c5ff5 Michael Hanselmann
def ListInstanceFields(opts, args):
227 b82c5ff5 Michael Hanselmann
  """List instance fields.
228 b82c5ff5 Michael Hanselmann

229 b82c5ff5 Michael Hanselmann
  @param opts: the command line options selected by the user
230 b82c5ff5 Michael Hanselmann
  @type args: list
231 b82c5ff5 Michael Hanselmann
  @param args: fields to list, or empty for all
232 b82c5ff5 Michael Hanselmann
  @rtype: int
233 b82c5ff5 Michael Hanselmann
  @return: the desired exit code
234 b82c5ff5 Michael Hanselmann

235 b82c5ff5 Michael Hanselmann
  """
236 b82c5ff5 Michael Hanselmann
  return GenericListFields(constants.QR_INSTANCE, args, opts.separator,
237 b82c5ff5 Michael Hanselmann
                           not opts.no_headers)
238 a8083063 Iustin Pop
239 a8083063 Iustin Pop
240 a8083063 Iustin Pop
def AddInstance(opts, args):
241 a8083063 Iustin Pop
  """Add an instance to the cluster.
242 a8083063 Iustin Pop

243 d77490c5 Iustin Pop
  This is just a wrapper over GenericInstanceCreate.
244 a8083063 Iustin Pop

245 a8083063 Iustin Pop
  """
246 d77490c5 Iustin Pop
  return GenericInstanceCreate(constants.INSTANCE_CREATE, opts, args)
247 a8083063 Iustin Pop
248 a8083063 Iustin Pop
249 0d0e9090 René Nussbaumer
def BatchCreate(opts, args):
250 7232c04c Iustin Pop
  """Create instances using a definition file.
251 7232c04c Iustin Pop

252 7232c04c Iustin Pop
  This function reads a json file with instances defined
253 7232c04c Iustin Pop
  in the form::
254 7232c04c Iustin Pop

255 7232c04c Iustin Pop
    {"instance-name":{
256 9939547b Iustin Pop
      "disk_size": [20480],
257 7232c04c Iustin Pop
      "template": "drbd",
258 7232c04c Iustin Pop
      "backend": {
259 7232c04c Iustin Pop
        "memory": 512,
260 7232c04c Iustin Pop
        "vcpus": 1 },
261 9939547b Iustin Pop
      "os": "debootstrap",
262 7232c04c Iustin Pop
      "primary_node": "firstnode",
263 7232c04c Iustin Pop
      "secondary_node": "secondnode",
264 7232c04c Iustin Pop
      "iallocator": "dumb"}
265 7232c04c Iustin Pop
    }
266 7232c04c Iustin Pop

267 7232c04c Iustin Pop
  Note that I{primary_node} and I{secondary_node} have precedence over
268 7232c04c Iustin Pop
  I{iallocator}.
269 7232c04c Iustin Pop

270 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
271 7232c04c Iustin Pop
  @type args: list
272 7232c04c Iustin Pop
  @param args: should contain one element, the json filename
273 7232c04c Iustin Pop
  @rtype: int
274 7232c04c Iustin Pop
  @return: the desired exit code
275 0d0e9090 René Nussbaumer

276 0d0e9090 René Nussbaumer
  """
277 9939547b Iustin Pop
  _DEFAULT_SPECS = {"disk_size": [20 * 1024],
278 0d0e9090 René Nussbaumer
                    "backend": {},
279 0d0e9090 René Nussbaumer
                    "iallocator": None,
280 0d0e9090 René Nussbaumer
                    "primary_node": None,
281 0d0e9090 René Nussbaumer
                    "secondary_node": None,
282 a379d9bd Guido Trotter
                    "nics": None,
283 0d0e9090 René Nussbaumer
                    "start": True,
284 0d0e9090 René Nussbaumer
                    "ip_check": True,
285 460d22be Iustin Pop
                    "name_check": True,
286 0d0e9090 René Nussbaumer
                    "hypervisor": None,
287 4082e6f9 Iustin Pop
                    "hvparams": {},
288 0d0e9090 René Nussbaumer
                    "file_storage_dir": None,
289 27ba7eab Guido Trotter
                    "force_variant": False,
290 0d0e9090 René Nussbaumer
                    "file_driver": 'loop'}
291 0d0e9090 René Nussbaumer
292 0d0e9090 René Nussbaumer
  def _PopulateWithDefaults(spec):
293 0d0e9090 René Nussbaumer
    """Returns a new hash combined with default values."""
294 2f79bd34 Iustin Pop
    mydict = _DEFAULT_SPECS.copy()
295 2f79bd34 Iustin Pop
    mydict.update(spec)
296 2f79bd34 Iustin Pop
    return mydict
297 0d0e9090 René Nussbaumer
298 0d0e9090 René Nussbaumer
  def _Validate(spec):
299 0d0e9090 René Nussbaumer
    """Validate the instance specs."""
300 0d0e9090 René Nussbaumer
    # Validate fields required under any circumstances
301 0d0e9090 René Nussbaumer
    for required_field in ('os', 'template'):
302 0d0e9090 René Nussbaumer
      if required_field not in spec:
303 0d0e9090 René Nussbaumer
        raise errors.OpPrereqError('Required field "%s" is missing.' %
304 debac808 Iustin Pop
                                   required_field, errors.ECODE_INVAL)
305 0d0e9090 René Nussbaumer
    # Validate special fields
306 0d0e9090 René Nussbaumer
    if spec['primary_node'] is not None:
307 3429a076 Apollon Oikonomopoulos
      if (spec['template'] in constants.DTS_INT_MIRROR and
308 0d0e9090 René Nussbaumer
          spec['secondary_node'] is None):
309 0d0e9090 René Nussbaumer
        raise errors.OpPrereqError('Template requires secondary node, but'
310 debac808 Iustin Pop
                                   ' there was no secondary provided.',
311 debac808 Iustin Pop
                                   errors.ECODE_INVAL)
312 0d0e9090 René Nussbaumer
    elif spec['iallocator'] is None:
313 0d0e9090 René Nussbaumer
      raise errors.OpPrereqError('You have to provide at least a primary_node'
314 debac808 Iustin Pop
                                 ' or an iallocator.',
315 debac808 Iustin Pop
                                 errors.ECODE_INVAL)
316 0d0e9090 René Nussbaumer
317 4082e6f9 Iustin Pop
    if (spec['hvparams'] and
318 4082e6f9 Iustin Pop
        not isinstance(spec['hvparams'], dict)):
319 debac808 Iustin Pop
      raise errors.OpPrereqError('Hypervisor parameters must be a dict.',
320 debac808 Iustin Pop
                                 errors.ECODE_INVAL)
321 0d0e9090 René Nussbaumer
322 0d0e9090 René Nussbaumer
  json_filename = args[0]
323 0d0e9090 René Nussbaumer
  try:
324 13998ef2 Michael Hanselmann
    instance_data = simplejson.loads(utils.ReadFile(json_filename))
325 7260cfbe Iustin Pop
  except Exception, err: # pylint: disable-msg=W0703
326 4082e6f9 Iustin Pop
    ToStderr("Can't parse the instance definition file: %s" % str(err))
327 4082e6f9 Iustin Pop
    return 1
328 0d0e9090 René Nussbaumer
329 fe7c59d5 Guido Trotter
  if not isinstance(instance_data, dict):
330 fe7c59d5 Guido Trotter
    ToStderr("The instance definition file is not in dict format.")
331 fe7c59d5 Guido Trotter
    return 1
332 fe7c59d5 Guido Trotter
333 cb573a31 Iustin Pop
  jex = JobExecutor(opts=opts)
334 d4dd4b74 Iustin Pop
335 0d0e9090 René Nussbaumer
  # Iterate over the instances and do:
336 0d0e9090 René Nussbaumer
  #  * Populate the specs with default value
337 0d0e9090 René Nussbaumer
  #  * Validate the instance specs
338 fe7c59d5 Guido Trotter
  i_names = utils.NiceSort(instance_data.keys()) # pylint: disable-msg=E1103
339 7312b33d Iustin Pop
  for name in i_names:
340 7312b33d Iustin Pop
    specs = instance_data[name]
341 0d0e9090 René Nussbaumer
    specs = _PopulateWithDefaults(specs)
342 0d0e9090 René Nussbaumer
    _Validate(specs)
343 0d0e9090 René Nussbaumer
344 4082e6f9 Iustin Pop
    hypervisor = specs['hypervisor']
345 4082e6f9 Iustin Pop
    hvparams = specs['hvparams']
346 0d0e9090 René Nussbaumer
347 9939547b Iustin Pop
    disks = []
348 9939547b Iustin Pop
    for elem in specs['disk_size']:
349 9939547b Iustin Pop
      try:
350 9939547b Iustin Pop
        size = utils.ParseUnit(elem)
351 691744c4 Iustin Pop
      except (TypeError, ValueError), err:
352 9939547b Iustin Pop
        raise errors.OpPrereqError("Invalid disk size '%s' for"
353 9939547b Iustin Pop
                                   " instance %s: %s" %
354 debac808 Iustin Pop
                                   (elem, name, err), errors.ECODE_INVAL)
355 9939547b Iustin Pop
      disks.append({"size": size})
356 9939547b Iustin Pop
357 a5728081 Guido Trotter
    utils.ForceDictType(specs['backend'], constants.BES_PARAMETER_TYPES)
358 a5728081 Guido Trotter
    utils.ForceDictType(hvparams, constants.HVS_PARAMETER_TYPES)
359 a5728081 Guido Trotter
360 a379d9bd Guido Trotter
    tmp_nics = []
361 717920a0 Guido Trotter
    for field in constants.INIC_PARAMS:
362 a379d9bd Guido Trotter
      if field in specs:
363 a379d9bd Guido Trotter
        if not tmp_nics:
364 a379d9bd Guido Trotter
          tmp_nics.append({})
365 a379d9bd Guido Trotter
        tmp_nics[0][field] = specs[field]
366 a379d9bd Guido Trotter
367 a379d9bd Guido Trotter
    if specs['nics'] is not None and tmp_nics:
368 a379d9bd Guido Trotter
      raise errors.OpPrereqError("'nics' list incompatible with using"
369 debac808 Iustin Pop
                                 " individual nic fields as well",
370 debac808 Iustin Pop
                                 errors.ECODE_INVAL)
371 a379d9bd Guido Trotter
    elif specs['nics'] is not None:
372 a379d9bd Guido Trotter
      tmp_nics = specs['nics']
373 a379d9bd Guido Trotter
    elif not tmp_nics:
374 a379d9bd Guido Trotter
      tmp_nics = [{}]
375 a379d9bd Guido Trotter
376 e1530b10 Iustin Pop
    op = opcodes.OpInstanceCreate(instance_name=name,
377 9939547b Iustin Pop
                                  disks=disks,
378 0d0e9090 René Nussbaumer
                                  disk_template=specs['template'],
379 0d0e9090 René Nussbaumer
                                  mode=constants.INSTANCE_CREATE,
380 0d0e9090 René Nussbaumer
                                  os_type=specs['os'],
381 de372295 Guido Trotter
                                  force_variant=specs["force_variant"],
382 0d0e9090 René Nussbaumer
                                  pnode=specs['primary_node'],
383 0d0e9090 René Nussbaumer
                                  snode=specs['secondary_node'],
384 a379d9bd Guido Trotter
                                  nics=tmp_nics,
385 0d0e9090 René Nussbaumer
                                  start=specs['start'],
386 0d0e9090 René Nussbaumer
                                  ip_check=specs['ip_check'],
387 460d22be Iustin Pop
                                  name_check=specs['name_check'],
388 0d0e9090 René Nussbaumer
                                  wait_for_sync=True,
389 0d0e9090 René Nussbaumer
                                  iallocator=specs['iallocator'],
390 0d0e9090 René Nussbaumer
                                  hypervisor=hypervisor,
391 0d0e9090 René Nussbaumer
                                  hvparams=hvparams,
392 0d0e9090 René Nussbaumer
                                  beparams=specs['backend'],
393 0d0e9090 René Nussbaumer
                                  file_storage_dir=specs['file_storage_dir'],
394 0d0e9090 René Nussbaumer
                                  file_driver=specs['file_driver'])
395 0d0e9090 René Nussbaumer
396 d4dd4b74 Iustin Pop
    jex.QueueJob(name, op)
397 d4dd4b74 Iustin Pop
  # we never want to wait, just show the submitted job IDs
398 d4dd4b74 Iustin Pop
  jex.WaitOrShow(False)
399 0d0e9090 René Nussbaumer
400 0d0e9090 René Nussbaumer
  return 0
401 0d0e9090 René Nussbaumer
402 0d0e9090 René Nussbaumer
403 fe7b0351 Michael Hanselmann
def ReinstallInstance(opts, args):
404 fe7b0351 Michael Hanselmann
  """Reinstall an instance.
405 fe7b0351 Michael Hanselmann

406 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
407 7232c04c Iustin Pop
  @type args: list
408 7232c04c Iustin Pop
  @param args: should contain only one element, the name of the
409 7232c04c Iustin Pop
      instance to be reinstalled
410 7232c04c Iustin Pop
  @rtype: int
411 7232c04c Iustin Pop
  @return: the desired exit code
412 fe7b0351 Michael Hanselmann

413 fe7b0351 Michael Hanselmann
  """
414 55efe6da Iustin Pop
  # first, compute the desired name list
415 55efe6da Iustin Pop
  if opts.multi_mode is None:
416 55efe6da Iustin Pop
    opts.multi_mode = _SHUTDOWN_INSTANCES
417 55efe6da Iustin Pop
418 55efe6da Iustin Pop
  inames = _ExpandMultiNames(opts.multi_mode, args)
419 55efe6da Iustin Pop
  if not inames:
420 debac808 Iustin Pop
    raise errors.OpPrereqError("Selection filter does not match any instances",
421 debac808 Iustin Pop
                               errors.ECODE_INVAL)
422 fe7b0351 Michael Hanselmann
423 55efe6da Iustin Pop
  # second, if requested, ask for an OS
424 20e23543 Alexander Schreiber
  if opts.select_os is True:
425 da2d02e7 Iustin Pop
    op = opcodes.OpOsDiagnose(output_fields=["name", "variants"], names=[])
426 400ca2f7 Iustin Pop
    result = SubmitOpCode(op, opts=opts)
427 20e23543 Alexander Schreiber
428 20e23543 Alexander Schreiber
    if not result:
429 3a24c527 Iustin Pop
      ToStdout("Can't get the OS list")
430 20e23543 Alexander Schreiber
      return 1
431 20e23543 Alexander Schreiber
432 3a24c527 Iustin Pop
    ToStdout("Available OS templates:")
433 20e23543 Alexander Schreiber
    number = 0
434 20e23543 Alexander Schreiber
    choices = []
435 d22dfef7 Iustin Pop
    for (name, variants) in result:
436 d22dfef7 Iustin Pop
      for entry in CalculateOSNames(name, variants):
437 d22dfef7 Iustin Pop
        ToStdout("%3s: %s", number, entry)
438 d22dfef7 Iustin Pop
        choices.append(("%s" % number, entry, entry))
439 d22dfef7 Iustin Pop
        number += 1
440 20e23543 Alexander Schreiber
441 20e23543 Alexander Schreiber
    choices.append(('x', 'exit', 'Exit gnt-instance reinstall'))
442 949bdabe Iustin Pop
    selected = AskUser("Enter OS template number (or x to abort):",
443 20e23543 Alexander Schreiber
                       choices)
444 20e23543 Alexander Schreiber
445 20e23543 Alexander Schreiber
    if selected == 'exit':
446 55efe6da Iustin Pop
      ToStderr("User aborted reinstall, exiting")
447 20e23543 Alexander Schreiber
      return 1
448 20e23543 Alexander Schreiber
449 2f79bd34 Iustin Pop
    os_name = selected
450 f86426f5 Iustin Pop
    os_msg = "change the OS to '%s'" % selected
451 20e23543 Alexander Schreiber
  else:
452 2f79bd34 Iustin Pop
    os_name = opts.os
453 f86426f5 Iustin Pop
    if opts.os is not None:
454 f86426f5 Iustin Pop
      os_msg = "change the OS to '%s'" % os_name
455 f86426f5 Iustin Pop
    else:
456 f86426f5 Iustin Pop
      os_msg = "keep the same OS"
457 20e23543 Alexander Schreiber
458 297ddce9 Iustin Pop
  # third, get confirmation: multi-reinstall requires --force-multi,
459 297ddce9 Iustin Pop
  # single-reinstall either --force or --force-multi (--force-multi is
460 297ddce9 Iustin Pop
  # a stronger --force)
461 55efe6da Iustin Pop
  multi_on = opts.multi_mode != _SHUTDOWN_INSTANCES or len(inames) > 1
462 55efe6da Iustin Pop
  if multi_on:
463 f86426f5 Iustin Pop
    warn_msg = ("Note: this will remove *all* data for the"
464 f86426f5 Iustin Pop
                " below instances! It will %s.\n" % os_msg)
465 297ddce9 Iustin Pop
    if not (opts.force_multi or
466 25bd815c René Nussbaumer
            ConfirmOperation(inames, "instances", "reinstall", extra=warn_msg)):
467 fe7b0351 Michael Hanselmann
      return 1
468 55efe6da Iustin Pop
  else:
469 297ddce9 Iustin Pop
    if not (opts.force or opts.force_multi):
470 f86426f5 Iustin Pop
      usertext = ("This will reinstall the instance '%s' (and %s) which"
471 f86426f5 Iustin Pop
                  " removes all data. Continue?") % (inames[0], os_msg)
472 55efe6da Iustin Pop
      if not AskUser(usertext):
473 55efe6da Iustin Pop
        return 1
474 55efe6da Iustin Pop
475 cb573a31 Iustin Pop
  jex = JobExecutor(verbose=multi_on, opts=opts)
476 55efe6da Iustin Pop
  for instance_name in inames:
477 5073fd8f Iustin Pop
    op = opcodes.OpInstanceReinstall(instance_name=instance_name,
478 06073e85 Guido Trotter
                                     os_type=os_name,
479 8d8c4eff Michael Hanselmann
                                     force_variant=opts.force_variant,
480 8d8c4eff Michael Hanselmann
                                     osparams=opts.osparams)
481 55efe6da Iustin Pop
    jex.QueueJob(instance_name, op)
482 fe7b0351 Michael Hanselmann
483 55efe6da Iustin Pop
  jex.WaitOrShow(not opts.submit_only)
484 fe7b0351 Michael Hanselmann
  return 0
485 fe7b0351 Michael Hanselmann
486 fe7b0351 Michael Hanselmann
487 a8083063 Iustin Pop
def RemoveInstance(opts, args):
488 a8083063 Iustin Pop
  """Remove an instance.
489 a8083063 Iustin Pop

490 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
491 7232c04c Iustin Pop
  @type args: list
492 7232c04c Iustin Pop
  @param args: should contain only one element, the name of
493 7232c04c Iustin Pop
      the instance to be removed
494 7232c04c Iustin Pop
  @rtype: int
495 7232c04c Iustin Pop
  @return: the desired exit code
496 a8083063 Iustin Pop

497 a8083063 Iustin Pop
  """
498 a8083063 Iustin Pop
  instance_name = args[0]
499 a8083063 Iustin Pop
  force = opts.force
500 a76f0c4a Iustin Pop
  cl = GetClient()
501 a8083063 Iustin Pop
502 a8083063 Iustin Pop
  if not force:
503 a76f0c4a Iustin Pop
    _EnsureInstancesExist(cl, [instance_name])
504 a76f0c4a Iustin Pop
505 a8083063 Iustin Pop
    usertext = ("This will remove the volumes of the instance %s"
506 a8083063 Iustin Pop
                " (including mirrors), thus removing all the data"
507 a8083063 Iustin Pop
                " of the instance. Continue?") % instance_name
508 47988778 Iustin Pop
    if not AskUser(usertext):
509 a8083063 Iustin Pop
      return 1
510 a8083063 Iustin Pop
511 3cd2d4b1 Iustin Pop
  op = opcodes.OpInstanceRemove(instance_name=instance_name,
512 17c3f802 Guido Trotter
                                ignore_failures=opts.ignore_failures,
513 4d98c565 Guido Trotter
                                shutdown_timeout=opts.shutdown_timeout)
514 a76f0c4a Iustin Pop
  SubmitOrSend(op, opts, cl=cl)
515 a8083063 Iustin Pop
  return 0
516 a8083063 Iustin Pop
517 a8083063 Iustin Pop
518 decd5f45 Iustin Pop
def RenameInstance(opts, args):
519 4ab0b9e3 Guido Trotter
  """Rename an instance.
520 decd5f45 Iustin Pop

521 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
522 7232c04c Iustin Pop
  @type args: list
523 7232c04c Iustin Pop
  @param args: should contain two elements, the old and the
524 7232c04c Iustin Pop
      new instance names
525 7232c04c Iustin Pop
  @rtype: int
526 7232c04c Iustin Pop
  @return: the desired exit code
527 decd5f45 Iustin Pop

528 decd5f45 Iustin Pop
  """
529 90ed09b0 René Nussbaumer
  if not opts.name_check:
530 1b6dddc8 René Nussbaumer
    if not AskUser("As you disabled the check of the DNS entry, please verify"
531 1b6dddc8 René Nussbaumer
                   " that '%s' is a FQDN. Continue?" % args[1]):
532 1b6dddc8 René Nussbaumer
      return 1
533 1b6dddc8 René Nussbaumer
534 5659e2e2 Iustin Pop
  op = opcodes.OpInstanceRename(instance_name=args[0],
535 decd5f45 Iustin Pop
                                new_name=args[1],
536 3fe11ba3 Manuel Franceschini
                                ip_check=opts.ip_check,
537 3fe11ba3 Manuel Franceschini
                                name_check=opts.name_check)
538 6a016df9 Michael Hanselmann
  result = SubmitOrSend(op, opts)
539 6a016df9 Michael Hanselmann
540 48418fea Iustin Pop
  if result:
541 48418fea Iustin Pop
    ToStdout("Instance '%s' renamed to '%s'", args[0], result)
542 6a016df9 Michael Hanselmann
543 decd5f45 Iustin Pop
  return 0
544 decd5f45 Iustin Pop
545 decd5f45 Iustin Pop
546 a8083063 Iustin Pop
def ActivateDisks(opts, args):
547 a8083063 Iustin Pop
  """Activate an instance's disks.
548 a8083063 Iustin Pop

549 a8083063 Iustin Pop
  This serves two purposes:
550 7232c04c Iustin Pop
    - it allows (as long as the instance is not running)
551 7232c04c Iustin Pop
      mounting the disks and modifying them from the node
552 a8083063 Iustin Pop
    - it repairs inactive secondary drbds
553 a8083063 Iustin Pop

554 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
555 7232c04c Iustin Pop
  @type args: list
556 7232c04c Iustin Pop
  @param args: should contain only one element, the instance name
557 7232c04c Iustin Pop
  @rtype: int
558 7232c04c Iustin Pop
  @return: the desired exit code
559 7232c04c Iustin Pop

560 a8083063 Iustin Pop
  """
561 a8083063 Iustin Pop
  instance_name = args[0]
562 83f5d475 Iustin Pop
  op = opcodes.OpInstanceActivateDisks(instance_name=instance_name,
563 b4ec07f8 Iustin Pop
                                       ignore_size=opts.ignore_size)
564 6340bb0a Iustin Pop
  disks_info = SubmitOrSend(op, opts)
565 a8083063 Iustin Pop
  for host, iname, nname in disks_info:
566 3a24c527 Iustin Pop
    ToStdout("%s:%s:%s", host, iname, nname)
567 a8083063 Iustin Pop
  return 0
568 a8083063 Iustin Pop
569 a8083063 Iustin Pop
570 a8083063 Iustin Pop
def DeactivateDisks(opts, args):
571 bd315bfa Iustin Pop
  """Deactivate an instance's disks.
572 a8083063 Iustin Pop

573 a8083063 Iustin Pop
  This function takes the instance name, looks for its primary node
574 a8083063 Iustin Pop
  and the tries to shutdown its block devices on that node.
575 a8083063 Iustin Pop

576 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
577 7232c04c Iustin Pop
  @type args: list
578 7232c04c Iustin Pop
  @param args: should contain only one element, the instance name
579 7232c04c Iustin Pop
  @rtype: int
580 7232c04c Iustin Pop
  @return: the desired exit code
581 7232c04c Iustin Pop

582 a8083063 Iustin Pop
  """
583 a8083063 Iustin Pop
  instance_name = args[0]
584 c9c41373 Iustin Pop
  op = opcodes.OpInstanceDeactivateDisks(instance_name=instance_name,
585 c9c41373 Iustin Pop
                                         force=opts.force)
586 6340bb0a Iustin Pop
  SubmitOrSend(op, opts)
587 a8083063 Iustin Pop
  return 0
588 a8083063 Iustin Pop
589 a8083063 Iustin Pop
590 bd315bfa Iustin Pop
def RecreateDisks(opts, args):
591 bd315bfa Iustin Pop
  """Recreate an instance's disks.
592 bd315bfa Iustin Pop

593 bd315bfa Iustin Pop
  @param opts: the command line options selected by the user
594 bd315bfa Iustin Pop
  @type args: list
595 bd315bfa Iustin Pop
  @param args: should contain only one element, the instance name
596 bd315bfa Iustin Pop
  @rtype: int
597 bd315bfa Iustin Pop
  @return: the desired exit code
598 bd315bfa Iustin Pop

599 bd315bfa Iustin Pop
  """
600 bd315bfa Iustin Pop
  instance_name = args[0]
601 bd315bfa Iustin Pop
  if opts.disks:
602 bd315bfa Iustin Pop
    try:
603 bd315bfa Iustin Pop
      opts.disks = [int(v) for v in opts.disks.split(",")]
604 bd315bfa Iustin Pop
    except (ValueError, TypeError), err:
605 bd315bfa Iustin Pop
      ToStderr("Invalid disks value: %s" % str(err))
606 bd315bfa Iustin Pop
      return 1
607 bd315bfa Iustin Pop
  else:
608 bd315bfa Iustin Pop
    opts.disks = []
609 bd315bfa Iustin Pop
610 c8a96ae7 Iustin Pop
  if opts.node:
611 c8a96ae7 Iustin Pop
    pnode, snode = SplitNodeOption(opts.node)
612 c8a96ae7 Iustin Pop
    nodes = [pnode]
613 c8a96ae7 Iustin Pop
    if snode is not None:
614 c8a96ae7 Iustin Pop
      nodes.append(snode)
615 c8a96ae7 Iustin Pop
  else:
616 c8a96ae7 Iustin Pop
    nodes = []
617 c8a96ae7 Iustin Pop
618 6b273e78 Iustin Pop
  op = opcodes.OpInstanceRecreateDisks(instance_name=instance_name,
619 c8a96ae7 Iustin Pop
                                       disks=opts.disks,
620 c8a96ae7 Iustin Pop
                                       nodes=nodes)
621 bd315bfa Iustin Pop
  SubmitOrSend(op, opts)
622 bd315bfa Iustin Pop
  return 0
623 bd315bfa Iustin Pop
624 bd315bfa Iustin Pop
625 c6e911bc Iustin Pop
def GrowDisk(opts, args):
626 7232c04c Iustin Pop
  """Grow an instance's disks.
627 c6e911bc Iustin Pop

628 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
629 7232c04c Iustin Pop
  @type args: list
630 7232c04c Iustin Pop
  @param args: should contain two elements, the instance name
631 7232c04c Iustin Pop
      whose disks we grow and the disk name, e.g. I{sda}
632 7232c04c Iustin Pop
  @rtype: int
633 7232c04c Iustin Pop
  @return: the desired exit code
634 c6e911bc Iustin Pop

635 c6e911bc Iustin Pop
  """
636 c6e911bc Iustin Pop
  instance = args[0]
637 c6e911bc Iustin Pop
  disk = args[1]
638 ad24e046 Iustin Pop
  try:
639 ad24e046 Iustin Pop
    disk = int(disk)
640 691744c4 Iustin Pop
  except (TypeError, ValueError), err:
641 debac808 Iustin Pop
    raise errors.OpPrereqError("Invalid disk index: %s" % str(err),
642 debac808 Iustin Pop
                               errors.ECODE_INVAL)
643 c6e911bc Iustin Pop
  amount = utils.ParseUnit(args[2])
644 60472d29 Iustin Pop
  op = opcodes.OpInstanceGrowDisk(instance_name=instance,
645 60472d29 Iustin Pop
                                  disk=disk, amount=amount,
646 60472d29 Iustin Pop
                                  wait_for_sync=opts.wait_for_sync)
647 6340bb0a Iustin Pop
  SubmitOrSend(op, opts)
648 c6e911bc Iustin Pop
  return 0
649 c6e911bc Iustin Pop
650 c6e911bc Iustin Pop
651 1c5945b6 Iustin Pop
def _StartupInstance(name, opts):
652 7232c04c Iustin Pop
  """Startup instances.
653 a8083063 Iustin Pop

654 1c5945b6 Iustin Pop
  This returns the opcode to start an instance, and its decorator will
655 1c5945b6 Iustin Pop
  wrap this into a loop starting all desired instances.
656 7232c04c Iustin Pop

657 1c5945b6 Iustin Pop
  @param name: the name of the instance to act on
658 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
659 1c5945b6 Iustin Pop
  @return: the opcode needed for the operation
660 a8083063 Iustin Pop

661 a8083063 Iustin Pop
  """
662 c873d91c Iustin Pop
  op = opcodes.OpInstanceStartup(instance_name=name,
663 b44bd844 Michael Hanselmann
                                 force=opts.force,
664 885a0fc4 Iustin Pop
                                 ignore_offline_nodes=opts.ignore_offline,
665 885a0fc4 Iustin Pop
                                 no_remember=opts.no_remember)
666 1c5945b6 Iustin Pop
  # do not add these parameters to the opcode unless they're defined
667 1c5945b6 Iustin Pop
  if opts.hvparams:
668 1c5945b6 Iustin Pop
    op.hvparams = opts.hvparams
669 1c5945b6 Iustin Pop
  if opts.beparams:
670 1c5945b6 Iustin Pop
    op.beparams = opts.beparams
671 1c5945b6 Iustin Pop
  return op
672 a8083063 Iustin Pop
673 7c0d6283 Michael Hanselmann
674 1c5945b6 Iustin Pop
def _RebootInstance(name, opts):
675 7232c04c Iustin Pop
  """Reboot instance(s).
676 7232c04c Iustin Pop

677 1c5945b6 Iustin Pop
  This returns the opcode to reboot an instance, and its decorator
678 1c5945b6 Iustin Pop
  will wrap this into a loop rebooting all desired instances.
679 579d4337 Alexander Schreiber

680 1c5945b6 Iustin Pop
  @param name: the name of the instance to act on
681 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
682 1c5945b6 Iustin Pop
  @return: the opcode needed for the operation
683 579d4337 Alexander Schreiber

684 579d4337 Alexander Schreiber
  """
685 90ab1a95 Iustin Pop
  return opcodes.OpInstanceReboot(instance_name=name,
686 579d4337 Alexander Schreiber
                                  reboot_type=opts.reboot_type,
687 17c3f802 Guido Trotter
                                  ignore_secondaries=opts.ignore_secondaries,
688 4d98c565 Guido Trotter
                                  shutdown_timeout=opts.shutdown_timeout)
689 a8083063 Iustin Pop
690 7c0d6283 Michael Hanselmann
691 1c5945b6 Iustin Pop
def _ShutdownInstance(name, opts):
692 a8083063 Iustin Pop
  """Shutdown an instance.
693 a8083063 Iustin Pop

694 1c5945b6 Iustin Pop
  This returns the opcode to shutdown an instance, and its decorator
695 1c5945b6 Iustin Pop
  will wrap this into a loop shutting down all desired instances.
696 1c5945b6 Iustin Pop

697 1c5945b6 Iustin Pop
  @param name: the name of the instance to act on
698 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
699 1c5945b6 Iustin Pop
  @return: the opcode needed for the operation
700 a8083063 Iustin Pop

701 a8083063 Iustin Pop
  """
702 ee3e37a7 Iustin Pop
  return opcodes.OpInstanceShutdown(instance_name=name,
703 b44bd844 Michael Hanselmann
                                    timeout=opts.timeout,
704 885a0fc4 Iustin Pop
                                    ignore_offline_nodes=opts.ignore_offline,
705 885a0fc4 Iustin Pop
                                    no_remember=opts.no_remember)
706 a8083063 Iustin Pop
707 a8083063 Iustin Pop
708 a8083063 Iustin Pop
def ReplaceDisks(opts, args):
709 a8083063 Iustin Pop
  """Replace the disks of an instance
710 a8083063 Iustin Pop

711 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
712 7232c04c Iustin Pop
  @type args: list
713 7232c04c Iustin Pop
  @param args: should contain only one element, the instance name
714 7232c04c Iustin Pop
  @rtype: int
715 7232c04c Iustin Pop
  @return: the desired exit code
716 a8083063 Iustin Pop

717 a8083063 Iustin Pop
  """
718 a14db5ff Iustin Pop
  new_2ndary = opts.dst_node
719 b6e82a65 Iustin Pop
  iallocator = opts.iallocator
720 a9e0c397 Iustin Pop
  if opts.disks is None:
721 54155f52 Iustin Pop
    disks = []
722 a9e0c397 Iustin Pop
  else:
723 54155f52 Iustin Pop
    try:
724 54155f52 Iustin Pop
      disks = [int(i) for i in opts.disks.split(",")]
725 691744c4 Iustin Pop
    except (TypeError, ValueError), err:
726 debac808 Iustin Pop
      raise errors.OpPrereqError("Invalid disk index passed: %s" % str(err),
727 debac808 Iustin Pop
                                 errors.ECODE_INVAL)
728 05d47e33 Michael Hanselmann
  cnt = [opts.on_primary, opts.on_secondary, opts.auto,
729 7e9366f7 Iustin Pop
         new_2ndary is not None, iallocator is not None].count(True)
730 7e9366f7 Iustin Pop
  if cnt != 1:
731 d8d838cb Michael Hanselmann
    raise errors.OpPrereqError("One and only one of the -p, -s, -a, -n and -I"
732 debac808 Iustin Pop
                               " options must be passed", errors.ECODE_INVAL)
733 7e9366f7 Iustin Pop
  elif opts.on_primary:
734 a9e0c397 Iustin Pop
    mode = constants.REPLACE_DISK_PRI
735 7e9366f7 Iustin Pop
  elif opts.on_secondary:
736 a9e0c397 Iustin Pop
    mode = constants.REPLACE_DISK_SEC
737 05d47e33 Michael Hanselmann
  elif opts.auto:
738 05d47e33 Michael Hanselmann
    mode = constants.REPLACE_DISK_AUTO
739 05d47e33 Michael Hanselmann
    if disks:
740 05d47e33 Michael Hanselmann
      raise errors.OpPrereqError("Cannot specify disks when using automatic"
741 debac808 Iustin Pop
                                 " mode", errors.ECODE_INVAL)
742 7e9366f7 Iustin Pop
  elif new_2ndary is not None or iallocator is not None:
743 7e9366f7 Iustin Pop
    # replace secondary
744 7e9366f7 Iustin Pop
    mode = constants.REPLACE_DISK_CHG
745 a9e0c397 Iustin Pop
746 668f755d Iustin Pop
  op = opcodes.OpInstanceReplaceDisks(instance_name=args[0], disks=disks,
747 668f755d Iustin Pop
                                      remote_node=new_2ndary, mode=mode,
748 668f755d Iustin Pop
                                      iallocator=iallocator,
749 668f755d Iustin Pop
                                      early_release=opts.early_release)
750 6340bb0a Iustin Pop
  SubmitOrSend(op, opts)
751 a8083063 Iustin Pop
  return 0
752 a8083063 Iustin Pop
753 a8083063 Iustin Pop
754 a8083063 Iustin Pop
def FailoverInstance(opts, args):
755 a8083063 Iustin Pop
  """Failover an instance.
756 a8083063 Iustin Pop

757 a8083063 Iustin Pop
  The failover is done by shutting it down on its present node and
758 a8083063 Iustin Pop
  starting it on the secondary.
759 a8083063 Iustin Pop

760 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
761 7232c04c Iustin Pop
  @type args: list
762 7232c04c Iustin Pop
  @param args: should contain only one element, the instance name
763 7232c04c Iustin Pop
  @rtype: int
764 7232c04c Iustin Pop
  @return: the desired exit code
765 a8083063 Iustin Pop

766 a8083063 Iustin Pop
  """
767 a76f0c4a Iustin Pop
  cl = GetClient()
768 80de0e3f Iustin Pop
  instance_name = args[0]
769 80de0e3f Iustin Pop
  force = opts.force
770 1b7761fd Apollon Oikonomopoulos
  iallocator = opts.iallocator
771 1b7761fd Apollon Oikonomopoulos
  target_node = opts.dst_node
772 1b7761fd Apollon Oikonomopoulos
773 1b7761fd Apollon Oikonomopoulos
  if iallocator and target_node:
774 1b7761fd Apollon Oikonomopoulos
    raise errors.OpPrereqError("Specify either an iallocator (-I), or a target"
775 1b7761fd Apollon Oikonomopoulos
                               " node (-n) but not both", errors.ECODE_INVAL)
776 a8083063 Iustin Pop
777 80de0e3f Iustin Pop
  if not force:
778 a76f0c4a Iustin Pop
    _EnsureInstancesExist(cl, [instance_name])
779 a76f0c4a Iustin Pop
780 80de0e3f Iustin Pop
    usertext = ("Failover will happen to image %s."
781 80de0e3f Iustin Pop
                " This requires a shutdown of the instance. Continue?" %
782 80de0e3f Iustin Pop
                (instance_name,))
783 80de0e3f Iustin Pop
    if not AskUser(usertext):
784 80de0e3f Iustin Pop
      return 1
785 a8083063 Iustin Pop
786 019dbee1 Iustin Pop
  op = opcodes.OpInstanceFailover(instance_name=instance_name,
787 17c3f802 Guido Trotter
                                  ignore_consistency=opts.ignore_consistency,
788 1b7761fd Apollon Oikonomopoulos
                                  shutdown_timeout=opts.shutdown_timeout,
789 1b7761fd Apollon Oikonomopoulos
                                  iallocator=iallocator,
790 1b7761fd Apollon Oikonomopoulos
                                  target_node=target_node)
791 a76f0c4a Iustin Pop
  SubmitOrSend(op, opts, cl=cl)
792 80de0e3f Iustin Pop
  return 0
793 a8083063 Iustin Pop
794 a8083063 Iustin Pop
795 53c776b5 Iustin Pop
def MigrateInstance(opts, args):
796 53c776b5 Iustin Pop
  """Migrate an instance.
797 53c776b5 Iustin Pop

798 53c776b5 Iustin Pop
  The migrate is done without shutdown.
799 53c776b5 Iustin Pop

800 2f907a8c Iustin Pop
  @param opts: the command line options selected by the user
801 2f907a8c Iustin Pop
  @type args: list
802 2f907a8c Iustin Pop
  @param args: should contain only one element, the instance name
803 2f907a8c Iustin Pop
  @rtype: int
804 2f907a8c Iustin Pop
  @return: the desired exit code
805 53c776b5 Iustin Pop

806 53c776b5 Iustin Pop
  """
807 a76f0c4a Iustin Pop
  cl = GetClient()
808 53c776b5 Iustin Pop
  instance_name = args[0]
809 53c776b5 Iustin Pop
  force = opts.force
810 1b7761fd Apollon Oikonomopoulos
  iallocator = opts.iallocator
811 1b7761fd Apollon Oikonomopoulos
  target_node = opts.dst_node
812 1b7761fd Apollon Oikonomopoulos
813 1b7761fd Apollon Oikonomopoulos
  if iallocator and target_node:
814 1b7761fd Apollon Oikonomopoulos
    raise errors.OpPrereqError("Specify either an iallocator (-I), or a target"
815 1b7761fd Apollon Oikonomopoulos
                               " node (-n) but not both", errors.ECODE_INVAL)
816 53c776b5 Iustin Pop
817 53c776b5 Iustin Pop
  if not force:
818 a76f0c4a Iustin Pop
    _EnsureInstancesExist(cl, [instance_name])
819 a76f0c4a Iustin Pop
820 53c776b5 Iustin Pop
    if opts.cleanup:
821 53c776b5 Iustin Pop
      usertext = ("Instance %s will be recovered from a failed migration."
822 53c776b5 Iustin Pop
                  " Note that the migration procedure (including cleanup)" %
823 53c776b5 Iustin Pop
                  (instance_name,))
824 53c776b5 Iustin Pop
    else:
825 53c776b5 Iustin Pop
      usertext = ("Instance %s will be migrated. Note that migration" %
826 53c776b5 Iustin Pop
                  (instance_name,))
827 cf29cfb6 Iustin Pop
    usertext += (" might impact the instance if anything goes wrong"
828 cf29cfb6 Iustin Pop
                 " (e.g. due to bugs in the hypervisor). Continue?")
829 53c776b5 Iustin Pop
    if not AskUser(usertext):
830 53c776b5 Iustin Pop
      return 1
831 53c776b5 Iustin Pop
832 e71b9ef4 Iustin Pop
  # this should be removed once --non-live is deprecated
833 783a6c0b Iustin Pop
  if not opts.live and opts.migration_mode is not None:
834 e71b9ef4 Iustin Pop
    raise errors.OpPrereqError("Only one of the --non-live and "
835 783a6c0b Iustin Pop
                               "--migration-mode options can be passed",
836 e71b9ef4 Iustin Pop
                               errors.ECODE_INVAL)
837 e71b9ef4 Iustin Pop
  if not opts.live: # --non-live passed
838 8c35561f Iustin Pop
    mode = constants.HT_MIGRATION_NONLIVE
839 e71b9ef4 Iustin Pop
  else:
840 8c35561f Iustin Pop
    mode = opts.migration_mode
841 e71b9ef4 Iustin Pop
842 75c866c2 Iustin Pop
  op = opcodes.OpInstanceMigrate(instance_name=instance_name, mode=mode,
843 1b7761fd Apollon Oikonomopoulos
                                 cleanup=opts.cleanup, iallocator=iallocator,
844 e9c487be René Nussbaumer
                                 target_node=target_node,
845 e9c487be René Nussbaumer
                                 allow_failover=opts.allow_failover)
846 400ca2f7 Iustin Pop
  SubmitOpCode(op, cl=cl, opts=opts)
847 53c776b5 Iustin Pop
  return 0
848 53c776b5 Iustin Pop
849 53c776b5 Iustin Pop
850 fbf5a861 Iustin Pop
def MoveInstance(opts, args):
851 fbf5a861 Iustin Pop
  """Move an instance.
852 fbf5a861 Iustin Pop

853 fbf5a861 Iustin Pop
  @param opts: the command line options selected by the user
854 fbf5a861 Iustin Pop
  @type args: list
855 fbf5a861 Iustin Pop
  @param args: should contain only one element, the instance name
856 fbf5a861 Iustin Pop
  @rtype: int
857 fbf5a861 Iustin Pop
  @return: the desired exit code
858 fbf5a861 Iustin Pop

859 fbf5a861 Iustin Pop
  """
860 fbf5a861 Iustin Pop
  cl = GetClient()
861 fbf5a861 Iustin Pop
  instance_name = args[0]
862 fbf5a861 Iustin Pop
  force = opts.force
863 fbf5a861 Iustin Pop
864 fbf5a861 Iustin Pop
  if not force:
865 fbf5a861 Iustin Pop
    usertext = ("Instance %s will be moved."
866 fbf5a861 Iustin Pop
                " This requires a shutdown of the instance. Continue?" %
867 fbf5a861 Iustin Pop
                (instance_name,))
868 fbf5a861 Iustin Pop
    if not AskUser(usertext):
869 fbf5a861 Iustin Pop
      return 1
870 fbf5a861 Iustin Pop
871 0091b480 Iustin Pop
  op = opcodes.OpInstanceMove(instance_name=instance_name,
872 17c3f802 Guido Trotter
                              target_node=opts.node,
873 bb851c63 Iustin Pop
                              shutdown_timeout=opts.shutdown_timeout,
874 bb851c63 Iustin Pop
                              ignore_consistency=opts.ignore_consistency)
875 fbf5a861 Iustin Pop
  SubmitOrSend(op, opts, cl=cl)
876 fbf5a861 Iustin Pop
  return 0
877 fbf5a861 Iustin Pop
878 fbf5a861 Iustin Pop
879 a8083063 Iustin Pop
def ConnectToInstanceConsole(opts, args):
880 a8083063 Iustin Pop
  """Connect to the console of an instance.
881 a8083063 Iustin Pop

882 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
883 7232c04c Iustin Pop
  @type args: list
884 7232c04c Iustin Pop
  @param args: should contain only one element, the instance name
885 7232c04c Iustin Pop
  @rtype: int
886 7232c04c Iustin Pop
  @return: the desired exit code
887 a8083063 Iustin Pop

888 a8083063 Iustin Pop
  """
889 a8083063 Iustin Pop
  instance_name = args[0]
890 a8083063 Iustin Pop
891 cc0dec7b Iustin Pop
  op = opcodes.OpInstanceConsole(instance_name=instance_name)
892 51c6e7b5 Michael Hanselmann
893 25ce3ec4 Michael Hanselmann
  cl = GetClient()
894 25ce3ec4 Michael Hanselmann
  try:
895 25ce3ec4 Michael Hanselmann
    cluster_name = cl.QueryConfigValues(["cluster_name"])[0]
896 25ce3ec4 Michael Hanselmann
    console_data = SubmitOpCode(op, opts=opts, cl=cl)
897 25ce3ec4 Michael Hanselmann
  finally:
898 25ce3ec4 Michael Hanselmann
    # Ensure client connection is closed while external commands are run
899 25ce3ec4 Michael Hanselmann
    cl.Close()
900 25ce3ec4 Michael Hanselmann
901 25ce3ec4 Michael Hanselmann
  del cl
902 25ce3ec4 Michael Hanselmann
903 25ce3ec4 Michael Hanselmann
  return _DoConsole(objects.InstanceConsole.FromDict(console_data),
904 25ce3ec4 Michael Hanselmann
                    opts.show_command, cluster_name)
905 25ce3ec4 Michael Hanselmann
906 25ce3ec4 Michael Hanselmann
907 25ce3ec4 Michael Hanselmann
def _DoConsole(console, show_command, cluster_name, feedback_fn=ToStdout,
908 25ce3ec4 Michael Hanselmann
               _runcmd_fn=utils.RunCmd):
909 cc0dec7b Iustin Pop
  """Acts based on the result of L{opcodes.OpInstanceConsole}.
910 25ce3ec4 Michael Hanselmann

911 25ce3ec4 Michael Hanselmann
  @type console: L{objects.InstanceConsole}
912 25ce3ec4 Michael Hanselmann
  @param console: Console object
913 25ce3ec4 Michael Hanselmann
  @type show_command: bool
914 25ce3ec4 Michael Hanselmann
  @param show_command: Whether to just display commands
915 25ce3ec4 Michael Hanselmann
  @type cluster_name: string
916 25ce3ec4 Michael Hanselmann
  @param cluster_name: Cluster name as retrieved from master daemon
917 25ce3ec4 Michael Hanselmann

918 25ce3ec4 Michael Hanselmann
  """
919 25ce3ec4 Michael Hanselmann
  assert console.Validate()
920 25ce3ec4 Michael Hanselmann
921 25ce3ec4 Michael Hanselmann
  if console.kind == constants.CONS_MESSAGE:
922 25ce3ec4 Michael Hanselmann
    feedback_fn(console.message)
923 25ce3ec4 Michael Hanselmann
  elif console.kind == constants.CONS_VNC:
924 25ce3ec4 Michael Hanselmann
    feedback_fn("Instance %s has VNC listening on %s:%s (display %s),"
925 25ce3ec4 Michael Hanselmann
                " URL <vnc://%s:%s/>",
926 25ce3ec4 Michael Hanselmann
                console.instance, console.host, console.port,
927 25ce3ec4 Michael Hanselmann
                console.display, console.host, console.port)
928 25ce3ec4 Michael Hanselmann
  elif console.kind == constants.CONS_SSH:
929 25ce3ec4 Michael Hanselmann
    # Convert to string if not already one
930 25ce3ec4 Michael Hanselmann
    if isinstance(console.command, basestring):
931 25ce3ec4 Michael Hanselmann
      cmd = console.command
932 25ce3ec4 Michael Hanselmann
    else:
933 25ce3ec4 Michael Hanselmann
      cmd = utils.ShellQuoteArgs(console.command)
934 25ce3ec4 Michael Hanselmann
935 25ce3ec4 Michael Hanselmann
    srun = ssh.SshRunner(cluster_name=cluster_name)
936 25ce3ec4 Michael Hanselmann
    ssh_cmd = srun.BuildCmd(console.host, console.user, cmd,
937 25ce3ec4 Michael Hanselmann
                            batch=True, quiet=False, tty=True)
938 25ce3ec4 Michael Hanselmann
939 25ce3ec4 Michael Hanselmann
    if show_command:
940 25ce3ec4 Michael Hanselmann
      feedback_fn(utils.ShellQuoteArgs(ssh_cmd))
941 25ce3ec4 Michael Hanselmann
    else:
942 25ce3ec4 Michael Hanselmann
      result = _runcmd_fn(ssh_cmd, interactive=True)
943 25ce3ec4 Michael Hanselmann
      if result.failed:
944 25ce3ec4 Michael Hanselmann
        logging.error("Console command \"%s\" failed with reason '%s' and"
945 25ce3ec4 Michael Hanselmann
                      " output %r", result.cmd, result.fail_reason,
946 25ce3ec4 Michael Hanselmann
                      result.output)
947 25ce3ec4 Michael Hanselmann
        raise errors.OpExecError("Connection to console of instance %s failed,"
948 25ce3ec4 Michael Hanselmann
                                 " please check cluster configuration" %
949 25ce3ec4 Michael Hanselmann
                                 console.instance)
950 51c6e7b5 Michael Hanselmann
  else:
951 25ce3ec4 Michael Hanselmann
    raise errors.GenericError("Unknown console type '%s'" % console.kind)
952 678aa6d3 Michael Hanselmann
953 678aa6d3 Michael Hanselmann
  return constants.EXIT_SUCCESS
954 a8083063 Iustin Pop
955 a8083063 Iustin Pop
956 e2736e40 Guido Trotter
def _FormatLogicalID(dev_type, logical_id, roman):
957 19708787 Iustin Pop
  """Formats the logical_id of a disk.
958 19708787 Iustin Pop

959 19708787 Iustin Pop
  """
960 19708787 Iustin Pop
  if dev_type == constants.LD_DRBD8:
961 19708787 Iustin Pop
    node_a, node_b, port, minor_a, minor_b, key = logical_id
962 19708787 Iustin Pop
    data = [
963 e2736e40 Guido Trotter
      ("nodeA", "%s, minor=%s" % (node_a, compat.TryToRoman(minor_a,
964 e2736e40 Guido Trotter
                                                            convert=roman))),
965 e2736e40 Guido Trotter
      ("nodeB", "%s, minor=%s" % (node_b, compat.TryToRoman(minor_b,
966 e2736e40 Guido Trotter
                                                            convert=roman))),
967 e2736e40 Guido Trotter
      ("port", compat.TryToRoman(port, convert=roman)),
968 19708787 Iustin Pop
      ("auth key", key),
969 19708787 Iustin Pop
      ]
970 19708787 Iustin Pop
  elif dev_type == constants.LD_LV:
971 19708787 Iustin Pop
    vg_name, lv_name = logical_id
972 19708787 Iustin Pop
    data = ["%s/%s" % (vg_name, lv_name)]
973 19708787 Iustin Pop
  else:
974 19708787 Iustin Pop
    data = [str(logical_id)]
975 19708787 Iustin Pop
976 19708787 Iustin Pop
  return data
977 19708787 Iustin Pop
978 19708787 Iustin Pop
979 e2736e40 Guido Trotter
def _FormatBlockDevInfo(idx, top_level, dev, static, roman):
980 a8083063 Iustin Pop
  """Show block device information.
981 a8083063 Iustin Pop

982 7232c04c Iustin Pop
  This is only used by L{ShowInstanceConfig}, but it's too big to be
983 a8083063 Iustin Pop
  left for an inline definition.
984 a8083063 Iustin Pop

985 19708787 Iustin Pop
  @type idx: int
986 19708787 Iustin Pop
  @param idx: the index of the current disk
987 19708787 Iustin Pop
  @type top_level: boolean
988 19708787 Iustin Pop
  @param top_level: if this a top-level disk?
989 7232c04c Iustin Pop
  @type dev: dict
990 7232c04c Iustin Pop
  @param dev: dictionary with disk information
991 7232c04c Iustin Pop
  @type static: boolean
992 7232c04c Iustin Pop
  @param static: wheter the device information doesn't contain
993 7232c04c Iustin Pop
      runtime information but only static data
994 e2736e40 Guido Trotter
  @type roman: boolean
995 e2736e40 Guido Trotter
  @param roman: whether to try to use roman integers
996 19708787 Iustin Pop
  @return: a list of either strings, tuples or lists
997 19708787 Iustin Pop
      (which should be formatted at a higher indent level)
998 7232c04c Iustin Pop

999 a8083063 Iustin Pop
  """
1000 19708787 Iustin Pop
  def helper(dtype, status):
1001 7232c04c Iustin Pop
    """Format one line for physical device status.
1002 7232c04c Iustin Pop

1003 7232c04c Iustin Pop
    @type dtype: str
1004 7232c04c Iustin Pop
    @param dtype: a constant from the L{constants.LDS_BLOCK} set
1005 7232c04c Iustin Pop
    @type status: tuple
1006 7232c04c Iustin Pop
    @param status: a tuple as returned from L{backend.FindBlockDevice}
1007 19708787 Iustin Pop
    @return: the string representing the status
1008 7232c04c Iustin Pop

1009 7232c04c Iustin Pop
    """
1010 a8083063 Iustin Pop
    if not status:
1011 19708787 Iustin Pop
      return "not active"
1012 19708787 Iustin Pop
    txt = ""
1013 f208978a Michael Hanselmann
    (path, major, minor, syncp, estt, degr, ldisk_status) = status
1014 19708787 Iustin Pop
    if major is None:
1015 19708787 Iustin Pop
      major_string = "N/A"
1016 a8083063 Iustin Pop
    else:
1017 e2736e40 Guido Trotter
      major_string = str(compat.TryToRoman(major, convert=roman))
1018 fd38ef95 Manuel Franceschini
1019 19708787 Iustin Pop
    if minor is None:
1020 19708787 Iustin Pop
      minor_string = "N/A"
1021 19708787 Iustin Pop
    else:
1022 e2736e40 Guido Trotter
      minor_string = str(compat.TryToRoman(minor, convert=roman))
1023 19708787 Iustin Pop
1024 19708787 Iustin Pop
    txt += ("%s (%s:%s)" % (path, major_string, minor_string))
1025 19708787 Iustin Pop
    if dtype in (constants.LD_DRBD8, ):
1026 19708787 Iustin Pop
      if syncp is not None:
1027 19708787 Iustin Pop
        sync_text = "*RECOVERING* %5.2f%%," % syncp
1028 19708787 Iustin Pop
        if estt:
1029 e2736e40 Guido Trotter
          sync_text += " ETA %ss" % compat.TryToRoman(estt, convert=roman)
1030 9db6dbce Iustin Pop
        else:
1031 19708787 Iustin Pop
          sync_text += " ETA unknown"
1032 19708787 Iustin Pop
      else:
1033 19708787 Iustin Pop
        sync_text = "in sync"
1034 19708787 Iustin Pop
      if degr:
1035 19708787 Iustin Pop
        degr_text = "*DEGRADED*"
1036 19708787 Iustin Pop
      else:
1037 19708787 Iustin Pop
        degr_text = "ok"
1038 f208978a Michael Hanselmann
      if ldisk_status == constants.LDS_FAULTY:
1039 19708787 Iustin Pop
        ldisk_text = " *MISSING DISK*"
1040 f208978a Michael Hanselmann
      elif ldisk_status == constants.LDS_UNKNOWN:
1041 f208978a Michael Hanselmann
        ldisk_text = " *UNCERTAIN STATE*"
1042 19708787 Iustin Pop
      else:
1043 19708787 Iustin Pop
        ldisk_text = ""
1044 19708787 Iustin Pop
      txt += (" %s, status %s%s" % (sync_text, degr_text, ldisk_text))
1045 19708787 Iustin Pop
    elif dtype == constants.LD_LV:
1046 f208978a Michael Hanselmann
      if ldisk_status == constants.LDS_FAULTY:
1047 19708787 Iustin Pop
        ldisk_text = " *FAILED* (failed drive?)"
1048 19708787 Iustin Pop
      else:
1049 19708787 Iustin Pop
        ldisk_text = ""
1050 19708787 Iustin Pop
      txt += ldisk_text
1051 19708787 Iustin Pop
    return txt
1052 19708787 Iustin Pop
1053 19708787 Iustin Pop
  # the header
1054 19708787 Iustin Pop
  if top_level:
1055 19708787 Iustin Pop
    if dev["iv_name"] is not None:
1056 19708787 Iustin Pop
      txt = dev["iv_name"]
1057 19708787 Iustin Pop
    else:
1058 e2736e40 Guido Trotter
      txt = "disk %s" % compat.TryToRoman(idx, convert=roman)
1059 a8083063 Iustin Pop
  else:
1060 e2736e40 Guido Trotter
    txt = "child %s" % compat.TryToRoman(idx, convert=roman)
1061 c98162a7 Iustin Pop
  if isinstance(dev["size"], int):
1062 c98162a7 Iustin Pop
    nice_size = utils.FormatUnit(dev["size"], "h")
1063 c98162a7 Iustin Pop
  else:
1064 c98162a7 Iustin Pop
    nice_size = dev["size"]
1065 c98162a7 Iustin Pop
  d1 = ["- %s: %s, size %s" % (txt, dev["dev_type"], nice_size)]
1066 19708787 Iustin Pop
  data = []
1067 19708787 Iustin Pop
  if top_level:
1068 19708787 Iustin Pop
    data.append(("access mode", dev["mode"]))
1069 a8083063 Iustin Pop
  if dev["logical_id"] is not None:
1070 19708787 Iustin Pop
    try:
1071 e2736e40 Guido Trotter
      l_id = _FormatLogicalID(dev["dev_type"], dev["logical_id"], roman)
1072 19708787 Iustin Pop
    except ValueError:
1073 19708787 Iustin Pop
      l_id = [str(dev["logical_id"])]
1074 19708787 Iustin Pop
    if len(l_id) == 1:
1075 19708787 Iustin Pop
      data.append(("logical_id", l_id[0]))
1076 19708787 Iustin Pop
    else:
1077 19708787 Iustin Pop
      data.extend(l_id)
1078 a8083063 Iustin Pop
  elif dev["physical_id"] is not None:
1079 19708787 Iustin Pop
    data.append("physical_id:")
1080 19708787 Iustin Pop
    data.append([dev["physical_id"]])
1081 57821cac Iustin Pop
  if not static:
1082 19708787 Iustin Pop
    data.append(("on primary", helper(dev["dev_type"], dev["pstatus"])))
1083 57821cac Iustin Pop
  if dev["sstatus"] and not static:
1084 19708787 Iustin Pop
    data.append(("on secondary", helper(dev["dev_type"], dev["sstatus"])))
1085 a8083063 Iustin Pop
1086 a8083063 Iustin Pop
  if dev["children"]:
1087 19708787 Iustin Pop
    data.append("child devices:")
1088 19708787 Iustin Pop
    for c_idx, child in enumerate(dev["children"]):
1089 e2736e40 Guido Trotter
      data.append(_FormatBlockDevInfo(c_idx, False, child, static, roman))
1090 19708787 Iustin Pop
  d1.append(data)
1091 19708787 Iustin Pop
  return d1
1092 a8083063 Iustin Pop
1093 a8083063 Iustin Pop
1094 19708787 Iustin Pop
def _FormatList(buf, data, indent_level):
1095 19708787 Iustin Pop
  """Formats a list of data at a given indent level.
1096 19708787 Iustin Pop

1097 19708787 Iustin Pop
  If the element of the list is:
1098 19708787 Iustin Pop
    - a string, it is simply formatted as is
1099 19708787 Iustin Pop
    - a tuple, it will be split into key, value and the all the
1100 19708787 Iustin Pop
      values in a list will be aligned all at the same start column
1101 19708787 Iustin Pop
    - a list, will be recursively formatted
1102 19708787 Iustin Pop

1103 19708787 Iustin Pop
  @type buf: StringIO
1104 19708787 Iustin Pop
  @param buf: the buffer into which we write the output
1105 19708787 Iustin Pop
  @param data: the list to format
1106 19708787 Iustin Pop
  @type indent_level: int
1107 19708787 Iustin Pop
  @param indent_level: the indent level to format at
1108 19708787 Iustin Pop

1109 19708787 Iustin Pop
  """
1110 19708787 Iustin Pop
  max_tlen = max([len(elem[0]) for elem in data
1111 19708787 Iustin Pop
                 if isinstance(elem, tuple)] or [0])
1112 19708787 Iustin Pop
  for elem in data:
1113 19708787 Iustin Pop
    if isinstance(elem, basestring):
1114 19708787 Iustin Pop
      buf.write("%*s%s\n" % (2*indent_level, "", elem))
1115 19708787 Iustin Pop
    elif isinstance(elem, tuple):
1116 19708787 Iustin Pop
      key, value = elem
1117 19708787 Iustin Pop
      spacer = "%*s" % (max_tlen - len(key), "")
1118 19708787 Iustin Pop
      buf.write("%*s%s:%s %s\n" % (2*indent_level, "", key, spacer, value))
1119 19708787 Iustin Pop
    elif isinstance(elem, list):
1120 19708787 Iustin Pop
      _FormatList(buf, elem, indent_level+1)
1121 19708787 Iustin Pop
1122 98825740 Michael Hanselmann
1123 a8083063 Iustin Pop
def ShowInstanceConfig(opts, args):
1124 a8083063 Iustin Pop
  """Compute instance run-time status.
1125 a8083063 Iustin Pop

1126 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
1127 7232c04c Iustin Pop
  @type args: list
1128 7232c04c Iustin Pop
  @param args: either an empty list, and then we query all
1129 7232c04c Iustin Pop
      instances, or should contain a list of instance names
1130 7232c04c Iustin Pop
  @rtype: int
1131 7232c04c Iustin Pop
  @return: the desired exit code
1132 7232c04c Iustin Pop

1133 a8083063 Iustin Pop
  """
1134 220cde0b Guido Trotter
  if not args and not opts.show_all:
1135 220cde0b Guido Trotter
    ToStderr("No instance selected."
1136 220cde0b Guido Trotter
             " Please pass in --all if you want to query all instances.\n"
1137 220cde0b Guido Trotter
             "Note that this can take a long time on a big cluster.")
1138 220cde0b Guido Trotter
    return 1
1139 220cde0b Guido Trotter
  elif args and opts.show_all:
1140 220cde0b Guido Trotter
    ToStderr("Cannot use --all if you specify instance names.")
1141 220cde0b Guido Trotter
    return 1
1142 220cde0b Guido Trotter
1143 a8083063 Iustin Pop
  retcode = 0
1144 5c097318 Iustin Pop
  op = opcodes.OpInstanceQueryData(instances=args, static=opts.static,
1145 5c097318 Iustin Pop
                                   use_locking=not opts.static)
1146 400ca2f7 Iustin Pop
  result = SubmitOpCode(op, opts=opts)
1147 a8083063 Iustin Pop
  if not result:
1148 3a24c527 Iustin Pop
    ToStdout("No instances.")
1149 a8083063 Iustin Pop
    return 1
1150 a8083063 Iustin Pop
1151 a8083063 Iustin Pop
  buf = StringIO()
1152 a8083063 Iustin Pop
  retcode = 0
1153 a8083063 Iustin Pop
  for instance_name in result:
1154 a8083063 Iustin Pop
    instance = result[instance_name]
1155 a8083063 Iustin Pop
    buf.write("Instance name: %s\n" % instance["name"])
1156 033d58b0 Iustin Pop
    buf.write("UUID: %s\n" % instance["uuid"])
1157 e2736e40 Guido Trotter
    buf.write("Serial number: %s\n" %
1158 e2736e40 Guido Trotter
              compat.TryToRoman(instance["serial_no"],
1159 e2736e40 Guido Trotter
                                convert=opts.roman_integers))
1160 90f72445 Iustin Pop
    buf.write("Creation time: %s\n" % utils.FormatTime(instance["ctime"]))
1161 90f72445 Iustin Pop
    buf.write("Modification time: %s\n" % utils.FormatTime(instance["mtime"]))
1162 57821cac Iustin Pop
    buf.write("State: configured to be %s" % instance["config_state"])
1163 57821cac Iustin Pop
    if not opts.static:
1164 57821cac Iustin Pop
      buf.write(", actual state is %s" % instance["run_state"])
1165 57821cac Iustin Pop
    buf.write("\n")
1166 57821cac Iustin Pop
    ##buf.write("Considered for memory checks in cluster verify: %s\n" %
1167 57821cac Iustin Pop
    ##          instance["auto_balance"])
1168 a8083063 Iustin Pop
    buf.write("  Nodes:\n")
1169 a8083063 Iustin Pop
    buf.write("    - primary: %s\n" % instance["pnode"])
1170 1f864b60 Iustin Pop
    buf.write("    - secondaries: %s\n" % utils.CommaJoin(instance["snodes"]))
1171 a8083063 Iustin Pop
    buf.write("  Operating system: %s\n" % instance["os"])
1172 acd19189 René Nussbaumer
    FormatParameterDict(buf, instance["os_instance"], instance["os_actual"],
1173 acd19189 René Nussbaumer
                        level=2)
1174 a8340917 Iustin Pop
    if instance.has_key("network_port"):
1175 e2736e40 Guido Trotter
      buf.write("  Allocated network port: %s\n" %
1176 e2736e40 Guido Trotter
                compat.TryToRoman(instance["network_port"],
1177 e2736e40 Guido Trotter
                                  convert=opts.roman_integers))
1178 24838135 Iustin Pop
    buf.write("  Hypervisor: %s\n" % instance["hypervisor"])
1179 dfff41f8 Guido Trotter
1180 dfff41f8 Guido Trotter
    # custom VNC console information
1181 dfff41f8 Guido Trotter
    vnc_bind_address = instance["hv_actual"].get(constants.HV_VNC_BIND_ADDRESS,
1182 dfff41f8 Guido Trotter
                                                 None)
1183 dfff41f8 Guido Trotter
    if vnc_bind_address:
1184 dfff41f8 Guido Trotter
      port = instance["network_port"]
1185 dfff41f8 Guido Trotter
      display = int(port) - constants.VNC_BASE_PORT
1186 9769bb78 Manuel Franceschini
      if display > 0 and vnc_bind_address == constants.IP4_ADDRESS_ANY:
1187 dfff41f8 Guido Trotter
        vnc_console_port = "%s:%s (display %s)" % (instance["pnode"],
1188 dfff41f8 Guido Trotter
                                                   port,
1189 dfff41f8 Guido Trotter
                                                   display)
1190 8b312c1d Manuel Franceschini
      elif display > 0 and netutils.IP4Address.IsValid(vnc_bind_address):
1191 dfff41f8 Guido Trotter
        vnc_console_port = ("%s:%s (node %s) (display %s)" %
1192 dfff41f8 Guido Trotter
                             (vnc_bind_address, port,
1193 dfff41f8 Guido Trotter
                              instance["pnode"], display))
1194 a8340917 Iustin Pop
      else:
1195 dfff41f8 Guido Trotter
        # vnc bind address is a file
1196 dfff41f8 Guido Trotter
        vnc_console_port = "%s:%s" % (instance["pnode"],
1197 dfff41f8 Guido Trotter
                                      vnc_bind_address)
1198 24838135 Iustin Pop
      buf.write("    - console connection: vnc to %s\n" % vnc_console_port)
1199 24838135 Iustin Pop
1200 acd19189 René Nussbaumer
    FormatParameterDict(buf, instance["hv_instance"], instance["hv_actual"],
1201 acd19189 René Nussbaumer
                        level=2)
1202 a8083063 Iustin Pop
    buf.write("  Hardware:\n")
1203 e2736e40 Guido Trotter
    buf.write("    - VCPUs: %s\n" %
1204 e2736e40 Guido Trotter
              compat.TryToRoman(instance["be_actual"][constants.BE_VCPUS],
1205 e2736e40 Guido Trotter
                                convert=opts.roman_integers))
1206 e2736e40 Guido Trotter
    buf.write("    - memory: %sMiB\n" %
1207 e2736e40 Guido Trotter
              compat.TryToRoman(instance["be_actual"][constants.BE_MEMORY],
1208 e2736e40 Guido Trotter
                                convert=opts.roman_integers))
1209 d2acfe27 Iustin Pop
    buf.write("    - NICs:\n")
1210 14ea9302 Guido Trotter
    for idx, (ip, mac, mode, link) in enumerate(instance["nics"]):
1211 0b13832c Guido Trotter
      buf.write("      - nic/%d: MAC: %s, IP: %s, mode: %s, link: %s\n" %
1212 0b13832c Guido Trotter
                (idx, mac, ip, mode, link))
1213 b577dac4 Michael Hanselmann
    buf.write("  Disk template: %s\n" % instance["disk_template"])
1214 19708787 Iustin Pop
    buf.write("  Disks:\n")
1215 a8083063 Iustin Pop
1216 19708787 Iustin Pop
    for idx, device in enumerate(instance["disks"]):
1217 e2736e40 Guido Trotter
      _FormatList(buf, _FormatBlockDevInfo(idx, True, device, opts.static,
1218 e2736e40 Guido Trotter
                  opts.roman_integers), 2)
1219 a8083063 Iustin Pop
1220 3a24c527 Iustin Pop
  ToStdout(buf.getvalue().rstrip('\n'))
1221 a8083063 Iustin Pop
  return retcode
1222 a8083063 Iustin Pop
1223 a8083063 Iustin Pop
1224 7767bbf5 Manuel Franceschini
def SetInstanceParams(opts, args):
1225 a8083063 Iustin Pop
  """Modifies an instance.
1226 a8083063 Iustin Pop

1227 a8083063 Iustin Pop
  All parameters take effect only at the next restart of the instance.
1228 a8083063 Iustin Pop

1229 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
1230 7232c04c Iustin Pop
  @type args: list
1231 7232c04c Iustin Pop
  @param args: should contain only one element, the instance name
1232 7232c04c Iustin Pop
  @rtype: int
1233 7232c04c Iustin Pop
  @return: the desired exit code
1234 a8083063 Iustin Pop

1235 a8083063 Iustin Pop
  """
1236 e29e9550 Iustin Pop
  if not (opts.nics or opts.disks or opts.disk_template or
1237 1052d622 Iustin Pop
          opts.hvparams or opts.beparams or opts.os or opts.osparams):
1238 3a24c527 Iustin Pop
    ToStderr("Please give at least one of the parameters.")
1239 a8083063 Iustin Pop
    return 1
1240 a8083063 Iustin Pop
1241 467ae11e Guido Trotter
  for param in opts.beparams:
1242 e9d622bc Guido Trotter
    if isinstance(opts.beparams[param], basestring):
1243 e9d622bc Guido Trotter
      if opts.beparams[param].lower() == "default":
1244 e9d622bc Guido Trotter
        opts.beparams[param] = constants.VALUE_DEFAULT
1245 a5728081 Guido Trotter
1246 a5728081 Guido Trotter
  utils.ForceDictType(opts.beparams, constants.BES_PARAMETER_TYPES,
1247 a5728081 Guido Trotter
                      allowed_values=[constants.VALUE_DEFAULT])
1248 467ae11e Guido Trotter
1249 48f212d7 Iustin Pop
  for param in opts.hvparams:
1250 48f212d7 Iustin Pop
    if isinstance(opts.hvparams[param], basestring):
1251 48f212d7 Iustin Pop
      if opts.hvparams[param].lower() == "default":
1252 48f212d7 Iustin Pop
        opts.hvparams[param] = constants.VALUE_DEFAULT
1253 a5728081 Guido Trotter
1254 48f212d7 Iustin Pop
  utils.ForceDictType(opts.hvparams, constants.HVS_PARAMETER_TYPES,
1255 a5728081 Guido Trotter
                      allowed_values=[constants.VALUE_DEFAULT])
1256 61be6ba4 Iustin Pop
1257 24991749 Iustin Pop
  for idx, (nic_op, nic_dict) in enumerate(opts.nics):
1258 24991749 Iustin Pop
    try:
1259 24991749 Iustin Pop
      nic_op = int(nic_op)
1260 24991749 Iustin Pop
      opts.nics[idx] = (nic_op, nic_dict)
1261 691744c4 Iustin Pop
    except (TypeError, ValueError):
1262 24991749 Iustin Pop
      pass
1263 24991749 Iustin Pop
1264 24991749 Iustin Pop
  for idx, (disk_op, disk_dict) in enumerate(opts.disks):
1265 24991749 Iustin Pop
    try:
1266 24991749 Iustin Pop
      disk_op = int(disk_op)
1267 24991749 Iustin Pop
      opts.disks[idx] = (disk_op, disk_dict)
1268 691744c4 Iustin Pop
    except (TypeError, ValueError):
1269 24991749 Iustin Pop
      pass
1270 24991749 Iustin Pop
    if disk_op == constants.DDM_ADD:
1271 24991749 Iustin Pop
      if 'size' not in disk_dict:
1272 debac808 Iustin Pop
        raise errors.OpPrereqError("Missing required parameter 'size'",
1273 debac808 Iustin Pop
                                   errors.ECODE_INVAL)
1274 24991749 Iustin Pop
      disk_dict['size'] = utils.ParseUnit(disk_dict['size'])
1275 24991749 Iustin Pop
1276 e29e9550 Iustin Pop
  if (opts.disk_template and
1277 3429a076 Apollon Oikonomopoulos
      opts.disk_template in constants.DTS_INT_MIRROR and
1278 e29e9550 Iustin Pop
      not opts.node):
1279 e29e9550 Iustin Pop
    ToStderr("Changing the disk template to a mirrored one requires"
1280 e29e9550 Iustin Pop
             " specifying a secondary node")
1281 e29e9550 Iustin Pop
    return 1
1282 e29e9550 Iustin Pop
1283 9a3cc7ae Iustin Pop
  op = opcodes.OpInstanceSetParams(instance_name=args[0],
1284 24991749 Iustin Pop
                                   nics=opts.nics,
1285 24991749 Iustin Pop
                                   disks=opts.disks,
1286 e29e9550 Iustin Pop
                                   disk_template=opts.disk_template,
1287 e29e9550 Iustin Pop
                                   remote_node=opts.node,
1288 48f212d7 Iustin Pop
                                   hvparams=opts.hvparams,
1289 338e51e8 Iustin Pop
                                   beparams=opts.beparams,
1290 96b39bcc Iustin Pop
                                   os_name=opts.os,
1291 1052d622 Iustin Pop
                                   osparams=opts.osparams,
1292 96b39bcc Iustin Pop
                                   force_variant=opts.force_variant,
1293 456798ab Iustin Pop
                                   force=opts.force,
1294 456798ab Iustin Pop
                                   wait_for_sync=opts.wait_for_sync)
1295 31a853d2 Iustin Pop
1296 6340bb0a Iustin Pop
  # even if here we process the result, we allow submit only
1297 6340bb0a Iustin Pop
  result = SubmitOrSend(op, opts)
1298 a8083063 Iustin Pop
1299 a8083063 Iustin Pop
  if result:
1300 3a24c527 Iustin Pop
    ToStdout("Modified instance %s", args[0])
1301 a8083063 Iustin Pop
    for param, data in result:
1302 3a24c527 Iustin Pop
      ToStdout(" - %-5s -> %s", param, data)
1303 e29e9550 Iustin Pop
    ToStdout("Please don't forget that most parameters take effect"
1304 3a24c527 Iustin Pop
             " only at the next start of the instance.")
1305 a8083063 Iustin Pop
  return 0
1306 a8083063 Iustin Pop
1307 a8083063 Iustin Pop
1308 312ac745 Iustin Pop
# multi-instance selection options
1309 c38c44ad Michael Hanselmann
m_force_multi = cli_option("--force-multiple", dest="force_multi",
1310 c38c44ad Michael Hanselmann
                           help="Do not ask for confirmation when more than"
1311 c38c44ad Michael Hanselmann
                           " one instance is affected",
1312 c38c44ad Michael Hanselmann
                           action="store_true", default=False)
1313 804a1e8e Iustin Pop
1314 c38c44ad Michael Hanselmann
m_pri_node_opt = cli_option("--primary", dest="multi_mode",
1315 c38c44ad Michael Hanselmann
                            help="Filter by nodes (primary only)",
1316 c38c44ad Michael Hanselmann
                            const=_SHUTDOWN_NODES_PRI, action="store_const")
1317 312ac745 Iustin Pop
1318 c38c44ad Michael Hanselmann
m_sec_node_opt = cli_option("--secondary", dest="multi_mode",
1319 c38c44ad Michael Hanselmann
                            help="Filter by nodes (secondary only)",
1320 c38c44ad Michael Hanselmann
                            const=_SHUTDOWN_NODES_SEC, action="store_const")
1321 312ac745 Iustin Pop
1322 c38c44ad Michael Hanselmann
m_node_opt = cli_option("--node", dest="multi_mode",
1323 c38c44ad Michael Hanselmann
                        help="Filter by nodes (primary and secondary)",
1324 c38c44ad Michael Hanselmann
                        const=_SHUTDOWN_NODES_BOTH, action="store_const")
1325 312ac745 Iustin Pop
1326 c38c44ad Michael Hanselmann
m_clust_opt = cli_option("--all", dest="multi_mode",
1327 c38c44ad Michael Hanselmann
                         help="Select all instances in the cluster",
1328 c38c44ad Michael Hanselmann
                         const=_SHUTDOWN_CLUSTER, action="store_const")
1329 312ac745 Iustin Pop
1330 c38c44ad Michael Hanselmann
m_inst_opt = cli_option("--instance", dest="multi_mode",
1331 c38c44ad Michael Hanselmann
                        help="Filter by instance name [default]",
1332 c38c44ad Michael Hanselmann
                        const=_SHUTDOWN_INSTANCES, action="store_const")
1333 312ac745 Iustin Pop
1334 39dfd93e René Nussbaumer
m_node_tags_opt = cli_option("--node-tags", dest="multi_mode",
1335 39dfd93e René Nussbaumer
                             help="Filter by node tag",
1336 39dfd93e René Nussbaumer
                             const=_SHUTDOWN_NODES_BOTH_BY_TAGS,
1337 39dfd93e René Nussbaumer
                             action="store_const")
1338 39dfd93e René Nussbaumer
1339 39dfd93e René Nussbaumer
m_pri_node_tags_opt = cli_option("--pri-node-tags", dest="multi_mode",
1340 39dfd93e René Nussbaumer
                                 help="Filter by primary node tag",
1341 39dfd93e René Nussbaumer
                                 const=_SHUTDOWN_NODES_PRI_BY_TAGS,
1342 39dfd93e René Nussbaumer
                                 action="store_const")
1343 39dfd93e René Nussbaumer
1344 39dfd93e René Nussbaumer
m_sec_node_tags_opt = cli_option("--sec-node-tags", dest="multi_mode",
1345 39dfd93e René Nussbaumer
                                 help="Filter by secondary node tag",
1346 39dfd93e René Nussbaumer
                                 const=_SHUTDOWN_NODES_SEC_BY_TAGS,
1347 39dfd93e René Nussbaumer
                                 action="store_const")
1348 39dfd93e René Nussbaumer
1349 39dfd93e René Nussbaumer
m_inst_tags_opt = cli_option("--tags", dest="multi_mode",
1350 39dfd93e René Nussbaumer
                             help="Filter by instance tag",
1351 39dfd93e René Nussbaumer
                             const=_SHUTDOWN_INSTANCES_BY_TAGS,
1352 39dfd93e René Nussbaumer
                             action="store_const")
1353 312ac745 Iustin Pop
1354 a8083063 Iustin Pop
# this is defined separately due to readability only
1355 a8083063 Iustin Pop
add_opts = [
1356 064c21f8 Iustin Pop
  NOSTART_OPT,
1357 064c21f8 Iustin Pop
  OS_OPT,
1358 06073e85 Guido Trotter
  FORCE_VARIANT_OPT,
1359 25a8792c Iustin Pop
  NO_INSTALL_OPT,
1360 a8083063 Iustin Pop
  ]
1361 a8083063 Iustin Pop
1362 a8083063 Iustin Pop
commands = {
1363 6ea815cf Iustin Pop
  'add': (
1364 eb28ecf6 Guido Trotter
    AddInstance, [ArgHost(min=1, max=1)], COMMON_CREATE_OPTS + add_opts,
1365 6ea815cf Iustin Pop
    "[...] -t disk-type -n node[:secondary-node] -o os-type <name>",
1366 6ea815cf Iustin Pop
    "Creates and adds a new instance to the cluster"),
1367 6ea815cf Iustin Pop
  'batch-create': (
1368 aa06f8c6 Michael Hanselmann
    BatchCreate, [ArgFile(min=1, max=1)], [DRY_RUN_OPT, PRIORITY_OPT],
1369 6ea815cf Iustin Pop
    "<instances.json>",
1370 6ea815cf Iustin Pop
    "Create a bunch of instances based on specs in the file."),
1371 6ea815cf Iustin Pop
  'console': (
1372 6ea815cf Iustin Pop
    ConnectToInstanceConsole, ARGS_ONE_INSTANCE,
1373 aa06f8c6 Michael Hanselmann
    [SHOWCMD_OPT, PRIORITY_OPT],
1374 6ea815cf Iustin Pop
    "[--show-cmd] <instance>", "Opens a console on the specified instance"),
1375 6ea815cf Iustin Pop
  'failover': (
1376 6ea815cf Iustin Pop
    FailoverInstance, ARGS_ONE_INSTANCE,
1377 db5a8a2d Iustin Pop
    [FORCE_OPT, IGNORE_CONSIST_OPT, SUBMIT_OPT, SHUTDOWN_TIMEOUT_OPT,
1378 1b7761fd Apollon Oikonomopoulos
     DRY_RUN_OPT, PRIORITY_OPT, DST_NODE_OPT, IALLOCATOR_OPT],
1379 6ea815cf Iustin Pop
    "[-f] <instance>", "Stops the instance and starts it on the backup node,"
1380 1b7761fd Apollon Oikonomopoulos
    " using the remote mirror (only for mirrored instances)"),
1381 6ea815cf Iustin Pop
  'migrate': (
1382 6ea815cf Iustin Pop
    MigrateInstance, ARGS_ONE_INSTANCE,
1383 aa06f8c6 Michael Hanselmann
    [FORCE_OPT, NONLIVE_OPT, MIGRATION_MODE_OPT, CLEANUP_OPT, DRY_RUN_OPT,
1384 e9c487be René Nussbaumer
     PRIORITY_OPT, DST_NODE_OPT, IALLOCATOR_OPT, ALLOW_FAILOVER_OPT],
1385 6ea815cf Iustin Pop
    "[-f] <instance>", "Migrate instance to its secondary node"
1386 1b7761fd Apollon Oikonomopoulos
    " (only for mirrored instances)"),
1387 6ea815cf Iustin Pop
  'move': (
1388 6ea815cf Iustin Pop
    MoveInstance, ARGS_ONE_INSTANCE,
1389 db5a8a2d Iustin Pop
    [FORCE_OPT, SUBMIT_OPT, SINGLE_NODE_OPT, SHUTDOWN_TIMEOUT_OPT,
1390 bb851c63 Iustin Pop
     DRY_RUN_OPT, PRIORITY_OPT, IGNORE_CONSIST_OPT],
1391 6ea815cf Iustin Pop
    "[-f] <instance>", "Move instance to an arbitrary node"
1392 6ea815cf Iustin Pop
    " (only for instances of type file and lv)"),
1393 6ea815cf Iustin Pop
  'info': (
1394 6ea815cf Iustin Pop
    ShowInstanceConfig, ARGS_MANY_INSTANCES,
1395 aa06f8c6 Michael Hanselmann
    [STATIC_OPT, ALL_OPT, ROMAN_OPT, PRIORITY_OPT],
1396 6ea815cf Iustin Pop
    "[-s] {--all | <instance>...}",
1397 6ea815cf Iustin Pop
    "Show information on the specified instance(s)"),
1398 6ea815cf Iustin Pop
  'list': (
1399 6ea815cf Iustin Pop
    ListInstances, ARGS_MANY_INSTANCES,
1400 87e87959 Michael Hanselmann
    [NOHDR_OPT, SEP_OPT, USEUNITS_OPT, FIELDS_OPT, VERBOSE_OPT,
1401 87e87959 Michael Hanselmann
     FORCE_FILTER_OPT],
1402 6ea815cf Iustin Pop
    "[<instance>...]",
1403 b82c5ff5 Michael Hanselmann
    "Lists the instances and their status. The available fields can be shown"
1404 b82c5ff5 Michael Hanselmann
    " using the \"list-fields\" command (see the man page for details)."
1405 b82c5ff5 Michael Hanselmann
    " The default field list is (in order): %s." %
1406 b82c5ff5 Michael Hanselmann
    utils.CommaJoin(_LIST_DEF_FIELDS),
1407 6ea815cf Iustin Pop
    ),
1408 b82c5ff5 Michael Hanselmann
  "list-fields": (
1409 b82c5ff5 Michael Hanselmann
    ListInstanceFields, [ArgUnknown()],
1410 b82c5ff5 Michael Hanselmann
    [NOHDR_OPT, SEP_OPT],
1411 b82c5ff5 Michael Hanselmann
    "[fields...]",
1412 b82c5ff5 Michael Hanselmann
    "Lists all available fields for instances"),
1413 6ea815cf Iustin Pop
  'reinstall': (
1414 3e54ace7 Iustin Pop
    ReinstallInstance, [ArgInstance()],
1415 06073e85 Guido Trotter
    [FORCE_OPT, OS_OPT, FORCE_VARIANT_OPT, m_force_multi, m_node_opt,
1416 39dfd93e René Nussbaumer
     m_pri_node_opt, m_sec_node_opt, m_clust_opt, m_inst_opt, m_node_tags_opt,
1417 39dfd93e René Nussbaumer
     m_pri_node_tags_opt, m_sec_node_tags_opt, m_inst_tags_opt, SELECT_OS_OPT,
1418 8d8c4eff Michael Hanselmann
     SUBMIT_OPT, DRY_RUN_OPT, PRIORITY_OPT, OSPARAMS_OPT],
1419 6ea815cf Iustin Pop
    "[-f] <instance>", "Reinstall a stopped instance"),
1420 6ea815cf Iustin Pop
  'remove': (
1421 6ea815cf Iustin Pop
    RemoveInstance, ARGS_ONE_INSTANCE,
1422 db5a8a2d Iustin Pop
    [FORCE_OPT, SHUTDOWN_TIMEOUT_OPT, IGNORE_FAILURES_OPT, SUBMIT_OPT,
1423 aa06f8c6 Michael Hanselmann
     DRY_RUN_OPT, PRIORITY_OPT],
1424 6ea815cf Iustin Pop
    "[-f] <instance>", "Shuts down the instance and removes it"),
1425 6ea815cf Iustin Pop
  'rename': (
1426 6ea815cf Iustin Pop
    RenameInstance,
1427 6ea815cf Iustin Pop
    [ArgInstance(min=1, max=1), ArgHost(min=1, max=1)],
1428 aa06f8c6 Michael Hanselmann
    [NOIPCHECK_OPT, NONAMECHECK_OPT, SUBMIT_OPT, DRY_RUN_OPT, PRIORITY_OPT],
1429 6ea815cf Iustin Pop
    "<instance> <new_name>", "Rename the instance"),
1430 6ea815cf Iustin Pop
  'replace-disks': (
1431 6ea815cf Iustin Pop
    ReplaceDisks, ARGS_ONE_INSTANCE,
1432 7ea7bcf6 Iustin Pop
    [AUTO_REPLACE_OPT, DISKIDX_OPT, IALLOCATOR_OPT, EARLY_RELEASE_OPT,
1433 db5a8a2d Iustin Pop
     NEW_SECONDARY_OPT, ON_PRIMARY_OPT, ON_SECONDARY_OPT, SUBMIT_OPT,
1434 aa06f8c6 Michael Hanselmann
     DRY_RUN_OPT, PRIORITY_OPT],
1435 6ea815cf Iustin Pop
    "[-s|-p|-n NODE|-I NAME] <instance>",
1436 6ea815cf Iustin Pop
    "Replaces all disks for the instance"),
1437 6ea815cf Iustin Pop
  'modify': (
1438 6ea815cf Iustin Pop
    SetInstanceParams, ARGS_ONE_INSTANCE,
1439 e29e9550 Iustin Pop
    [BACKEND_OPT, DISK_OPT, FORCE_OPT, HVOPTS_OPT, NET_OPT, SUBMIT_OPT,
1440 1052d622 Iustin Pop
     DISK_TEMPLATE_OPT, SINGLE_NODE_OPT, OS_OPT, FORCE_VARIANT_OPT,
1441 456798ab Iustin Pop
     OSPARAMS_OPT, DRY_RUN_OPT, PRIORITY_OPT, NWSYNC_OPT],
1442 6ea815cf Iustin Pop
    "<instance>", "Alters the parameters of an instance"),
1443 6ea815cf Iustin Pop
  'shutdown': (
1444 1c5945b6 Iustin Pop
    GenericManyOps("shutdown", _ShutdownInstance), [ArgInstance()],
1445 064c21f8 Iustin Pop
    [m_node_opt, m_pri_node_opt, m_sec_node_opt, m_clust_opt,
1446 39dfd93e René Nussbaumer
     m_node_tags_opt, m_pri_node_tags_opt, m_sec_node_tags_opt,
1447 db5a8a2d Iustin Pop
     m_inst_tags_opt, m_inst_opt, m_force_multi, TIMEOUT_OPT, SUBMIT_OPT,
1448 885a0fc4 Iustin Pop
     DRY_RUN_OPT, PRIORITY_OPT, IGNORE_OFFLINE_OPT, NO_REMEMBER_OPT],
1449 6ea815cf Iustin Pop
    "<instance>", "Stops an instance"),
1450 6ea815cf Iustin Pop
  'startup': (
1451 1c5945b6 Iustin Pop
    GenericManyOps("startup", _StartupInstance), [ArgInstance()],
1452 39dfd93e René Nussbaumer
    [FORCE_OPT, m_force_multi, m_node_opt, m_pri_node_opt, m_sec_node_opt,
1453 39dfd93e René Nussbaumer
     m_node_tags_opt, m_pri_node_tags_opt, m_sec_node_tags_opt,
1454 39dfd93e René Nussbaumer
     m_inst_tags_opt, m_clust_opt, m_inst_opt, SUBMIT_OPT, HVOPTS_OPT,
1455 885a0fc4 Iustin Pop
     BACKEND_OPT, DRY_RUN_OPT, PRIORITY_OPT, IGNORE_OFFLINE_OPT,
1456 885a0fc4 Iustin Pop
     NO_REMEMBER_OPT],
1457 6ea815cf Iustin Pop
    "<instance>", "Starts an instance"),
1458 6ea815cf Iustin Pop
  'reboot': (
1459 1c5945b6 Iustin Pop
    GenericManyOps("reboot", _RebootInstance), [ArgInstance()],
1460 064c21f8 Iustin Pop
    [m_force_multi, REBOOT_TYPE_OPT, IGNORE_SECONDARIES_OPT, m_node_opt,
1461 17c3f802 Guido Trotter
     m_pri_node_opt, m_sec_node_opt, m_clust_opt, m_inst_opt, SUBMIT_OPT,
1462 39dfd93e René Nussbaumer
     m_node_tags_opt, m_pri_node_tags_opt, m_sec_node_tags_opt,
1463 aa06f8c6 Michael Hanselmann
     m_inst_tags_opt, SHUTDOWN_TIMEOUT_OPT, DRY_RUN_OPT, PRIORITY_OPT],
1464 6ea815cf Iustin Pop
    "<instance>", "Reboots an instance"),
1465 6ea815cf Iustin Pop
  'activate-disks': (
1466 db5a8a2d Iustin Pop
    ActivateDisks, ARGS_ONE_INSTANCE,
1467 aa06f8c6 Michael Hanselmann
    [SUBMIT_OPT, IGNORE_SIZE_OPT, PRIORITY_OPT],
1468 6ea815cf Iustin Pop
    "<instance>", "Activate an instance's disks"),
1469 6ea815cf Iustin Pop
  'deactivate-disks': (
1470 aa06f8c6 Michael Hanselmann
    DeactivateDisks, ARGS_ONE_INSTANCE,
1471 c9c41373 Iustin Pop
    [FORCE_OPT, SUBMIT_OPT, DRY_RUN_OPT, PRIORITY_OPT],
1472 c9c41373 Iustin Pop
    "[-f] <instance>", "Deactivate an instance's disks"),
1473 6ea815cf Iustin Pop
  'recreate-disks': (
1474 aa06f8c6 Michael Hanselmann
    RecreateDisks, ARGS_ONE_INSTANCE,
1475 c8a96ae7 Iustin Pop
    [SUBMIT_OPT, DISKIDX_OPT, NODE_PLACEMENT_OPT, DRY_RUN_OPT, PRIORITY_OPT],
1476 6ea815cf Iustin Pop
    "<instance>", "Recreate an instance's disks"),
1477 6ea815cf Iustin Pop
  'grow-disk': (
1478 6ea815cf Iustin Pop
    GrowDisk,
1479 6ea815cf Iustin Pop
    [ArgInstance(min=1, max=1), ArgUnknown(min=1, max=1),
1480 6ea815cf Iustin Pop
     ArgUnknown(min=1, max=1)],
1481 aa06f8c6 Michael Hanselmann
    [SUBMIT_OPT, NWSYNC_OPT, DRY_RUN_OPT, PRIORITY_OPT],
1482 6ea815cf Iustin Pop
    "<instance> <disk> <size>", "Grow an instance's disk"),
1483 6ea815cf Iustin Pop
  'list-tags': (
1484 aa06f8c6 Michael Hanselmann
    ListTags, ARGS_ONE_INSTANCE, [PRIORITY_OPT],
1485 6ea815cf Iustin Pop
    "<instance_name>", "List the tags of the given instance"),
1486 6ea815cf Iustin Pop
  'add-tags': (
1487 6ea815cf Iustin Pop
    AddTags, [ArgInstance(min=1, max=1), ArgUnknown()],
1488 aa06f8c6 Michael Hanselmann
    [TAG_SRC_OPT, PRIORITY_OPT],
1489 6ea815cf Iustin Pop
    "<instance_name> tag...", "Add tags to the given instance"),
1490 6ea815cf Iustin Pop
  'remove-tags': (
1491 6ea815cf Iustin Pop
    RemoveTags, [ArgInstance(min=1, max=1), ArgUnknown()],
1492 aa06f8c6 Michael Hanselmann
    [TAG_SRC_OPT, PRIORITY_OPT],
1493 6ea815cf Iustin Pop
    "<instance_name> tag...", "Remove tags from given instance"),
1494 a8083063 Iustin Pop
  }
1495 a8083063 Iustin Pop
1496 7232c04c Iustin Pop
#: dictionary with aliases for commands
1497 dbfd89dd Guido Trotter
aliases = {
1498 536fda25 Guido Trotter
  'start': 'startup',
1499 536fda25 Guido Trotter
  'stop': 'shutdown',
1500 dbfd89dd Guido Trotter
  }
1501 dbfd89dd Guido Trotter
1502 a8005e17 Michael Hanselmann
1503 e792102d Michael Hanselmann
def Main():
1504 e792102d Michael Hanselmann
  return GenericMain(commands, aliases=aliases,
1505 e792102d Michael Hanselmann
                     override={"tag_type": constants.TAG_INSTANCE})