Statistics
| Branch: | Tag: | Revision:

root / lib / client / gnt_instance.py @ 080fbeea

History | View | Annotate | Download (55.1 kB)

1 e792102d Michael Hanselmann
#
2 a8083063 Iustin Pop
#
3 a8083063 Iustin Pop
4 60472d29 Iustin Pop
# Copyright (C) 2006, 2007, 2008, 2009, 2010, 2011 Google Inc.
5 a8083063 Iustin Pop
#
6 a8083063 Iustin Pop
# This program is free software; you can redistribute it and/or modify
7 a8083063 Iustin Pop
# it under the terms of the GNU General Public License as published by
8 a8083063 Iustin Pop
# the Free Software Foundation; either version 2 of the License, or
9 a8083063 Iustin Pop
# (at your option) any later version.
10 a8083063 Iustin Pop
#
11 a8083063 Iustin Pop
# This program is distributed in the hope that it will be useful, but
12 a8083063 Iustin Pop
# WITHOUT ANY WARRANTY; without even the implied warranty of
13 a8083063 Iustin Pop
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 a8083063 Iustin Pop
# General Public License for more details.
15 a8083063 Iustin Pop
#
16 a8083063 Iustin Pop
# You should have received a copy of the GNU General Public License
17 a8083063 Iustin Pop
# along with this program; if not, write to the Free Software
18 a8083063 Iustin Pop
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
19 a8083063 Iustin Pop
# 02110-1301, USA.
20 a8083063 Iustin Pop
21 7260cfbe Iustin Pop
"""Instance related commands"""
22 a8083063 Iustin Pop
23 b459a848 Andrea Spadaccini
# pylint: disable=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 c20efaa8 Michael Hanselmann
_EXPAND_CLUSTER = "cluster"
45 c20efaa8 Michael Hanselmann
_EXPAND_NODES_BOTH = "nodes"
46 c20efaa8 Michael Hanselmann
_EXPAND_NODES_PRI = "nodes-pri"
47 c20efaa8 Michael Hanselmann
_EXPAND_NODES_SEC = "nodes-sec"
48 c20efaa8 Michael Hanselmann
_EXPAND_NODES_BOTH_BY_TAGS = "nodes-by-tags"
49 c20efaa8 Michael Hanselmann
_EXPAND_NODES_PRI_BY_TAGS = "nodes-pri-by-tags"
50 c20efaa8 Michael Hanselmann
_EXPAND_NODES_SEC_BY_TAGS = "nodes-sec-by-tags"
51 c20efaa8 Michael Hanselmann
_EXPAND_INSTANCES = "instances"
52 c20efaa8 Michael Hanselmann
_EXPAND_INSTANCES_BY_TAGS = "instances-by-tags"
53 c20efaa8 Michael Hanselmann
54 c20efaa8 Michael Hanselmann
_EXPAND_NODES_TAGS_MODES = frozenset([
55 c20efaa8 Michael Hanselmann
  _EXPAND_NODES_BOTH_BY_TAGS,
56 c20efaa8 Michael Hanselmann
  _EXPAND_NODES_PRI_BY_TAGS,
57 c20efaa8 Michael Hanselmann
  _EXPAND_NODES_SEC_BY_TAGS,
58 c20efaa8 Michael Hanselmann
  ])
59 312ac745 Iustin Pop
60 7c0d6283 Michael Hanselmann
61 7232c04c Iustin Pop
#: default list of options for L{ListInstances}
62 48c4dfa8 Iustin Pop
_LIST_DEF_FIELDS = [
63 e69d05fd Iustin Pop
  "name", "hypervisor", "os", "pnode", "status", "oper_ram",
64 48c4dfa8 Iustin Pop
  ]
65 48c4dfa8 Iustin Pop
66 bdb7d4e8 Michael Hanselmann
67 479636a3 Iustin Pop
def _ExpandMultiNames(mode, names, client=None):
68 312ac745 Iustin Pop
  """Expand the given names using the passed mode.
69 312ac745 Iustin Pop

70 c20efaa8 Michael Hanselmann
  For _EXPAND_CLUSTER, all instances will be returned. For
71 c20efaa8 Michael Hanselmann
  _EXPAND_NODES_PRI/SEC, all instances having those nodes as
72 c20efaa8 Michael Hanselmann
  primary/secondary will be returned. For _EXPAND_NODES_BOTH, all
73 312ac745 Iustin Pop
  instances having those nodes as either primary or secondary will be
74 c20efaa8 Michael Hanselmann
  returned. For _EXPAND_INSTANCES, the given instances will be
75 312ac745 Iustin Pop
  returned.
76 312ac745 Iustin Pop

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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 a8083063 Iustin Pop

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

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

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

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

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

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

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

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

807 53c776b5 Iustin Pop
  """
808 a76f0c4a Iustin Pop
  cl = GetClient()
809 53c776b5 Iustin Pop
  instance_name = args[0]
810 53c776b5 Iustin Pop
  force = opts.force
811 1b7761fd Apollon Oikonomopoulos
  iallocator = opts.iallocator
812 1b7761fd Apollon Oikonomopoulos
  target_node = opts.dst_node
813 1b7761fd Apollon Oikonomopoulos
814 1b7761fd Apollon Oikonomopoulos
  if iallocator and target_node:
815 1b7761fd Apollon Oikonomopoulos
    raise errors.OpPrereqError("Specify either an iallocator (-I), or a target"
816 1b7761fd Apollon Oikonomopoulos
                               " node (-n) but not both", errors.ECODE_INVAL)
817 53c776b5 Iustin Pop
818 53c776b5 Iustin Pop
  if not force:
819 a76f0c4a Iustin Pop
    _EnsureInstancesExist(cl, [instance_name])
820 a76f0c4a Iustin Pop
821 53c776b5 Iustin Pop
    if opts.cleanup:
