Statistics
| Branch: | Tag: | Revision:

root / lib / client / gnt_instance.py @ a6682fdc

History | View | Annotate | Download (52.2 kB)

1 e792102d Michael Hanselmann
#
2 a8083063 Iustin Pop
#
3 a8083063 Iustin Pop
4 783a6c0b Iustin Pop
# Copyright (C) 2006, 2007, 2008, 2009, 2010 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 31a853d2 Iustin Pop
_VALUE_TRUE = "true"
61 31a853d2 Iustin Pop
62 7232c04c Iustin Pop
#: default list of options for L{ListInstances}
63 48c4dfa8 Iustin Pop
_LIST_DEF_FIELDS = [
64 e69d05fd Iustin Pop
  "name", "hypervisor", "os", "pnode", "status", "oper_ram",
65 48c4dfa8 Iustin Pop
  ]
66 48c4dfa8 Iustin Pop
67 bdb7d4e8 Michael Hanselmann
68 479636a3 Iustin Pop
def _ExpandMultiNames(mode, names, client=None):
69 312ac745 Iustin Pop
  """Expand the given names using the passed mode.
70 312ac745 Iustin Pop

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

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

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

150 804a1e8e Iustin Pop
  This function is used to request confirmation for doing an operation
151 804a1e8e Iustin Pop
  on a given list of instances.
152 804a1e8e Iustin Pop

153 7232c04c Iustin Pop
  @type inames: list
154 7232c04c Iustin Pop
  @param inames: the list of names that we display when
155 7232c04c Iustin Pop
      we ask for confirmation
156 7232c04c Iustin Pop
  @type text: str
157 7232c04c Iustin Pop
  @param text: the operation that the user should confirm
158 7232c04c Iustin Pop
      (e.g. I{shutdown} or I{startup})
159 7232c04c Iustin Pop
  @rtype: boolean
160 7232c04c Iustin Pop
  @return: True or False depending on user's confirmation.
161 804a1e8e Iustin Pop

162 804a1e8e Iustin Pop
  """
163 804a1e8e Iustin Pop
  count = len(inames)
164 55efe6da Iustin Pop
  msg = ("The %s will operate on %d instances.\n%s"
165 55efe6da Iustin Pop
         "Do you want to continue?" % (text, count, extra))
166 804a1e8e Iustin Pop
  affected = ("\nAffected instances:\n" +
167 804a1e8e Iustin Pop
              "\n".join(["  %s" % name for name in inames]))
168 804a1e8e Iustin Pop
169 804a1e8e Iustin Pop
  choices = [('y', True, 'Yes, execute the %s' % text),
170 804a1e8e Iustin Pop
             ('n', False, 'No, abort the %s' % text)]
171 804a1e8e Iustin Pop
172 804a1e8e Iustin Pop
  if count > 20:
173 804a1e8e Iustin Pop
    choices.insert(1, ('v', 'v', 'View the list of affected instances'))
174 804a1e8e Iustin Pop
    ask = msg
175 804a1e8e Iustin Pop
  else:
176 804a1e8e Iustin Pop
    ask = msg + affected
177 804a1e8e Iustin Pop
178 804a1e8e Iustin Pop
  choice = AskUser(ask, choices)
179 804a1e8e Iustin Pop
  if choice == 'v':
180 804a1e8e Iustin Pop
    choices.pop(1)
181 5e66b7e6 Iustin Pop
    choice = AskUser(msg + affected, choices)
182 804a1e8e Iustin Pop
  return choice
183 804a1e8e Iustin Pop
184 804a1e8e Iustin Pop
185 a76f0c4a Iustin Pop
def _EnsureInstancesExist(client, names):
186 a76f0c4a Iustin Pop
  """Check for and ensure the given instance names exist.
187 a76f0c4a Iustin Pop

188 a76f0c4a Iustin Pop
  This function will raise an OpPrereqError in case they don't
189 a76f0c4a Iustin Pop
  exist. Otherwise it will exit cleanly.
190 a76f0c4a Iustin Pop

191 f2fd87d7 Iustin Pop
  @type client: L{ganeti.luxi.Client}
192 a76f0c4a Iustin Pop
  @param client: the client to use for the query
193 a76f0c4a Iustin Pop
  @type names: list
194 a76f0c4a Iustin Pop
  @param names: the list of instance names to query
195 a76f0c4a Iustin Pop
  @raise errors.OpPrereqError: in case any instance is missing
196 a76f0c4a Iustin Pop

197 a76f0c4a Iustin Pop
  """
198 a76f0c4a Iustin Pop
  # TODO: change LUQueryInstances to that it actually returns None
199 a76f0c4a Iustin Pop
  # instead of raising an exception, or devise a better mechanism
200 ec79568d Iustin Pop
  result = client.QueryInstances(names, ["name"], False)
201 a76f0c4a Iustin Pop
  for orig_name, row in zip(names, result):
202 a76f0c4a Iustin Pop
    if row[0] is None:
203 debac808 Iustin Pop
      raise errors.OpPrereqError("Instance '%s' does not exist" % orig_name,
204 debac808 Iustin Pop
                                 errors.ECODE_NOENT)
205 a76f0c4a Iustin Pop
206 a76f0c4a Iustin Pop
207 1c5945b6 Iustin Pop
def GenericManyOps(operation, fn):
208 1c5945b6 Iustin Pop
  """Generic multi-instance operations.
209 1c5945b6 Iustin Pop

210 1c5945b6 Iustin Pop
  The will return a wrapper that processes the options and arguments
211 1c5945b6 Iustin Pop
  given, and uses the passed function to build the opcode needed for
212 1c5945b6 Iustin Pop
  the specific operation. Thus all the generic loop/confirmation code
213 1c5945b6 Iustin Pop
  is abstracted into this function.
214 1c5945b6 Iustin Pop

215 1c5945b6 Iustin Pop
  """
216 1c5945b6 Iustin Pop
  def realfn(opts, args):
217 1c5945b6 Iustin Pop
    if opts.multi_mode is None:
218 1c5945b6 Iustin Pop
      opts.multi_mode = _SHUTDOWN_INSTANCES
219 1c5945b6 Iustin Pop
    cl = GetClient()
220 1c5945b6 Iustin Pop
    inames = _ExpandMultiNames(opts.multi_mode, args, client=cl)
221 1c5945b6 Iustin Pop
    if not inames:
222 1c5945b6 Iustin Pop
      raise errors.OpPrereqError("Selection filter does not match"
223 debac808 Iustin Pop
                                 " any instances", errors.ECODE_INVAL)
224 1c5945b6 Iustin Pop
    multi_on = opts.multi_mode != _SHUTDOWN_INSTANCES or len(inames) > 1
225 1c5945b6 Iustin Pop
    if not (opts.force_multi or not multi_on
226 1c5945b6 Iustin Pop
            or _ConfirmOperation(inames, operation)):