822 53c776b5 Iustin Pop
      usertext = ("Instance %s will be recovered from a failed migration."
823 53c776b5 Iustin Pop
                  " Note that the migration procedure (including cleanup)" %
824 53c776b5 Iustin Pop
                  (instance_name,))
825 53c776b5 Iustin Pop
    else:
826 53c776b5 Iustin Pop
      usertext = ("Instance %s will be migrated. Note that migration" %
827 53c776b5 Iustin Pop
                  (instance_name,))
828 cf29cfb6 Iustin Pop
    usertext += (" might impact the instance if anything goes wrong"
829 cf29cfb6 Iustin Pop
                 " (e.g. due to bugs in the hypervisor). Continue?")
830 53c776b5 Iustin Pop
    if not AskUser(usertext):
831 53c776b5 Iustin Pop
      return 1
832 53c776b5 Iustin Pop
833 e71b9ef4 Iustin Pop
  # this should be removed once --non-live is deprecated
834 783a6c0b Iustin Pop
  if not opts.live and opts.migration_mode is not None:
835 e71b9ef4 Iustin Pop
    raise errors.OpPrereqError("Only one of the --non-live and "
836 783a6c0b Iustin Pop
                               "--migration-mode options can be passed",
837 e71b9ef4 Iustin Pop
                               errors.ECODE_INVAL)
838 e71b9ef4 Iustin Pop
  if not opts.live: # --non-live passed
839 8c35561f Iustin Pop
    mode = constants.HT_MIGRATION_NONLIVE
840 e71b9ef4 Iustin Pop
  else:
841 8c35561f Iustin Pop
    mode = opts.migration_mode
842 e71b9ef4 Iustin Pop
843 75c866c2 Iustin Pop
  op = opcodes.OpInstanceMigrate(instance_name=instance_name, mode=mode,
844 1b7761fd Apollon Oikonomopoulos
                                 cleanup=opts.cleanup, iallocator=iallocator,
845 e9c487be René Nussbaumer
                                 target_node=target_node,
846 e9c487be René Nussbaumer
                                 allow_failover=opts.allow_failover)
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 0091b480 Iustin Pop
  op = opcodes.OpInstanceMove(instance_name=instance_name,
873 17c3f802 Guido Trotter
                              target_node=opts.node,
874 bb851c63 Iustin Pop
                              shutdown_timeout=opts.shutdown_timeout,
875 bb851c63 Iustin Pop
                              ignore_consistency=opts.ignore_consistency)
876 fbf5a861 Iustin Pop
  SubmitOrSend(op, opts, cl=cl)
877 fbf5a861 Iustin Pop
  return 0
878 fbf5a861 Iustin Pop
879 fbf5a861 Iustin Pop
880 a8083063 Iustin Pop
def ConnectToInstanceConsole(opts, args):
881 a8083063 Iustin Pop
  """Connect to the console of an instance.
882 a8083063 Iustin Pop

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

889 a8083063 Iustin Pop
  """
890 a8083063 Iustin Pop
  instance_name = args[0]
891 a8083063 Iustin Pop
892 25ce3ec4 Michael Hanselmann
  cl = GetClient()
893 25ce3ec4 Michael Hanselmann
  try:
894 25ce3ec4 Michael Hanselmann
    cluster_name = cl.QueryConfigValues(["cluster_name"])[0]
895 d6f46b6a Michael Hanselmann
    ((console_data, oper_state), ) = \
896 d6f46b6a Michael Hanselmann
      cl.QueryInstances([instance_name], ["console", "oper_state"], False)
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 d6f46b6a Michael Hanselmann
  if not console_data:
904 d6f46b6a Michael Hanselmann
    if oper_state:
905 d6f46b6a Michael Hanselmann
      # Instance is running
906 d6f46b6a Michael Hanselmann
      raise errors.OpExecError("Console information for instance %s is"
907 d6f46b6a Michael Hanselmann
                               " unavailable" % instance_name)
908 d6f46b6a Michael Hanselmann
    else:
909 d6f46b6a Michael Hanselmann
      raise errors.OpExecError("Instance %s is not running, can't get console" %
910 d6f46b6a Michael Hanselmann
                               instance_name)
911 d6f46b6a Michael Hanselmann
912 25ce3ec4 Michael Hanselmann
  return _DoConsole(objects.InstanceConsole.FromDict(console_data),
913 25ce3ec4 Michael Hanselmann
                    opts.show_command, cluster_name)
914 25ce3ec4 Michael Hanselmann
915 25ce3ec4 Michael Hanselmann
916 25ce3ec4 Michael Hanselmann
def _DoConsole(console, show_command, cluster_name, feedback_fn=ToStdout,
917 25ce3ec4 Michael Hanselmann
               _runcmd_fn=utils.RunCmd):
918 cc0dec7b Iustin Pop
  """Acts based on the result of L{opcodes.OpInstanceConsole}.
919 25ce3ec4 Michael Hanselmann

920 25ce3ec4 Michael Hanselmann
  @type console: L{objects.InstanceConsole}
921 25ce3ec4 Michael Hanselmann
  @param console: Console object
922 25ce3ec4 Michael Hanselmann
  @type show_command: bool
923 25ce3ec4 Michael Hanselmann
  @param show_command: Whether to just display commands
924 25ce3ec4 Michael Hanselmann
  @type cluster_name: string
925 25ce3ec4 Michael Hanselmann
  @param cluster_name: Cluster name as retrieved from master daemon
926 25ce3ec4 Michael Hanselmann

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

971 19708787 Iustin Pop
  """
972 19708787 Iustin Pop
  if dev_type == constants.LD_DRBD8:
973 19708787 Iustin Pop
    node_a, node_b, port, minor_a, minor_b, key = logical_id