227 1c5945b6 Iustin Pop
      return 1
228 cb573a31 Iustin Pop
    jex = JobExecutor(verbose=multi_on, cl=cl, opts=opts)
229 1c5945b6 Iustin Pop
    for name in inames:
230 1c5945b6 Iustin Pop
      op = fn(name, opts)
231 1c5945b6 Iustin Pop
      jex.QueueJob(name, op)
232 b4e68848 Iustin Pop
    results = jex.WaitOrShow(not opts.submit_only)
233 b4e68848 Iustin Pop
    rcode = compat.all(row[0] for row in results)
234 b4e68848 Iustin Pop
    return int(not rcode)
235 1c5945b6 Iustin Pop
  return realfn
236 1c5945b6 Iustin Pop
237 1c5945b6 Iustin Pop
238 a8083063 Iustin Pop
def ListInstances(opts, args):
239 f5abe9bd Oleksiy Mishchenko
  """List instances and their properties.
240 a8083063 Iustin Pop

241 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
242 7232c04c Iustin Pop
  @type args: list
243 7232c04c Iustin Pop
  @param args: should be an empty list
244 7232c04c Iustin Pop
  @rtype: int
245 7232c04c Iustin Pop
  @return: the desired exit code
246 7232c04c Iustin Pop

247 a8083063 Iustin Pop
  """
248 a4ebd726 Michael Hanselmann
  selected_fields = ParseFields(opts.output, _LIST_DEF_FIELDS)
249 a8083063 Iustin Pop
250 b82c5ff5 Michael Hanselmann
  fmtoverride = dict.fromkeys(["tags", "disk.sizes", "nic.macs", "nic.ips",
251 b82c5ff5 Michael Hanselmann
                               "nic.modes", "nic.links", "nic.bridges",
252 b82c5ff5 Michael Hanselmann
                               "snodes"],
253 b82c5ff5 Michael Hanselmann
                              (lambda value: ",".join(str(item)
254 b82c5ff5 Michael Hanselmann
                                                      for item in value),
255 b82c5ff5 Michael Hanselmann
                               False))
256 a8083063 Iustin Pop
257 b82c5ff5 Michael Hanselmann
  return GenericList(constants.QR_INSTANCE, selected_fields, args, opts.units,
258 b82c5ff5 Michael Hanselmann
                     opts.separator, not opts.no_headers,
259 b82c5ff5 Michael Hanselmann
                     format_override=fmtoverride)
260 b82c5ff5 Michael Hanselmann
261 b82c5ff5 Michael Hanselmann
262 b82c5ff5 Michael Hanselmann
def ListInstanceFields(opts, args):
263 b82c5ff5 Michael Hanselmann
  """List instance fields.
264 b82c5ff5 Michael Hanselmann

265 b82c5ff5 Michael Hanselmann
  @param opts: the command line options selected by the user
266 b82c5ff5 Michael Hanselmann
  @type args: list
267 b82c5ff5 Michael Hanselmann
  @param args: fields to list, or empty for all
268 b82c5ff5 Michael Hanselmann
  @rtype: int
269 b82c5ff5 Michael Hanselmann
  @return: the desired exit code
270 b82c5ff5 Michael Hanselmann

271 b82c5ff5 Michael Hanselmann
  """
272 b82c5ff5 Michael Hanselmann
  return GenericListFields(constants.QR_INSTANCE, args, opts.separator,
273 b82c5ff5 Michael Hanselmann
                           not opts.no_headers)
274 a8083063 Iustin Pop
275 a8083063 Iustin Pop
276 a8083063 Iustin Pop
def AddInstance(opts, args):
277 a8083063 Iustin Pop
  """Add an instance to the cluster.
278 a8083063 Iustin Pop

279 d77490c5 Iustin Pop
  This is just a wrapper over GenericInstanceCreate.
280 a8083063 Iustin Pop

281 a8083063 Iustin Pop
  """
282 d77490c5 Iustin Pop
  return GenericInstanceCreate(constants.INSTANCE_CREATE, opts, args)
283 a8083063 Iustin Pop
284 a8083063 Iustin Pop
285 0d0e9090 René Nussbaumer
def BatchCreate(opts, args):
286 7232c04c Iustin Pop
  """Create instances using a definition file.
287 7232c04c Iustin Pop

288 7232c04c Iustin Pop
  This function reads a json file with instances defined
289 7232c04c Iustin Pop
  in the form::
290 7232c04c Iustin Pop

291 7232c04c Iustin Pop
    {"instance-name":{
292 9939547b Iustin Pop
      "disk_size": [20480],
293 7232c04c Iustin Pop
      "template": "drbd",
294 7232c04c Iustin Pop
      "backend": {
295 7232c04c Iustin Pop
        "memory": 512,
296 7232c04c Iustin Pop
        "vcpus": 1 },
297 9939547b Iustin Pop
      "os": "debootstrap",
298 7232c04c Iustin Pop
      "primary_node": "firstnode",
299 7232c04c Iustin Pop
      "secondary_node": "secondnode",
300 7232c04c Iustin Pop
      "iallocator": "dumb"}
301 7232c04c Iustin Pop
    }
302 7232c04c Iustin Pop

303 7232c04c Iustin Pop
  Note that I{primary_node} and I{secondary_node} have precedence over
304 7232c04c Iustin Pop
  I{iallocator}.
305 7232c04c Iustin Pop

306 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
307 7232c04c Iustin Pop
  @type args: list
308 7232c04c Iustin Pop
  @param args: should contain one element, the json filename
309 7232c04c Iustin Pop
  @rtype: int
310 7232c04c Iustin Pop
  @return: the desired exit code
311 0d0e9090 René Nussbaumer

312 0d0e9090 René Nussbaumer
  """
313 9939547b Iustin Pop
  _DEFAULT_SPECS = {"disk_size": [20 * 1024],
314 0d0e9090 René Nussbaumer
                    "backend": {},
315 0d0e9090 René Nussbaumer
                    "iallocator": None,
316 0d0e9090 René Nussbaumer
                    "primary_node": None,
317 0d0e9090 René Nussbaumer
                    "secondary_node": None,
318 a379d9bd Guido Trotter
                    "nics": None,
319 0d0e9090 René Nussbaumer
                    "start": True,
320 0d0e9090 René Nussbaumer
                    "ip_check": True,
321 460d22be Iustin Pop
                    "name_check": True,
322 0d0e9090 René Nussbaumer
                    "hypervisor": None,
323 4082e6f9 Iustin Pop
                    "hvparams": {},
324 0d0e9090 René Nussbaumer
                    "file_storage_dir": None,
325 27ba7eab Guido Trotter
                    "force_variant": False,
326 0d0e9090 René Nussbaumer
                    "file_driver": 'loop'}
327 0d0e9090 René Nussbaumer
328 0d0e9090 René Nussbaumer
  def _PopulateWithDefaults(spec):
329 0d0e9090 René Nussbaumer
    """Returns a new hash combined with default values."""
330 2f79bd34 Iustin Pop
    mydict = _DEFAULT_SPECS.copy()
331 2f79bd34 Iustin Pop
    mydict.update(spec)
332 2f79bd34 Iustin Pop
    return mydict
333 0d0e9090 René Nussbaumer
334 0d0e9090 René Nussbaumer
  def _Validate(spec):
335 0d0e9090 René Nussbaumer
    """Validate the instance specs."""
336 0d0e9090 René Nussbaumer
    # Validate fields required under any circumstances
337 0d0e9090 René Nussbaumer
    for required_field in ('os', 'template'):
338 0d0e9090 René Nussbaumer
      if required_field not in spec:
339 0d0e9090 René Nussbaumer
        raise errors.OpPrereqError('Required field "%s" is missing.' %
340 debac808 Iustin Pop
                                   required_field, errors.ECODE_INVAL)
341 0d0e9090 René Nussbaumer
    # Validate special fields
342 0d0e9090 René Nussbaumer
    if spec['primary_node'] is not None:
343 0d0e9090 René Nussbaumer
      if (spec['template'] in constants.DTS_NET_MIRROR and
344 0d0e9090 René Nussbaumer
          spec['secondary_node'] is None):
345 0d0e9090 René Nussbaumer
        raise errors.OpPrereqError('Template requires secondary node, but'
346 debac808 Iustin Pop
                                   ' there was no secondary provided.',
347 debac808 Iustin Pop
                                   errors.ECODE_INVAL)
348 0d0e9090 René Nussbaumer
    elif spec['iallocator'] is None:
349 0d0e9090 René Nussbaumer
      raise errors.OpPrereqError('You have to provide at least a primary_node'
350 debac808 Iustin Pop
                                 ' or an iallocator.',
351 debac808 Iustin Pop
                                 errors.ECODE_INVAL)
352 0d0e9090 René Nussbaumer
353 4082e6f9 Iustin Pop
    if (spec['hvparams'] and
354 4082e6f9 Iustin Pop
        not isinstance(spec['hvparams'], dict)):
355 debac808 Iustin Pop
      raise errors.OpPrereqError('Hypervisor parameters must be a dict.',
356 debac808 Iustin Pop
                                 errors.ECODE_INVAL)
357 0d0e9090 René Nussbaumer
358 0d0e9090 René Nussbaumer
  json_filename = args[0]
359 0d0e9090 René Nussbaumer
  try:
360 13998ef2 Michael Hanselmann
    instance_data = simplejson.loads(utils.ReadFile(json_filename))
361 7260cfbe Iustin Pop
  except Exception, err: # pylint: disable-msg=W0703
362 4082e6f9 Iustin Pop
    ToStderr("Can't parse the instance definition file: %s" % str(err))
363 4082e6f9 Iustin Pop
    return 1
364 0d0e9090 René Nussbaumer
365 fe7c59d5 Guido Trotter
  if not isinstance(instance_data, dict):
366 fe7c59d5 Guido Trotter
    ToStderr("The instance definition file is not in dict format.")
367 fe7c59d5 Guido Trotter
    return 1
368 fe7c59d5 Guido Trotter
369 cb573a31 Iustin Pop
  jex = JobExecutor(opts=opts)
370 d4dd4b74 Iustin Pop
371 0d0e9090 René Nussbaumer
  # Iterate over the instances and do:
372 0d0e9090 René Nussbaumer
  #  * Populate the specs with default value
373 0d0e9090 René Nussbaumer
  #  * Validate the instance specs
374 fe7c59d5 Guido Trotter
  i_names = utils.NiceSort(instance_data.keys()) # pylint: disable-msg=E1103
375 7312b33d Iustin Pop
  for name in i_names:
376 7312b33d Iustin Pop
    specs = instance_data[name]
377 0d0e9090 René Nussbaumer
    specs = _PopulateWithDefaults(specs)
378 0d0e9090 René Nussbaumer
    _Validate(specs)
379 0d0e9090 René Nussbaumer
380 4082e6f9 Iustin Pop
    hypervisor = specs['hypervisor']
381 4082e6f9 Iustin Pop
    hvparams = specs['hvparams']
382 0d0e9090 René Nussbaumer
383 9939547b Iustin Pop
    disks = []
384 9939547b Iustin Pop
    for elem in specs['disk_size']:
385 9939547b Iustin Pop
      try:
386 9939547b Iustin Pop
        size = utils.ParseUnit(elem)
387 691744c4 Iustin Pop
      except (TypeError, ValueError), err:
388 9939547b Iustin Pop
        raise errors.OpPrereqError("Invalid disk size '%s' for"
389 9939547b Iustin Pop
                                   " instance %s: %s" %
390 debac808 Iustin Pop
                                   (elem, name, err), errors.ECODE_INVAL)
391 9939547b Iustin Pop
      disks.append({"size": size})
392 9939547b Iustin Pop
393 a5728081 Guido Trotter
    utils.ForceDictType(specs['backend'], constants.BES_PARAMETER_TYPES)
394 a5728081 Guido Trotter
    utils.ForceDictType(hvparams, constants.HVS_PARAMETER_TYPES)
395 a5728081 Guido Trotter
396 a379d9bd Guido Trotter
    tmp_nics = []
397 a379d9bd Guido Trotter
    for field in ('ip', 'mac', 'mode', 'link', 'bridge'):
398 a379d9bd Guido Trotter
      if field in specs:
399 a379d9bd Guido Trotter
        if not tmp_nics:
400 a379d9bd Guido Trotter
          tmp_nics.append({})
401 a379d9bd Guido Trotter
        tmp_nics[0][field] = specs[field]
402 a379d9bd Guido Trotter
403 a379d9bd Guido Trotter
    if specs['nics'] is not None and tmp_nics:
404 a379d9bd Guido Trotter
      raise errors.OpPrereqError("'nics' list incompatible with using"
405 debac808 Iustin Pop
                                 " individual nic fields as well",
406 debac808 Iustin Pop
                                 errors.ECODE_INVAL)
407 a379d9bd Guido Trotter
    elif specs['nics'] is not None:
408 a379d9bd Guido Trotter
      tmp_nics = specs['nics']
409 a379d9bd Guido Trotter
    elif not tmp_nics:
410 a379d9bd Guido Trotter
      tmp_nics = [{}]