974 19708787 Iustin Pop
    data = [
975 e2736e40 Guido Trotter
      ("nodeA", "%s, minor=%s" % (node_a, compat.TryToRoman(minor_a,
976 e2736e40 Guido Trotter
                                                            convert=roman))),
977 e2736e40 Guido Trotter
      ("nodeB", "%s, minor=%s" % (node_b, compat.TryToRoman(minor_b,
978 e2736e40 Guido Trotter
                                                            convert=roman))),
979 e2736e40 Guido Trotter
      ("port", compat.TryToRoman(port, convert=roman)),
980 19708787 Iustin Pop
      ("auth key", key),
981 19708787 Iustin Pop
      ]
982 19708787 Iustin Pop
  elif dev_type == constants.LD_LV:
983 19708787 Iustin Pop
    vg_name, lv_name = logical_id
984 19708787 Iustin Pop
    data = ["%s/%s" % (vg_name, lv_name)]
985 19708787 Iustin Pop
  else:
986 19708787 Iustin Pop
    data = [str(logical_id)]
987 19708787 Iustin Pop
988 19708787 Iustin Pop
  return data
989 19708787 Iustin Pop
990 19708787 Iustin Pop
991 f965260c Michael Hanselmann
def _FormatBlockDevInfo(idx, top_level, dev, roman):
992 a8083063 Iustin Pop
  """Show block device information.
993 a8083063 Iustin Pop

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

997 19708787 Iustin Pop
  @type idx: int
998 19708787 Iustin Pop
  @param idx: the index of the current disk
999 19708787 Iustin Pop
  @type top_level: boolean
1000 19708787 Iustin Pop
  @param top_level: if this a top-level disk?
1001 7232c04c Iustin Pop
  @type dev: dict
1002 7232c04c Iustin Pop
  @param dev: dictionary with disk information
1003 e2736e40 Guido Trotter
  @type roman: boolean
1004 e2736e40 Guido Trotter
  @param roman: whether to try to use roman integers
1005 19708787 Iustin Pop
  @return: a list of either strings, tuples or lists
1006 19708787 Iustin Pop
      (which should be formatted at a higher indent level)
1007 7232c04c Iustin Pop

1008 a8083063 Iustin Pop
  """
1009 19708787 Iustin Pop
  def helper(dtype, status):
1010 7232c04c Iustin Pop
    """Format one line for physical device status.
1011 7232c04c Iustin Pop

1012 7232c04c Iustin Pop
    @type dtype: str
1013 7232c04c Iustin Pop
    @param dtype: a constant from the L{constants.LDS_BLOCK} set
1014 7232c04c Iustin Pop
    @type status: tuple
1015 7232c04c Iustin Pop
    @param status: a tuple as returned from L{backend.FindBlockDevice}
1016 19708787 Iustin Pop
    @return: the string representing the status
1017 7232c04c Iustin Pop

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

1108 19708787 Iustin Pop
  If the element of the list is:
1109 19708787 Iustin Pop
    - a string, it is simply formatted as is
1110 19708787 Iustin Pop
    - a tuple, it will be split into key, value and the all the
1111 19708787 Iustin Pop
      values in a list will be aligned all at the same start column
1112 19708787 Iustin Pop
    - a list, will be recursively formatted
1113 19708787 Iustin Pop

1114 19708787 Iustin Pop
  @type buf: StringIO
1115 19708787 Iustin Pop
  @param buf: the buffer into which we write the output
1116 19708787 Iustin Pop
  @param data: the list to format
1117 19708787 Iustin Pop
  @type indent_level: int
1118 19708787 Iustin Pop
  @param indent_level: the indent level to format at
1119 19708787 Iustin Pop

1120 19708787 Iustin Pop
  """
1121 19708787 Iustin Pop
  max_tlen = max([len(elem[0]) for elem in data
1122 19708787 Iustin Pop
                 if isinstance(elem, tuple)] or [0])
1123 19708787 Iustin Pop
  for elem in data:
1124 19708787 Iustin Pop
    if isinstance(elem, basestring):
1125 e687ec01 Michael Hanselmann
      buf.write("%*s%s\n" % (2 * indent_level, "", elem))
1126 19708787 Iustin Pop
    elif isinstance(elem, tuple):
1127 19708787 Iustin Pop
      key, value = elem
1128 19708787 Iustin Pop
      spacer = "%*s" % (max_tlen - len(key), "")
1129 e687ec01 Michael Hanselmann
      buf.write("%*s%s:%s %s\n" % (2 * indent_level, "", key, spacer, value))
1130 19708787 Iustin Pop
    elif isinstance(elem, list):
1131 e687ec01 Michael Hanselmann
      _FormatList(buf, elem, indent_level + 1)
1132 19708787 Iustin Pop
1133 98825740 Michael Hanselmann
1134 a8083063 Iustin Pop
def ShowInstanceConfig(opts, args):
1135 a8083063 Iustin Pop
  """Compute instance run-time status.
1136 a8083063 Iustin Pop

1137 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
1138 7232c04c Iustin Pop
  @type args: list
1139 7232c04c Iustin Pop
  @param args: either an empty list, and then we query all
1140 7232c04c Iustin Pop
      instances, or should contain a list of instance names
1141 7232c04c Iustin Pop
  @rtype: int
1142 7232c04c Iustin Pop
  @return: the desired exit code
1143 7232c04c Iustin Pop

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

1246 a8083063 Iustin Pop
  All parameters take effect only at the next restart of the instance.
1247 a8083063 Iustin Pop

1248 7232c04c Iustin Pop
  @param opts: the command line options selected by the user
1249 7232c04c Iustin Pop
  @type args: list
1250 7232c04c Iustin Pop
  @param args: should contain only one element, the instance name
1251 7232c04c Iustin Pop
  @rtype: int
1252 7232c04c Iustin Pop
  @return: the desired exit code
1253 a8083063 Iustin Pop

1254 a8083063 Iustin Pop
  """
1255 e29e9550 Iustin Pop
  if not (opts.nics or opts.disks or opts.disk_template or
1256 1052d622 Iustin Pop
          opts.hvparams or opts.beparams or opts.os or opts.osparams):