411 a379d9bd Guido Trotter
412 0d0e9090 René Nussbaumer
    op = opcodes.OpCreateInstance(instance_name=name,
413 9939547b Iustin Pop
                                  disks=disks,
414 0d0e9090 René Nussbaumer
                                  disk_template=specs['template'],
415 0d0e9090 René Nussbaumer
                                  mode=constants.INSTANCE_CREATE,
416 0d0e9090 René Nussbaumer
                                  os_type=specs['os'],
417 de372295 Guido Trotter
                                  force_variant=specs["force_variant"],
418 0d0e9090 René Nussbaumer
                                  pnode=specs['primary_node'],
419 0d0e9090 René Nussbaumer
                                  snode=specs['secondary_node'],
420 a379d9bd Guido Trotter
                                  nics=tmp_nics,
421 0d0e9090 René Nussbaumer
                                  start=specs['start'],
422 0d0e9090 René Nussbaumer
                                  ip_check=specs['ip_check'],
423 460d22be Iustin Pop
                                  name_check=specs['name_check'],
424 0d0e9090 René Nussbaumer
                                  wait_for_sync=True,
425 0d0e9090 René Nussbaumer
                                  iallocator=specs['iallocator'],
426 0d0e9090 René Nussbaumer
                                  hypervisor=hypervisor,
427 0d0e9090 René Nussbaumer
                                  hvparams=hvparams,
428 0d0e9090 René Nussbaumer
                                  beparams=specs['backend'],
429 0d0e9090 René Nussbaumer
                                  file_storage_dir=specs['file_storage_dir'],
430 0d0e9090 René Nussbaumer
                                  file_driver=specs['file_driver'])
431 0d0e9090 René Nussbaumer
432 d4dd4b74 Iustin Pop
    jex.QueueJob(name, op)
433 d4dd4b74 Iustin Pop
  # we never want to wait, just show the submitted job IDs
434 d4dd4b74 Iustin Pop
  jex.WaitOrShow(False)
435 0d0e9090 René Nussbaumer
436 0d0e9090 René Nussbaumer
  return 0
437 0d0e9090 René Nussbaumer
438 0d0e9090 René Nussbaumer
439 fe7b0351 Michael Hanselmann
def ReinstallInstance(opts, args):
440 fe7b0351 Michael Hanselmann
  """Reinstall an instance.
441 fe7b0351 Michael Hanselmann

442 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
443 7232c04c Iustin Pop
  @type args: list
444 7232c04c Iustin Pop
  @param args: should contain only one element, the name of the
445 7232c04c Iustin Pop
      instance to be reinstalled
446 7232c04c Iustin Pop
  @rtype: int
447 7232c04c Iustin Pop
  @return: the desired exit code
448 fe7b0351 Michael Hanselmann

449 fe7b0351 Michael Hanselmann
  """
450 55efe6da Iustin Pop
  # first, compute the desired name list
451 55efe6da Iustin Pop
  if opts.multi_mode is None:
452 55efe6da Iustin Pop
    opts.multi_mode = _SHUTDOWN_INSTANCES
453 55efe6da Iustin Pop
454 55efe6da Iustin Pop
  inames = _ExpandMultiNames(opts.multi_mode, args)
455 55efe6da Iustin Pop
  if not inames:
456 debac808 Iustin Pop
    raise errors.OpPrereqError("Selection filter does not match any instances",
457 debac808 Iustin Pop
                               errors.ECODE_INVAL)
458 fe7b0351 Michael Hanselmann
459 55efe6da Iustin Pop
  # second, if requested, ask for an OS
460 20e23543 Alexander Schreiber
  if opts.select_os is True:
461 d22dfef7 Iustin Pop
    op = opcodes.OpDiagnoseOS(output_fields=["name", "variants"], names=[])
462 400ca2f7 Iustin Pop
    result = SubmitOpCode(op, opts=opts)
463 20e23543 Alexander Schreiber
464 20e23543 Alexander Schreiber
    if not result:
465 3a24c527 Iustin Pop
      ToStdout("Can't get the OS list")
466 20e23543 Alexander Schreiber
      return 1
467 20e23543 Alexander Schreiber
468 3a24c527 Iustin Pop
    ToStdout("Available OS templates:")
469 20e23543 Alexander Schreiber
    number = 0
470 20e23543 Alexander Schreiber
    choices = []
471 d22dfef7 Iustin Pop
    for (name, variants) in result:
472 d22dfef7 Iustin Pop
      for entry in CalculateOSNames(name, variants):
473 d22dfef7 Iustin Pop
        ToStdout("%3s: %s", number, entry)
474 d22dfef7 Iustin Pop
        choices.append(("%s" % number, entry, entry))
475 d22dfef7 Iustin Pop
        number += 1
476 20e23543 Alexander Schreiber
477 20e23543 Alexander Schreiber
    choices.append(('x', 'exit', 'Exit gnt-instance reinstall'))
478 949bdabe Iustin Pop
    selected = AskUser("Enter OS template number (or x to abort):",
479 20e23543 Alexander Schreiber
                       choices)
480 20e23543 Alexander Schreiber
481 20e23543 Alexander Schreiber
    if selected == 'exit':
482 55efe6da Iustin Pop
      ToStderr("User aborted reinstall, exiting")
483 20e23543 Alexander Schreiber
      return 1
484 20e23543 Alexander Schreiber
485 2f79bd34 Iustin Pop
    os_name = selected
486 20e23543 Alexander Schreiber
  else:
487 2f79bd34 Iustin Pop
    os_name = opts.os
488 20e23543 Alexander Schreiber
489 297ddce9 Iustin Pop
  # third, get confirmation: multi-reinstall requires --force-multi,
490 297ddce9 Iustin Pop
  # single-reinstall either --force or --force-multi (--force-multi is
491 297ddce9 Iustin Pop
  # a stronger --force)
492 55efe6da Iustin Pop
  multi_on = opts.multi_mode != _SHUTDOWN_INSTANCES or len(inames) > 1
493 55efe6da Iustin Pop
  if multi_on:
494 55efe6da Iustin Pop
    warn_msg = "Note: this will remove *all* data for the below instances!\n"
495 297ddce9 Iustin Pop
    if not (opts.force_multi or
496 55efe6da Iustin Pop
            _ConfirmOperation(inames, "reinstall", extra=warn_msg)):
497 fe7b0351 Michael Hanselmann
      return 1
498 55efe6da Iustin Pop
  else:
499 297ddce9 Iustin Pop
    if not (opts.force or opts.force_multi):
500 55efe6da Iustin Pop
      usertext = ("This will reinstall the instance %s and remove"
501 b6e243ab Iustin Pop
                  " all data. Continue?") % inames[0]
502 55efe6da Iustin Pop
      if not AskUser(usertext):
503 55efe6da Iustin Pop
        return 1
504 55efe6da Iustin Pop
505 cb573a31 Iustin Pop
  jex = JobExecutor(verbose=multi_on, opts=opts)
506 55efe6da Iustin Pop
  for instance_name in inames:
507 55efe6da Iustin Pop
    op = opcodes.OpReinstallInstance(instance_name=instance_name,
508 06073e85 Guido Trotter
                                     os_type=os_name,
509 8d8c4eff Michael Hanselmann
                                     force_variant=opts.force_variant,
510 8d8c4eff Michael Hanselmann
                                     osparams=opts.osparams)
511 55efe6da Iustin Pop
    jex.QueueJob(instance_name, op)
512 fe7b0351 Michael Hanselmann
513 55efe6da Iustin Pop
  jex.WaitOrShow(not opts.submit_only)
514 fe7b0351 Michael Hanselmann
  return 0
515 fe7b0351 Michael Hanselmann
516 fe7b0351 Michael Hanselmann
517 a8083063 Iustin Pop
def RemoveInstance(opts, args):
518 a8083063 Iustin Pop
  """Remove an instance.
519 a8083063 Iustin Pop

520 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
521 7232c04c Iustin Pop
  @type args: list
522 7232c04c Iustin Pop
  @param args: should contain only one element, the name of
523 7232c04c Iustin Pop
      the instance to be removed
524 7232c04c Iustin Pop
  @rtype: int
525 7232c04c Iustin Pop
  @return: the desired exit code
526 a8083063 Iustin Pop

527 a8083063 Iustin Pop
  """
528 a8083063 Iustin Pop
  instance_name = args[0]
529 a8083063 Iustin Pop
  force = opts.force
530 a76f0c4a Iustin Pop
  cl = GetClient()
531 a8083063 Iustin Pop
532 a8083063 Iustin Pop
  if not force:
533 a76f0c4a Iustin Pop
    _EnsureInstancesExist(cl, [instance_name])
534 a76f0c4a Iustin Pop
535 a8083063 Iustin Pop
    usertext = ("This will remove the volumes of the instance %s"
536 a8083063 Iustin Pop
                " (including mirrors), thus removing all the data"
537 a8083063 Iustin Pop
                " of the instance. Continue?") % instance_name
538 47988778 Iustin Pop
    if not AskUser(usertext):
539 a8083063 Iustin Pop
      return 1
540 a8083063 Iustin Pop
541 1d67656e Iustin Pop
  op = opcodes.OpRemoveInstance(instance_name=instance_name,
542 17c3f802 Guido Trotter
                                ignore_failures=opts.ignore_failures,
543 4d98c565 Guido Trotter
                                shutdown_timeout=opts.shutdown_timeout)
544 a76f0c4a Iustin Pop
  SubmitOrSend(op, opts, cl=cl)
545 a8083063 Iustin Pop
  return 0
546 a8083063 Iustin Pop
547 a8083063 Iustin Pop
548 decd5f45 Iustin Pop
def RenameInstance(opts, args):
549 4ab0b9e3 Guido Trotter
  """Rename an instance.
550 decd5f45 Iustin Pop

551 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
552 7232c04c Iustin Pop
  @type args: list
553 7232c04c Iustin Pop
  @param args: should contain two elements, the old and the
554 7232c04c Iustin Pop
      new instance names
555 7232c04c Iustin Pop
  @rtype: int
556 7232c04c Iustin Pop
  @return: the desired exit code
557 decd5f45 Iustin Pop

558 decd5f45 Iustin Pop
  """
559 90ed09b0 René Nussbaumer
  if not opts.name_check:
560 1b6dddc8 René Nussbaumer
    if not AskUser("As you disabled the check of the DNS entry, please verify"
561 1b6dddc8 René Nussbaumer
                   " that '%s' is a FQDN. Continue?" % args[1]):
562 1b6dddc8 René Nussbaumer
      return 1
563 1b6dddc8 René Nussbaumer
564 decd5f45 Iustin Pop
  op = opcodes.OpRenameInstance(instance_name=args[0],
565 decd5f45 Iustin Pop
                                new_name=args[1],
566 3fe11ba3 Manuel Franceschini
                                ip_check=opts.ip_check,
567 3fe11ba3 Manuel Franceschini
                                name_check=opts.name_check)
568 6a016df9 Michael Hanselmann
  result = SubmitOrSend(op, opts)
569 6a016df9 Michael Hanselmann
570 48418fea Iustin Pop
  if result:
571 48418fea Iustin Pop
    ToStdout("Instance '%s' renamed to '%s'", args[0], result)
572 6a016df9 Michael Hanselmann
573 decd5f45 Iustin Pop
  return 0
574 decd5f45 Iustin Pop
575 decd5f45 Iustin Pop
576 a8083063 Iustin Pop
def ActivateDisks(opts, args):
577 a8083063 Iustin Pop
  """Activate an instance's disks.
578 a8083063 Iustin Pop

579 a8083063 Iustin Pop
  This serves two purposes:
580 7232c04c Iustin Pop
    - it allows (as long as the instance is not running)
581 7232c04c Iustin Pop
      mounting the disks and modifying them from the node
582 a8083063 Iustin Pop
    - it repairs inactive secondary drbds
583 a8083063 Iustin Pop

584 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
585 7232c04c Iustin Pop
  @type args: list
586 7232c04c Iustin Pop
  @param args: should contain only one element, the instance name
587 7232c04c Iustin Pop
  @rtype: int
588 7232c04c Iustin Pop
  @return: the desired exit code
589 7232c04c Iustin Pop

590 a8083063 Iustin Pop
  """
591 a8083063 Iustin Pop
  instance_name = args[0]
592 b4ec07f8 Iustin Pop
  op = opcodes.OpActivateInstanceDisks(instance_name=instance_name,
593 b4ec07f8 Iustin Pop
                                       ignore_size=opts.ignore_size)
594 6340bb0a Iustin Pop
  disks_info = SubmitOrSend(op, opts)
595 a8083063 Iustin Pop
  for host, iname, nname in disks_info:
596 3a24c527 Iustin Pop
    ToStdout("%s:%s:%s", host, iname, nname)
597 a8083063 Iustin Pop
  return 0
598 a8083063 Iustin Pop
599 a8083063 Iustin Pop
600 a8083063 Iustin Pop
def DeactivateDisks(opts, args):
601 bd315bfa Iustin Pop
  """Deactivate an instance's disks.
602 a8083063 Iustin Pop

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

606 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
607 7232c04c Iustin Pop
  @type args: list
608 7232c04c Iustin Pop
  @param args: should contain only one element, the instance name
609 7232c04c Iustin Pop
  @rtype: int
610 7232c04c Iustin Pop
  @return: the desired exit code
611 7232c04c Iustin Pop

612 a8083063 Iustin Pop
  """
613 a8083063 Iustin Pop
  instance_name = args[0]
614 a8083063 Iustin Pop
  op = opcodes.OpDeactivateInstanceDisks(instance_name=instance_name)
615 6340bb0a Iustin Pop
  SubmitOrSend(op, opts)
616 a8083063 Iustin Pop
  return 0