1257 3a24c527 Iustin Pop
    ToStderr("Please give at least one of the parameters.")
1258 a8083063 Iustin Pop
    return 1
1259 a8083063 Iustin Pop
1260 467ae11e Guido Trotter
  for param in opts.beparams:
1261 e9d622bc Guido Trotter
    if isinstance(opts.beparams[param], basestring):
1262 e9d622bc Guido Trotter
      if opts.beparams[param].lower() == "default":
1263 e9d622bc Guido Trotter
        opts.beparams[param] = constants.VALUE_DEFAULT
1264 a5728081 Guido Trotter
1265 a5728081 Guido Trotter
  utils.ForceDictType(opts.beparams, constants.BES_PARAMETER_TYPES,
1266 a5728081 Guido Trotter
                      allowed_values=[constants.VALUE_DEFAULT])
1267 467ae11e Guido Trotter
1268 48f212d7 Iustin Pop
  for param in opts.hvparams:
1269 48f212d7 Iustin Pop
    if isinstance(opts.hvparams[param], basestring):
1270 48f212d7 Iustin Pop
      if opts.hvparams[param].lower() == "default":
1271 48f212d7 Iustin Pop
        opts.hvparams[param] = constants.VALUE_DEFAULT
1272 a5728081 Guido Trotter
1273 48f212d7 Iustin Pop
  utils.ForceDictType(opts.hvparams, constants.HVS_PARAMETER_TYPES,
1274 a5728081 Guido Trotter
                      allowed_values=[constants.VALUE_DEFAULT])
1275 61be6ba4 Iustin Pop
1276 24991749 Iustin Pop
  for idx, (nic_op, nic_dict) in enumerate(opts.nics):
1277 24991749 Iustin Pop
    try:
1278 24991749 Iustin Pop
      nic_op = int(nic_op)
1279 24991749 Iustin Pop
      opts.nics[idx] = (nic_op, nic_dict)
1280 691744c4 Iustin Pop
    except (TypeError, ValueError):
1281 24991749 Iustin Pop
      pass
1282 24991749 Iustin Pop
1283 24991749 Iustin Pop
  for idx, (disk_op, disk_dict) in enumerate(opts.disks):
1284 24991749 Iustin Pop
    try:
1285 24991749 Iustin Pop
      disk_op = int(disk_op)
1286 24991749 Iustin Pop
      opts.disks[idx] = (disk_op, disk_dict)
1287 691744c4 Iustin Pop
    except (TypeError, ValueError):
1288 24991749 Iustin Pop
      pass
1289 24991749 Iustin Pop
    if disk_op == constants.DDM_ADD:
1290 d0c8c01d Iustin Pop
      if "size" not in disk_dict:
1291 debac808 Iustin Pop
        raise errors.OpPrereqError("Missing required parameter 'size'",
1292 debac808 Iustin Pop
                                   errors.ECODE_INVAL)
1293 d0c8c01d Iustin Pop
      disk_dict["size"] = utils.ParseUnit(disk_dict["size"])
1294 24991749 Iustin Pop
1295 e29e9550 Iustin Pop
  if (opts.disk_template and
1296 3429a076 Apollon Oikonomopoulos
      opts.disk_template in constants.DTS_INT_MIRROR and
1297 e29e9550 Iustin Pop
      not opts.node):
1298 e29e9550 Iustin Pop
    ToStderr("Changing the disk template to a mirrored one requires"
1299 e29e9550 Iustin Pop
             " specifying a secondary node")
1300 e29e9550 Iustin Pop
    return 1
1301 e29e9550 Iustin Pop
1302 9a3cc7ae Iustin Pop
  op = opcodes.OpInstanceSetParams(instance_name=args[0],
1303 24991749 Iustin Pop
                                   nics=opts.nics,
1304 24991749 Iustin Pop
                                   disks=opts.disks,
1305 e29e9550 Iustin Pop
                                   disk_template=opts.disk_template,
1306 e29e9550 Iustin Pop
                                   remote_node=opts.node,
1307 48f212d7 Iustin Pop
                                   hvparams=opts.hvparams,
1308 338e51e8 Iustin Pop
                                   beparams=opts.beparams,
1309 96b39bcc Iustin Pop
                                   os_name=opts.os,
1310 1052d622 Iustin Pop
                                   osparams=opts.osparams,
1311 96b39bcc Iustin Pop
                                   force_variant=opts.force_variant,
1312 456798ab Iustin Pop
                                   force=opts.force,
1313 456798ab Iustin Pop
                                   wait_for_sync=opts.wait_for_sync)
1314 31a853d2 Iustin Pop
1315 6340bb0a Iustin Pop
  # even if here we process the result, we allow submit only
1316 6340bb0a Iustin Pop
  result = SubmitOrSend(op, opts)
1317 a8083063 Iustin Pop
1318 a8083063 Iustin Pop
  if result:
1319 3a24c527 Iustin Pop
    ToStdout("Modified instance %s", args[0])
1320 a8083063 Iustin Pop
    for param, data in result:
1321 3a24c527 Iustin Pop
      ToStdout(" - %-5s -> %s", param, data)
1322 e29e9550 Iustin Pop
    ToStdout("Please don't forget that most parameters take effect"
1323 3a24c527 Iustin Pop
             " only at the next start of the instance.")
1324 a8083063 Iustin Pop
  return 0
1325 a8083063 Iustin Pop
1326 a8083063 Iustin Pop
1327 bd2a5569 Michael Hanselmann
def ChangeGroup(opts, args):
1328 bd2a5569 Michael Hanselmann
  """Moves an instance to another group.
1329 bd2a5569 Michael Hanselmann

1330 bd2a5569 Michael Hanselmann
  """
1331 bd2a5569 Michael Hanselmann
  (instance_name, ) = args