617 a8083063 Iustin Pop
618 a8083063 Iustin Pop
619 bd315bfa Iustin Pop
def RecreateDisks(opts, args):
620 bd315bfa Iustin Pop
  """Recreate an instance's disks.
621 bd315bfa Iustin Pop

622 bd315bfa Iustin Pop
  @param opts: the command line options selected by the user
623 bd315bfa Iustin Pop
  @type args: list
624 bd315bfa Iustin Pop
  @param args: should contain only one element, the instance name
625 bd315bfa Iustin Pop
  @rtype: int
626 bd315bfa Iustin Pop
  @return: the desired exit code
627 bd315bfa Iustin Pop

628 bd315bfa Iustin Pop
  """
629 bd315bfa Iustin Pop
  instance_name = args[0]
630 bd315bfa Iustin Pop
  if opts.disks:
631 bd315bfa Iustin Pop
    try:
632 bd315bfa Iustin Pop
      opts.disks = [int(v) for v in opts.disks.split(",")]
633 bd315bfa Iustin Pop
    except (ValueError, TypeError), err:
634 bd315bfa Iustin Pop
      ToStderr("Invalid disks value: %s" % str(err))
635 bd315bfa Iustin Pop
      return 1
636 bd315bfa Iustin Pop
  else:
637 bd315bfa Iustin Pop
    opts.disks = []
638 bd315bfa Iustin Pop
639 bd315bfa Iustin Pop
  op = opcodes.OpRecreateInstanceDisks(instance_name=instance_name,
640 bd315bfa Iustin Pop
                                       disks=opts.disks)
641 bd315bfa Iustin Pop
  SubmitOrSend(op, opts)
642 bd315bfa Iustin Pop
  return 0
643 bd315bfa Iustin Pop
644 bd315bfa Iustin Pop
645 c6e911bc Iustin Pop
def GrowDisk(opts, args):
646 7232c04c Iustin Pop
  """Grow an instance's disks.
647 c6e911bc Iustin Pop

648 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
649 7232c04c Iustin Pop
  @type args: list
650 7232c04c Iustin Pop
  @param args: should contain two elements, the instance name
651 7232c04c Iustin Pop
      whose disks we grow and the disk name, e.g. I{sda}
652 7232c04c Iustin Pop
  @rtype: int
653 7232c04c Iustin Pop
  @return: the desired exit code
654 c6e911bc Iustin Pop

655 c6e911bc Iustin Pop
  """
656 c6e911bc Iustin Pop
  instance = args[0]
657 c6e911bc Iustin Pop
  disk = args[1]
658 ad24e046 Iustin Pop
  try:
659 ad24e046 Iustin Pop
    disk = int(disk)
660 691744c4 Iustin Pop
  except (TypeError, ValueError), err:
661 debac808 Iustin Pop
    raise errors.OpPrereqError("Invalid disk index: %s" % str(err),
662 debac808 Iustin Pop
                               errors.ECODE_INVAL)
663 c6e911bc Iustin Pop
  amount = utils.ParseUnit(args[2])
664 6605411d Iustin Pop
  op = opcodes.OpGrowDisk(instance_name=instance, disk=disk, amount=amount,
665 6605411d Iustin Pop
                          wait_for_sync=opts.wait_for_sync)
666 6340bb0a Iustin Pop
  SubmitOrSend(op, opts)
667 c6e911bc Iustin Pop
  return 0
668 c6e911bc Iustin Pop
669 c6e911bc Iustin Pop
670 1c5945b6 Iustin Pop
def _StartupInstance(name, opts):
671 7232c04c Iustin Pop
  """Startup instances.
672 a8083063 Iustin Pop

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

676 1c5945b6 Iustin Pop
  @param name: the name of the instance to act on
677 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
678 1c5945b6 Iustin Pop
  @return: the opcode needed for the operation
679 a8083063 Iustin Pop

680 a8083063 Iustin Pop
  """
681 1c5945b6 Iustin Pop
  op = opcodes.OpStartupInstance(instance_name=name,
682 b44bd844 Michael Hanselmann
                                 force=opts.force,
683 b44bd844 Michael Hanselmann
                                 ignore_offline_nodes=opts.ignore_offline)
684 1c5945b6 Iustin Pop
  # do not add these parameters to the opcode unless they're defined
685 1c5945b6 Iustin Pop
  if opts.hvparams:
686 1c5945b6 Iustin Pop
    op.hvparams = opts.hvparams
687 1c5945b6 Iustin Pop
  if opts.beparams:
688 1c5945b6 Iustin Pop
    op.beparams = opts.beparams
689 1c5945b6 Iustin Pop
  return op
690 a8083063 Iustin Pop
691 7c0d6283 Michael Hanselmann
692 1c5945b6 Iustin Pop
def _RebootInstance(name, opts):
693 7232c04c Iustin Pop
  """Reboot instance(s).
694 7232c04c Iustin Pop

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

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

702 579d4337 Alexander Schreiber
  """
703 1c5945b6 Iustin Pop
  return opcodes.OpRebootInstance(instance_name=name,
704 579d4337 Alexander Schreiber
                                  reboot_type=opts.reboot_type,
705 17c3f802 Guido Trotter
                                  ignore_secondaries=opts.ignore_secondaries,
706 4d98c565 Guido Trotter
                                  shutdown_timeout=opts.shutdown_timeout)
707 a8083063 Iustin Pop
708 7c0d6283 Michael Hanselmann
709 1c5945b6 Iustin Pop
def _ShutdownInstance(name, opts):
710 a8083063 Iustin Pop
  """Shutdown an instance.
711 a8083063 Iustin Pop

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

715 1c5945b6 Iustin Pop
  @param name: the name of the instance to act on
716 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
717 1c5945b6 Iustin Pop
  @return: the opcode needed for the operation
718 a8083063 Iustin Pop

719 a8083063 Iustin Pop
  """
720 6263189c Guido Trotter
  return opcodes.OpShutdownInstance(instance_name=name,
721 b44bd844 Michael Hanselmann
                                    timeout=opts.timeout,
722 b44bd844 Michael Hanselmann
                                    ignore_offline_nodes=opts.ignore_offline)
723 a8083063 Iustin Pop
724 a8083063 Iustin Pop
725 a8083063 Iustin Pop
def ReplaceDisks(opts, args):
726 a8083063 Iustin Pop
  """Replace the disks of an instance
727 a8083063 Iustin Pop

728 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
729 7232c04c Iustin Pop
  @type args: list
730 7232c04c Iustin Pop
  @param args: should contain only one element, the instance name
731 7232c04c Iustin Pop
  @rtype: int
732 7232c04c Iustin Pop
  @return: the desired exit code
733 a8083063 Iustin Pop

734 a8083063 Iustin Pop
  """
735 a14db5ff Iustin Pop
  new_2ndary = opts.dst_node
736 b6e82a65 Iustin Pop
  iallocator = opts.iallocator
737 a9e0c397 Iustin Pop
  if opts.disks is None:
738 54155f52 Iustin Pop
    disks = []
739 a9e0c397 Iustin Pop
  else:
740 54155f52 Iustin Pop
    try:
741 54155f52 Iustin Pop
      disks = [int(i) for i in opts.disks.split(",")]
742 691744c4 Iustin Pop
    except (TypeError, ValueError), err:
743 debac808 Iustin Pop
      raise errors.OpPrereqError("Invalid disk index passed: %s" % str(err),
744 debac808 Iustin Pop
                                 errors.ECODE_INVAL)
745 05d47e33 Michael Hanselmann
  cnt = [opts.on_primary, opts.on_secondary, opts.auto,
746 7e9366f7 Iustin Pop
         new_2ndary is not None, iallocator is not None].count(True)
747 7e9366f7 Iustin Pop
  if cnt != 1:
748 05d47e33 Michael Hanselmann
    raise errors.OpPrereqError("One and only one of the -p, -s, -a, -n and -i"
749 debac808 Iustin Pop
                               " options must be passed", errors.ECODE_INVAL)
750 7e9366f7 Iustin Pop
  elif opts.on_primary:
751 a9e0c397 Iustin Pop
    mode = constants.REPLACE_DISK_PRI
752 7e9366f7 Iustin Pop
  elif opts.on_secondary:
753 a9e0c397 Iustin Pop
    mode = constants.REPLACE_DISK_SEC
754 05d47e33 Michael Hanselmann
  elif opts.auto:
755 05d47e33 Michael Hanselmann
    mode = constants.REPLACE_DISK_AUTO
756 05d47e33 Michael Hanselmann
    if disks:
757 05d47e33 Michael Hanselmann
      raise errors.OpPrereqError("Cannot specify disks when using automatic"
758 debac808 Iustin Pop
                                 " mode", errors.ECODE_INVAL)
759 7e9366f7 Iustin Pop
  elif new_2ndary is not None or iallocator is not None:
760 7e9366f7 Iustin Pop
    # replace secondary
761 7e9366f7 Iustin Pop
    mode = constants.REPLACE_DISK_CHG
762 a9e0c397 Iustin Pop
763 a9e0c397 Iustin Pop
  op = opcodes.OpReplaceDisks(instance_name=args[0], disks=disks,
764 b6e82a65 Iustin Pop
                              remote_node=new_2ndary, mode=mode,
765 7ea7bcf6 Iustin Pop
                              iallocator=iallocator,
766 7ea7bcf6 Iustin Pop
                              early_release=opts.early_release)
767 6340bb0a Iustin Pop
  SubmitOrSend(op, opts)
768 a8083063 Iustin Pop
  return 0
769 a8083063 Iustin Pop
770 a8083063 Iustin Pop
771 a8083063 Iustin Pop
def FailoverInstance(opts, args):
772 a8083063 Iustin Pop
  """Failover an instance.
773 a8083063 Iustin Pop

774 a8083063 Iustin Pop
  The failover is done by shutting it down on its present node and
775 a8083063 Iustin Pop
  starting it on the secondary.
776 a8083063 Iustin Pop

777 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
778 7232c04c Iustin Pop
  @type args: list
779 7232c04c Iustin Pop
  @param args: should contain only one element, the instance name
780 7232c04c Iustin Pop
  @rtype: int
781 7232c04c Iustin Pop
  @return: the desired exit code
782 a8083063 Iustin Pop

783 a8083063 Iustin Pop
  """
784 a76f0c4a Iustin Pop
  cl = GetClient()
785 80de0e3f Iustin Pop
  instance_name = args[0]
786 80de0e3f Iustin Pop
  force = opts.force
787 a8083063 Iustin Pop
788 80de0e3f Iustin Pop
  if not force:
789 a76f0c4a Iustin Pop
    _EnsureInstancesExist(cl, [instance_name])
790 a76f0c4a Iustin Pop
791 80de0e3f Iustin Pop
    usertext = ("Failover will happen to image %s."
792 80de0e3f Iustin Pop
                " This requires a shutdown of the instance. Continue?" %
793 80de0e3f Iustin Pop
                (instance_name,))
794 80de0e3f Iustin Pop
    if not AskUser(usertext):
795 80de0e3f Iustin Pop
      return 1
796 a8083063 Iustin Pop
797 80de0e3f Iustin Pop
  op = opcodes.OpFailoverInstance(instance_name=instance_name,
798 17c3f802 Guido Trotter
                                  ignore_consistency=opts.ignore_consistency,
799 4d98c565 Guido Trotter
                                  shutdown_timeout=opts.shutdown_timeout)
800 a76f0c4a Iustin Pop
  SubmitOrSend(op, opts, cl=cl)
801 80de0e3f Iustin Pop
  return 0
802 a8083063 Iustin Pop
803 a8083063 Iustin Pop
804 53c776b5 Iustin Pop
def MigrateInstance(opts, args):
805 53c776b5 Iustin Pop
  """Migrate an instance.
806 53c776b5 Iustin Pop

807 53c776b5 Iustin Pop
  The migrate is done without shutdown.
808 53c776b5 Iustin Pop

809 2f907a8c Iustin Pop
  @param opts: the command line options selected by the user
810 2f907a8c Iustin Pop
  @type args: list
811 2f907a8c Iustin Pop
  @param args: should contain only one element, the instance name
812 2f907a8c Iustin Pop
  @rtype: int
813 2f907a8c Iustin Pop
  @return: the desired exit code
814 53c776b5 Iustin Pop

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

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

860 fbf5a861 Iustin Pop
  """
861 fbf5a861 Iustin Pop
  cl = GetClient()
862 fbf5a861 Iustin Pop
  instance_name = args[0]
863 fbf5a861 Iustin Pop
  force = opts.force
864 fbf5a861 Iustin Pop
865 fbf5a861 Iustin Pop
  if not force:
866 fbf5a861 Iustin Pop
    usertext = ("Instance %s will be moved."
867 fbf5a861 Iustin Pop
                " This requires a shutdown of the instance. Continue?" %
868 fbf5a861 Iustin Pop
                (instance_name,))
869 fbf5a861 Iustin Pop
    if not AskUser(usertext):
870 fbf5a861 Iustin Pop
      return 1