1332 bd2a5569 Michael Hanselmann
1333 bd2a5569 Michael Hanselmann
  cl = GetClient()
1334 bd2a5569 Michael Hanselmann
1335 bd2a5569 Michael Hanselmann
  op = opcodes.OpInstanceChangeGroup(instance_name=instance_name,
1336 bd2a5569 Michael Hanselmann
                                     iallocator=opts.iallocator,
1337 bd2a5569 Michael Hanselmann
                                     target_groups=opts.to,
1338 bd2a5569 Michael Hanselmann
                                     early_release=opts.early_release)
1339 bd2a5569 Michael Hanselmann
  result = SubmitOpCode(op, cl=cl, opts=opts)
1340 bd2a5569 Michael Hanselmann
1341 bd2a5569 Michael Hanselmann
  # Keep track of submitted jobs
1342 bd2a5569 Michael Hanselmann
  jex = JobExecutor(cl=cl, opts=opts)
1343 bd2a5569 Michael Hanselmann
1344 bd2a5569 Michael Hanselmann
  for (status, job_id) in result[constants.JOB_IDS_KEY]:
1345 bd2a5569 Michael Hanselmann
    jex.AddJobId(None, status, job_id)
1346 bd2a5569 Michael Hanselmann
1347 bd2a5569 Michael Hanselmann
  results = jex.GetResults()
1348 bd2a5569 Michael Hanselmann
  bad_cnt = len([row for row in results if not row[0]])
1349 bd2a5569 Michael Hanselmann
  if bad_cnt == 0:
1350 bd2a5569 Michael Hanselmann
    ToStdout("Instance '%s' changed group successfully.", instance_name)
1351 bd2a5569 Michael Hanselmann
    rcode = constants.EXIT_SUCCESS
1352 bd2a5569 Michael Hanselmann
  else:
1353 bd2a5569 Michael Hanselmann
    ToStdout("There were %s errors while changing group of instance '%s'.",
1354 bd2a5569 Michael Hanselmann
             bad_cnt, instance_name)
1355 bd2a5569 Michael Hanselmann
    rcode = constants.EXIT_FAILURE
1356 bd2a5569 Michael Hanselmann
1357 bd2a5569 Michael Hanselmann
  return rcode
1358 bd2a5569 Michael Hanselmann
1359 bd2a5569 Michael Hanselmann
1360 312ac745 Iustin Pop
# multi-instance selection options
1361 c38c44ad Michael Hanselmann
m_force_multi = cli_option("--force-multiple", dest="force_multi",
1362 c38c44ad Michael Hanselmann
                           help="Do not ask for confirmation when more than"
1363 c38c44ad Michael Hanselmann
                           " one instance is affected",
1364 c38c44ad Michael Hanselmann
                           action="store_true", default=False)
1365 804a1e8e Iustin Pop
1366 c38c44ad Michael Hanselmann
m_pri_node_opt = cli_option("--primary", dest="multi_mode",
1367 c38c44ad Michael Hanselmann
                            help="Filter by nodes (primary only)",
1368 c20efaa8 Michael Hanselmann
                            const=_EXPAND_NODES_PRI, action="store_const")
1369 312ac745 Iustin Pop
1370 c38c44ad Michael Hanselmann
m_sec_node_opt = cli_option("--secondary", dest="multi_mode",
1371 c38c44ad Michael Hanselmann
                            help="Filter by nodes (secondary only)",
1372 c20efaa8 Michael Hanselmann
                            const=_EXPAND_NODES_SEC, action="store_const")
1373 312ac745 Iustin Pop
1374 c38c44ad Michael Hanselmann
m_node_opt = cli_option("--node", dest="multi_mode",
1375 c38c44ad Michael Hanselmann
                        help="Filter by nodes (primary and secondary)",
1376 c20efaa8 Michael Hanselmann
                        const=_EXPAND_NODES_BOTH, action="store_const")
1377 312ac745 Iustin Pop
1378 c38c44ad Michael Hanselmann
m_clust_opt = cli_option("--all", dest="multi_mode",
1379 c38c44ad Michael Hanselmann
                         help="Select all instances in the cluster",
1380 c20efaa8 Michael Hanselmann
                         const=_EXPAND_CLUSTER, action="store_const")
1381 312ac745 Iustin Pop
1382 c38c44ad Michael Hanselmann
m_inst_opt = cli_option("--instance", dest="multi_mode",
1383 c38c44ad Michael Hanselmann
                        help="Filter by instance name [default]",
1384 c20efaa8 Michael Hanselmann
                        const=_EXPAND_INSTANCES, action="store_const")
1385 312ac745 Iustin Pop
1386 39dfd93e René Nussbaumer
m_node_tags_opt = cli_option("--node-tags", dest="multi_mode",
1387 39dfd93e René Nussbaumer
                             help="Filter by node tag",
1388 c20efaa8 Michael Hanselmann
                             const=_EXPAND_NODES_BOTH_BY_TAGS,
1389 39dfd93e René Nussbaumer
                             action="store_const")
1390 39dfd93e René Nussbaumer
1391 39dfd93e René Nussbaumer
m_pri_node_tags_opt = cli_option("--pri-node-tags", dest="multi_mode",
1392 39dfd93e René Nussbaumer
                                 help="Filter by primary node tag",
1393 c20efaa8 Michael Hanselmann
                                 const=_EXPAND_NODES_PRI_BY_TAGS,
1394 39dfd93e René Nussbaumer
                                 action="store_const")
1395 39dfd93e René Nussbaumer
1396 39dfd93e René Nussbaumer
m_sec_node_tags_opt = cli_option("--sec-node-tags", dest="multi_mode",
1397 39dfd93e René Nussbaumer
                                 help="Filter by secondary node tag",
1398 c20efaa8 Michael Hanselmann
                                 const=_EXPAND_NODES_SEC_BY_TAGS,
1399 39dfd93e René Nussbaumer
                                 action="store_const")