871 fbf5a861 Iustin Pop
872 fbf5a861 Iustin Pop
  op = opcodes.OpMoveInstance(instance_name=instance_name,
873 17c3f802 Guido Trotter
                              target_node=opts.node,
874 4d98c565 Guido Trotter
                              shutdown_timeout=opts.shutdown_timeout)
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 a8083063 Iustin Pop
  op = opcodes.OpConnectConsole(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 25ce3ec4 Michael Hanselmann
  """Acts based on the result of L{opcodes.OpConnectConsole}.
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 57821cac Iustin Pop
  op = opcodes.OpQueryInstanceData(instances=args, static=opts.static)
1145 400ca2f7 Iustin Pop
  result = SubmitOpCode(op, opts=opts)
1146 a8083063 Iustin Pop
  if not result:
1147 3a24c527 Iustin Pop
    ToStdout("No instances.")
1148 a8083063 Iustin Pop
    return 1
1149 a8083063 Iustin Pop
1150 a8083063 Iustin Pop
  buf = StringIO()
1151 a8083063 Iustin Pop
  retcode = 0
1152 a8083063 Iustin Pop
  for instance_name in result:
1153 a8083063 Iustin Pop
    instance = result[instance_name]
1154 a8083063 Iustin Pop
    buf.write("Instance name: %s\n" % instance["name"])
1155 033d58b0 Iustin Pop
    buf.write("UUID: %s\n" % instance["uuid"])
1156 e2736e40 Guido Trotter
    buf.write("Serial number: %s\n" %
1157 e2736e40 Guido Trotter
              compat.TryToRoman(instance["serial_no"],
1158 e2736e40 Guido Trotter
                                convert=opts.roman_integers))
1159 90f72445 Iustin Pop
    buf.write("Creation time: %s\n" % utils.FormatTime(instance["ctime"]))
1160 90f72445 Iustin Pop
    buf.write("Modification time: %s\n" % utils.FormatTime(instance["mtime"]))
1161 57821cac Iustin Pop
    buf.write("State: configured to be %s" % instance["config_state"])
1162 57821cac Iustin Pop
    if not opts.static:
1163 57821cac Iustin Pop
      buf.write(", actual state is %s" % instance["run_state"])
1164 57821cac Iustin Pop
    buf.write("\n")
1165 57821cac Iustin Pop
    ##buf.write("Considered for memory checks in cluster verify: %s\n" %
1166 57821cac Iustin Pop
    ##          instance["auto_balance"])
1167 a8083063 Iustin Pop
    buf.write("  Nodes:\n")
1168 a8083063 Iustin Pop
    buf.write("    - primary: %s\n" % instance["pnode"])
1169 1f864b60 Iustin Pop
    buf.write("    - secondaries: %s\n" % utils.CommaJoin(instance["snodes"]))
1170 a8083063 Iustin Pop
    buf.write("  Operating system: %s\n" % instance["os"])
1171 acd19189 René Nussbaumer
    FormatParameterDict(buf, instance["os_instance"], instance["os_actual"],
1172 acd19189 René Nussbaumer
                        level=2)
1173 a8340917 Iustin Pop
    if instance.has_key("network_port"):
1174 e2736e40 Guido Trotter
      buf.write("  Allocated network port: %s\n" %
1175 e2736e40 Guido Trotter
                compat.TryToRoman(instance["network_port"],
1176 e2736e40 Guido Trotter
                                  convert=opts.roman_integers))
1177 24838135 Iustin Pop
    buf.write("  Hypervisor: %s\n" % instance["hypervisor"])
1178 dfff41f8 Guido Trotter
1179 dfff41f8 Guido Trotter
    # custom VNC console information
1180 dfff41f8 Guido Trotter
    vnc_bind_address = instance["hv_actual"].get(constants.HV_VNC_BIND_ADDRESS,
1181 dfff41f8 Guido Trotter
                                                 None)
1182 dfff41f8 Guido Trotter
    if vnc_bind_address:
1183 dfff41f8 Guido Trotter
      port = instance["network_port"]
1184 dfff41f8 Guido Trotter
      display = int(port) - constants.VNC_BASE_PORT
1185 9769bb78 Manuel Franceschini
      if display > 0 and vnc_bind_address == constants.IP4_ADDRESS_ANY:
1186 dfff41f8 Guido Trotter
        vnc_console_port = "%s:%s (display %s)" % (instance["pnode"],
1187 dfff41f8 Guido Trotter
                                                   port,
1188 dfff41f8 Guido Trotter
                                                   display)
1189 8b312c1d Manuel Franceschini
      elif display > 0 and netutils.IP4Address.IsValid(vnc_bind_address):
1190 dfff41f8 Guido Trotter
        vnc_console_port = ("%s:%s (node %s) (display %s)" %
1191 dfff41f8 Guido Trotter
                             (vnc_bind_address, port,
1192 dfff41f8 Guido Trotter
                              instance["pnode"], display))
1193 a8340917 Iustin Pop
      else:
1194 dfff41f8 Guido Trotter
        # vnc bind address is a file
1195 dfff41f8 Guido Trotter
        vnc_console_port = "%s:%s" % (instance["pnode"],
1196 dfff41f8 Guido Trotter
                                      vnc_bind_address)
1197 24838135 Iustin Pop
      buf.write("    - console connection: vnc to %s\n" % vnc_console_port)
1198 24838135 Iustin Pop
1199 acd19189 René Nussbaumer
    FormatParameterDict(buf, instance["hv_instance"], instance["hv_actual"],
1200 acd19189 René Nussbaumer
                        level=2)
1201 a8083063 Iustin Pop
    buf.write("  Hardware:\n")
1202 e2736e40 Guido Trotter
    buf.write("    - VCPUs: %s\n" %
1203 e2736e40 Guido Trotter
              compat.TryToRoman(instance["be_actual"][constants.BE_VCPUS],
1204 e2736e40 Guido Trotter
                                convert=opts.roman_integers))
1205 e2736e40 Guido Trotter
    buf.write("    - memory: %sMiB\n" %
1206 e2736e40 Guido Trotter
              compat.TryToRoman(instance["be_actual"][constants.BE_MEMORY],
1207 e2736e40 Guido Trotter
                                convert=opts.roman_integers))
1208 d2acfe27 Iustin Pop
    buf.write("    - NICs:\n")
1209 14ea9302 Guido Trotter
    for idx, (ip, mac, mode, link) in enumerate(instance["nics"]):
1210 0b13832c Guido Trotter
      buf.write("      - nic/%d: MAC: %s, IP: %s, mode: %s, link: %s\n" %
1211 0b13832c Guido Trotter
                (idx, mac, ip, mode, link))
1212 b577dac4 Michael Hanselmann
    buf.write("  Disk template: %s\n" % instance["disk_template"])
1213 19708787 Iustin Pop
    buf.write("  Disks:\n")
1214 a8083063 Iustin Pop
1215 19708787 Iustin Pop
    for idx, device in enumerate(instance["disks"]):
1216 e2736e40 Guido Trotter
      _FormatList(buf, _FormatBlockDevInfo(idx, True, device, opts.static,
1217 e2736e40 Guido Trotter
                  opts.roman_integers), 2)
1218 a8083063 Iustin Pop
1219 3a24c527 Iustin Pop
  ToStdout(buf.getvalue().rstrip('\n'))
1220 a8083063 Iustin Pop
  return retcode
1221 a8083063 Iustin Pop
1222 a8083063 Iustin Pop
1223 7767bbf5 Manuel Franceschini
def SetInstanceParams(opts, args):
1224 a8083063 Iustin Pop
  """Modifies an instance.
1225 a8083063 Iustin Pop

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

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

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