1400 39dfd93e René Nussbaumer
1401 39dfd93e René Nussbaumer
m_inst_tags_opt = cli_option("--tags", dest="multi_mode",
1402 39dfd93e René Nussbaumer
                             help="Filter by instance tag",
1403 c20efaa8 Michael Hanselmann
                             const=_EXPAND_INSTANCES_BY_TAGS,
1404 39dfd93e René Nussbaumer
                             action="store_const")
1405 312ac745 Iustin Pop
1406 a8083063 Iustin Pop
# this is defined separately due to readability only
1407 a8083063 Iustin Pop
add_opts = [
1408 064c21f8 Iustin Pop
  NOSTART_OPT,
1409 064c21f8 Iustin Pop
  OS_OPT,
1410 06073e85 Guido Trotter
  FORCE_VARIANT_OPT,
1411 25a8792c Iustin Pop
  NO_INSTALL_OPT,
1412 a8083063 Iustin Pop
  ]
1413 a8083063 Iustin Pop
1414 a8083063 Iustin Pop
commands = {
1415 d0c8c01d Iustin Pop
  "add": (
1416 eb28ecf6 Guido Trotter
    AddInstance, [ArgHost(min=1, max=1)], COMMON_CREATE_OPTS + add_opts,
1417 6ea815cf Iustin Pop
    "[...] -t disk-type -n node[:secondary-node] -o os-type <name>",
1418 6ea815cf Iustin Pop
    "Creates and adds a new instance to the cluster"),
1419 d0c8c01d Iustin Pop
  "batch-create": (
1420 aa06f8c6 Michael Hanselmann
    BatchCreate, [ArgFile(min=1, max=1)], [DRY_RUN_OPT, PRIORITY_OPT],
1421 6ea815cf Iustin Pop
    "<instances.json>",
1422 6ea815cf Iustin Pop
    "Create a bunch of instances based on specs in the file."),
1423 d0c8c01d Iustin Pop
  "console": (
1424 6ea815cf Iustin Pop
    ConnectToInstanceConsole, ARGS_ONE_INSTANCE,
1425 aa06f8c6 Michael Hanselmann
    [SHOWCMD_OPT, PRIORITY_OPT],
1426 6ea815cf Iustin Pop
    "[--show-cmd] <instance>", "Opens a console on the specified instance"),
1427 d0c8c01d Iustin Pop
  "failover": (
1428 6ea815cf Iustin Pop
    FailoverInstance, ARGS_ONE_INSTANCE,
1429 db5a8a2d Iustin Pop
    [FORCE_OPT, IGNORE_CONSIST_OPT, SUBMIT_OPT, SHUTDOWN_TIMEOUT_OPT,
1430 1b7761fd Apollon Oikonomopoulos
     DRY_RUN_OPT, PRIORITY_OPT, DST_NODE_OPT, IALLOCATOR_OPT],
1431 a6a3efe4 Iustin Pop
    "[-f] <instance>", "Stops the instance, changes its primary node and"
1432 a6a3efe4 Iustin Pop
    " (if it was originally running) starts it on the new node"
1433 a6a3efe4 Iustin Pop
    " (the secondary for mirrored instances or any node"
1434 a6a3efe4 Iustin Pop
    " for shared storage)."),
1435 d0c8c01d Iustin Pop
  "migrate": (
1436 6ea815cf Iustin Pop
    MigrateInstance, ARGS_ONE_INSTANCE,
1437 aa06f8c6 Michael Hanselmann
    [FORCE_OPT, NONLIVE_OPT, MIGRATION_MODE_OPT, CLEANUP_OPT, DRY_RUN_OPT,
1438 e9c487be René Nussbaumer
     PRIORITY_OPT, DST_NODE_OPT, IALLOCATOR_OPT, ALLOW_FAILOVER_OPT],
1439 6ea815cf Iustin Pop
    "[-f] <instance>", "Migrate instance to its secondary node"
1440 1b7761fd Apollon Oikonomopoulos
    " (only for mirrored instances)"),
1441 d0c8c01d Iustin Pop
  "move": (
1442 6ea815cf Iustin Pop
    MoveInstance, ARGS_ONE_INSTANCE,
1443 db5a8a2d Iustin Pop
    [FORCE_OPT, SUBMIT_OPT, SINGLE_NODE_OPT, SHUTDOWN_TIMEOUT_OPT,
1444 bb851c63 Iustin Pop
     DRY_RUN_OPT, PRIORITY_OPT, IGNORE_CONSIST_OPT],
1445 6ea815cf Iustin Pop
    "[-f] <instance>", "Move instance to an arbitrary node"
1446 6ea815cf Iustin Pop
    " (only for instances of type file and lv)"),
1447 d0c8c01d Iustin Pop
  "info": (
1448 6ea815cf Iustin Pop
    ShowInstanceConfig, ARGS_MANY_INSTANCES,
1449 aa06f8c6 Michael Hanselmann
    [STATIC_OPT, ALL_OPT, ROMAN_OPT, PRIORITY_OPT],
1450 6ea815cf Iustin Pop
    "[-s] {--all | <instance>...}",
1451 6ea815cf Iustin Pop
    "Show information on the specified instance(s)"),
1452 d0c8c01d Iustin Pop
  "list": (
1453 6ea815cf Iustin Pop
    ListInstances, ARGS_MANY_INSTANCES,
1454 87e87959 Michael Hanselmann
    [NOHDR_OPT, SEP_OPT, USEUNITS_OPT, FIELDS_OPT, VERBOSE_OPT,
1455 87e87959 Michael Hanselmann
     FORCE_FILTER_OPT],
1456 6ea815cf Iustin Pop
    "[<instance>...]",
1457 b82c5ff5 Michael Hanselmann
    "Lists the instances and their status. The available fields can be shown"
1458 b82c5ff5 Michael Hanselmann
    " using the \"list-fields\" command (see the man page for details)."
1459 b82c5ff5 Michael Hanselmann
    " The default field list is (in order): %s." %
1460 b82c5ff5 Michael Hanselmann
    utils.CommaJoin(_LIST_DEF_FIELDS),
1461 6ea815cf Iustin Pop
    ),
1462 b82c5ff5 Michael Hanselmann
  "list-fields": (
1463 b82c5ff5 Michael Hanselmann
    ListInstanceFields, [ArgUnknown()],
1464 b82c5ff5 Michael Hanselmann
    [NOHDR_OPT, SEP_OPT],
1465 b82c5ff5 Michael Hanselmann
    "[fields...]",
1466 b82c5ff5 Michael Hanselmann
    "Lists all available fields for instances"),
1467 d0c8c01d Iustin Pop
  "reinstall": (
1468 3e54ace7 Iustin Pop
    ReinstallInstance, [ArgInstance()],
1469 06073e85 Guido Trotter
    [FORCE_OPT, OS_OPT, FORCE_VARIANT_OPT, m_force_multi, m_node_opt,
1470 39dfd93e René Nussbaumer
     m_pri_node_opt, m_sec_node_opt, m_clust_opt, m_inst_opt, m_node_tags_opt,
1471 39dfd93e René Nussbaumer
     m_pri_node_tags_opt, m_sec_node_tags_opt, m_inst_tags_opt, SELECT_OS_OPT,
1472 8d8c4eff Michael Hanselmann
     SUBMIT_OPT, DRY_RUN_OPT, PRIORITY_OPT, OSPARAMS_OPT],
1473 6ea815cf Iustin Pop
    "[-f] <instance>", "Reinstall a stopped instance"),
1474 d0c8c01d Iustin Pop
  "remove": (
1475 6ea815cf Iustin Pop
    RemoveInstance, ARGS_ONE_INSTANCE,
1476 db5a8a2d Iustin Pop
    [FORCE_OPT, SHUTDOWN_TIMEOUT_OPT, IGNORE_FAILURES_OPT, SUBMIT_OPT,
1477 aa06f8c6 Michael Hanselmann
     DRY_RUN_OPT, PRIORITY_OPT],
1478 6ea815cf Iustin Pop
    "[-f] <instance>", "Shuts down the instance and removes it"),
1479 d0c8c01d Iustin Pop
  "rename": (
1480 6ea815cf Iustin Pop
    RenameInstance,
1481 6ea815cf Iustin Pop
    [ArgInstance(min=1, max=1), ArgHost(min=1, max=1)],
1482 aa06f8c6 Michael Hanselmann
    [NOIPCHECK_OPT, NONAMECHECK_OPT, SUBMIT_OPT, DRY_RUN_OPT, PRIORITY_OPT],
1483 6ea815cf Iustin Pop
    "<instance> <new_name>", "Rename the instance"),
1484 d0c8c01d Iustin Pop
  "replace-disks": (
1485 6ea815cf Iustin Pop
    ReplaceDisks, ARGS_ONE_INSTANCE,
1486 7ea7bcf6 Iustin Pop
    [AUTO_REPLACE_OPT, DISKIDX_OPT, IALLOCATOR_OPT, EARLY_RELEASE_OPT,
1487 db5a8a2d Iustin Pop
     NEW_SECONDARY_OPT, ON_PRIMARY_OPT, ON_SECONDARY_OPT, SUBMIT_OPT,
1488 aa06f8c6 Michael Hanselmann
     DRY_RUN_OPT, PRIORITY_OPT],
1489 6ea815cf Iustin Pop
    "[-s|-p|-n NODE|-I NAME] <instance>",
1490 6ea815cf Iustin Pop
    "Replaces all disks for the instance"),
1491 d0c8c01d Iustin Pop
  "modify": (
1492 6ea815cf Iustin Pop
    SetInstanceParams, ARGS_ONE_INSTANCE,
1493 e29e9550 Iustin Pop
    [BACKEND_OPT, DISK_OPT, FORCE_OPT, HVOPTS_OPT, NET_OPT, SUBMIT_OPT,
1494 1052d622 Iustin Pop
     DISK_TEMPLATE_OPT, SINGLE_NODE_OPT, OS_OPT, FORCE_VARIANT_OPT,
1495 456798ab Iustin Pop
     OSPARAMS_OPT, DRY_RUN_OPT, PRIORITY_OPT, NWSYNC_OPT],
1496 6ea815cf Iustin Pop
    "<instance>", "Alters the parameters of an instance"),
1497 d0c8c01d Iustin Pop
  "shutdown": (
1498 1c5945b6 Iustin Pop
    GenericManyOps("shutdown", _ShutdownInstance), [ArgInstance()],
1499 064c21f8 Iustin Pop
    [m_node_opt, m_pri_node_opt, m_sec_node_opt, m_clust_opt,
1500 39dfd93e René Nussbaumer
     m_node_tags_opt, m_pri_node_tags_opt, m_sec_node_tags_opt,
1501 db5a8a2d Iustin Pop
     m_inst_tags_opt, m_inst_opt, m_force_multi, TIMEOUT_OPT, SUBMIT_OPT,
1502 885a0fc4 Iustin Pop
     DRY_RUN_OPT, PRIORITY_OPT, IGNORE_OFFLINE_OPT, NO_REMEMBER_OPT],
1503 6ea815cf Iustin Pop
    "<instance>", "Stops an instance"),
1504 d0c8c01d Iustin Pop
  "startup": (
1505 1c5945b6 Iustin Pop
    GenericManyOps("startup", _StartupInstance), [ArgInstance()],
1506 39dfd93e René Nussbaumer
    [FORCE_OPT, m_force_multi, m_node_opt, m_pri_node_opt, m_sec_node_opt,
1507 39dfd93e René Nussbaumer
     m_node_tags_opt, m_pri_node_tags_opt, m_sec_node_tags_opt,
1508 39dfd93e René Nussbaumer
     m_inst_tags_opt, m_clust_opt, m_inst_opt, SUBMIT_OPT, HVOPTS_OPT,
1509 885a0fc4 Iustin Pop
     BACKEND_OPT, DRY_RUN_OPT, PRIORITY_OPT, IGNORE_OFFLINE_OPT,
1510 323f9095 Stephen Shirley
     NO_REMEMBER_OPT, STARTUP_PAUSED_OPT],
1511 6ea815cf Iustin Pop
    "<instance>", "Starts an instance"),
1512 d0c8c01d Iustin Pop
  "reboot": (
1513 1c5945b6 Iustin Pop
    GenericManyOps("reboot", _RebootInstance), [ArgInstance()],
1514 064c21f8 Iustin Pop
    [m_force_multi, REBOOT_TYPE_OPT, IGNORE_SECONDARIES_OPT, m_node_opt,
1515 17c3f802 Guido Trotter
     m_pri_node_opt, m_sec_node_opt, m_clust_opt, m_inst_opt, SUBMIT_OPT,
1516 39dfd93e René Nussbaumer
     m_node_tags_opt, m_pri_node_tags_opt, m_sec_node_tags_opt,
1517 aa06f8c6 Michael Hanselmann
     m_inst_tags_opt, SHUTDOWN_TIMEOUT_OPT, DRY_RUN_OPT, PRIORITY_OPT],
1518 6ea815cf Iustin Pop
    "<instance>", "Reboots an instance"),
1519 d0c8c01d Iustin Pop
  "activate-disks": (
1520 db5a8a2d Iustin Pop
    ActivateDisks, ARGS_ONE_INSTANCE,
1521 aa06f8c6 Michael Hanselmann
    [SUBMIT_OPT, IGNORE_SIZE_OPT, PRIORITY_OPT],
1522 6ea815cf Iustin Pop
    "<instance>", "Activate an instance's disks"),
1523 d0c8c01d Iustin Pop
  "deactivate-disks": (
1524 aa06f8c6 Michael Hanselmann
    DeactivateDisks, ARGS_ONE_INSTANCE,
1525 c9c41373 Iustin Pop
    [FORCE_OPT, SUBMIT_OPT, DRY_RUN_OPT, PRIORITY_OPT],
1526 c9c41373 Iustin Pop
    "[-f] <instance>", "Deactivate an instance's disks"),
1527 d0c8c01d Iustin Pop
  "recreate-disks": (
1528 aa06f8c6 Michael Hanselmann
    RecreateDisks, ARGS_ONE_INSTANCE,
1529 c8a96ae7 Iustin Pop
    [SUBMIT_OPT, DISKIDX_OPT, NODE_PLACEMENT_OPT, DRY_RUN_OPT, PRIORITY_OPT],
1530 6ea815cf Iustin Pop
    "<instance>", "Recreate an instance's disks"),
1531 d0c8c01d Iustin Pop
  "grow-disk": (
1532 6ea815cf Iustin Pop
    GrowDisk,
1533 6ea815cf Iustin Pop
    [ArgInstance(min=1, max=1), ArgUnknown(min=1, max=1),
1534 6ea815cf Iustin Pop
     ArgUnknown(min=1, max=1)],
1535 aa06f8c6 Michael Hanselmann
    [SUBMIT_OPT, NWSYNC_OPT, DRY_RUN_OPT, PRIORITY_OPT],
1536 6ea815cf Iustin Pop
    "<instance> <disk> <size>", "Grow an instance's disk"),
1537 bd2a5569 Michael Hanselmann
  "change-group": (
1538 bd2a5569 Michael Hanselmann
    ChangeGroup, ARGS_ONE_INSTANCE,
1539 bd2a5569 Michael Hanselmann
    [TO_GROUP_OPT, IALLOCATOR_OPT, EARLY_RELEASE_OPT],
1540 bd2a5569 Michael Hanselmann
    "[-I <iallocator>] [--to <group>]", "Change group of instance"),
1541 d0c8c01d Iustin Pop
  "list-tags": (
1542 aa06f8c6 Michael Hanselmann
    ListTags, ARGS_ONE_INSTANCE, [PRIORITY_OPT],
1543 6ea815cf Iustin Pop
    "<instance_name>", "List the tags of the given instance"),
1544 d0c8c01d Iustin Pop
  "add-tags": (
1545 6ea815cf Iustin Pop
    AddTags, [ArgInstance(min=1, max=1), ArgUnknown()],
1546 aa06f8c6 Michael Hanselmann
    [TAG_SRC_OPT, PRIORITY_OPT],
1547 6ea815cf Iustin Pop
    "<instance_name> tag...", "Add tags to the given instance"),
1548 d0c8c01d Iustin Pop
  "remove-tags": (
1549 6ea815cf Iustin Pop
    RemoveTags, [ArgInstance(min=1, max=1), ArgUnknown()],
1550 aa06f8c6 Michael Hanselmann
    [TAG_SRC_OPT, PRIORITY_OPT],
1551 6ea815cf Iustin Pop
    "<instance_name> tag...", "Remove tags from given instance"),
1552 a8083063 Iustin Pop
  }
1553 a8083063 Iustin Pop
1554 7232c04c Iustin Pop
#: dictionary with aliases for commands
1555 dbfd89dd Guido Trotter
aliases = {
1556 d0c8c01d Iustin Pop
  "start": "startup",
1557 d0c8c01d Iustin Pop
  "stop": "shutdown",
1558 dbfd89dd Guido Trotter
  }
1559 dbfd89dd Guido Trotter
1560 a8005e17 Michael Hanselmann
1561 e792102d Michael Hanselmann
def Main():
1562 e792102d Michael Hanselmann
  return GenericMain(commands, aliases=aliases,
1563 e792102d Michael Hanselmann
                     override={"tag_type": constants.TAG_INSTANCE})