gnt-instance 53.3 KB
Newer Older
Iustin Pop's avatar
Iustin Pop committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#!/usr/bin/python
#

# Copyright (C) 2006, 2007 Google Inc.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
# 02110-1301, USA.


22
23
24
25
# pylint: disable-msg=W0401,W0614
# W0401: Wildcard import ganeti.cli
# W0614: Unused import %s from wildcard import (since we need cli)

Iustin Pop's avatar
Iustin Pop committed
26
27
import sys
import os
28
import itertools
29
import simplejson
Iustin Pop's avatar
Iustin Pop committed
30
31
32
33
from optparse import make_option
from cStringIO import StringIO

from ganeti.cli import *
34
from ganeti import cli
Iustin Pop's avatar
Iustin Pop committed
35
36
37
from ganeti import opcodes
from ganeti import constants
from ganeti import utils
38
39
40
41
42
43
44
45
46
from ganeti import errors


_SHUTDOWN_CLUSTER = "cluster"
_SHUTDOWN_NODES_BOTH = "nodes"
_SHUTDOWN_NODES_PRI = "nodes-pri"
_SHUTDOWN_NODES_SEC = "nodes-sec"
_SHUTDOWN_INSTANCES = "instances"

47

48
49
_VALUE_TRUE = "true"

50
#: default list of options for L{ListInstances}
51
_LIST_DEF_FIELDS = [
52
  "name", "hypervisor", "os", "pnode", "status", "oper_ram",
53
54
  ]

55

56
def _ExpandMultiNames(mode, names, client=None):
57
58
59
60
  """Expand the given names using the passed mode.

  For _SHUTDOWN_CLUSTER, all instances will be returned. For
  _SHUTDOWN_NODES_PRI/SEC, all instances having those nodes as
61
  primary/secondary will be returned. For _SHUTDOWN_NODES_BOTH, all
62
63
64
65
  instances having those nodes as either primary or secondary will be
  returned. For _SHUTDOWN_INSTANCES, the given instances will be
  returned.

66
67
68
69
70
71
72
73
74
75
76
77
  @param mode: one of L{_SHUTDOWN_CLUSTER}, L{_SHUTDOWN_NODES_BOTH},
      L{_SHUTDOWN_NODES_PRI}, L{_SHUTDOWN_NODES_SEC} or
      L{_SHUTDOWN_INSTANCES}
  @param names: a list of names; for cluster, it must be empty,
      and for node and instance it must be a list of valid item
      names (short names are valid as usual, e.g. node1 instead of
      node1.example.com)
  @rtype: list
  @return: the list of names after the expansion
  @raise errors.ProgrammerError: for unknown selection type
  @raise errors.OpPrereqError: for invalid input parameters

78
  """
79
80
  if client is None:
    client = GetClient()
81
82
83
  if mode == _SHUTDOWN_CLUSTER:
    if names:
      raise errors.OpPrereqError("Cluster filter mode takes no arguments")
84
    idata = client.QueryInstances([], ["name"], False)
85
86
87
88
89
90
91
    inames = [row[0] for row in idata]

  elif mode in (_SHUTDOWN_NODES_BOTH,
                _SHUTDOWN_NODES_PRI,
                _SHUTDOWN_NODES_SEC):
    if not names:
      raise errors.OpPrereqError("No node names passed")
92
93
    ndata = client.QueryNodes(names, ["name", "pinst_list", "sinst_list"],
                              True)
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
    ipri = [row[1] for row in ndata]
    pri_names = list(itertools.chain(*ipri))
    isec = [row[2] for row in ndata]
    sec_names = list(itertools.chain(*isec))
    if mode == _SHUTDOWN_NODES_BOTH:
      inames = pri_names + sec_names
    elif mode == _SHUTDOWN_NODES_PRI:
      inames = pri_names
    elif mode == _SHUTDOWN_NODES_SEC:
      inames = sec_names
    else:
      raise errors.ProgrammerError("Unhandled shutdown type")

  elif mode == _SHUTDOWN_INSTANCES:
    if not names:
      raise errors.OpPrereqError("No instance names passed")
110
    idata = client.QueryInstances(names, ["name"], False)
111
112
113
114
115
116
    inames = [row[0] for row in idata]

  else:
    raise errors.OpPrereqError("Unknown mode '%s'" % mode)

  return inames
Iustin Pop's avatar
Iustin Pop committed
117
118


119
120
121
122
123
124
def _ConfirmOperation(inames, text):
  """Ask the user to confirm an operation on a list of instances.

  This function is used to request confirmation for doing an operation
  on a given list of instances.

125
126
127
128
129
130
131
132
  @type inames: list
  @param inames: the list of names that we display when
      we ask for confirmation
  @type text: str
  @param text: the operation that the user should confirm
      (e.g. I{shutdown} or I{startup})
  @rtype: boolean
  @return: True or False depending on user's confirmation.
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152

  """
  count = len(inames)
  msg = ("The %s will operate on %d instances.\n"
         "Do you want to continue?" % (text, count))
  affected = ("\nAffected instances:\n" +
              "\n".join(["  %s" % name for name in inames]))

  choices = [('y', True, 'Yes, execute the %s' % text),
             ('n', False, 'No, abort the %s' % text)]

  if count > 20:
    choices.insert(1, ('v', 'v', 'View the list of affected instances'))
    ask = msg
  else:
    ask = msg + affected

  choice = AskUser(ask, choices)
  if choice == 'v':
    choices.pop(1)
153
    choice = AskUser(msg + affected, choices)
154
155
156
  return choice


157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
def _TransformPath(user_input):
  """Transform a user path into a canonical value.

  This function transforms the a path passed as textual information
  into the constants that the LU code expects.

  """
  if user_input:
    if user_input.lower() == "default":
      result_path = constants.VALUE_DEFAULT
    elif user_input.lower() == "none":
      result_path = constants.VALUE_NONE
    else:
      if not os.path.isabs(user_input):
        raise errors.OpPrereqError("Path '%s' is not an absolute filename" %
                                   user_input)
      result_path = user_input
  else:
    result_path = constants.VALUE_DEFAULT

  return result_path


180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
def _EnsureInstancesExist(client, names):
  """Check for and ensure the given instance names exist.

  This function will raise an OpPrereqError in case they don't
  exist. Otherwise it will exit cleanly.

  @type client: L{luxi.Client}
  @param client: the client to use for the query
  @type names: list
  @param names: the list of instance names to query
  @raise errors.OpPrereqError: in case any instance is missing

  """
  # TODO: change LUQueryInstances to that it actually returns None
  # instead of raising an exception, or devise a better mechanism
195
  result = client.QueryInstances(names, ["name"], False)
196
197
198
199
200
  for orig_name, row in zip(names, result):
    if row[0] is None:
      raise errors.OpPrereqError("Instance '%s' does not exist" % orig_name)


Iustin Pop's avatar
Iustin Pop committed
201
def ListInstances(opts, args):
Oleksiy Mishchenko's avatar
Oleksiy Mishchenko committed
202
  """List instances and their properties.
Iustin Pop's avatar
Iustin Pop committed
203

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

Iustin Pop's avatar
Iustin Pop committed
210
211
  """
  if opts.output is None:
212
213
214
    selected_fields = _LIST_DEF_FIELDS
  elif opts.output.startswith("+"):
    selected_fields = _LIST_DEF_FIELDS + opts.output[1:].split(",")
Iustin Pop's avatar
Iustin Pop committed
215
216
217
  else:
    selected_fields = opts.output.split(",")

218
  output = GetClient().QueryInstances(args, selected_fields, opts.do_locking)
Iustin Pop's avatar
Iustin Pop committed
219
220

  if not opts.no_headers:
221
222
223
    headers = {
      "name": "Instance", "os": "OS", "pnode": "Primary_node",
      "snodes": "Secondary_Nodes", "admin_state": "Autostart",
Iustin Pop's avatar
Iustin Pop committed
224
      "oper_state": "Running",
225
      "oper_ram": "Memory", "disk_template": "Disk_template",
226
      "ip": "IP_address", "mac": "MAC_address",
Iustin Pop's avatar
Iustin Pop committed
227
      "bridge": "Bridge",
228
      "sda_size": "Disk/0", "sdb_size": "Disk/1",
229
      "disk_usage": "DiskUsage",
230
      "status": "Status", "tags": "Tags",
231
      "network_port": "Network_port",
232
233
234
235
236
237
238
239
      "hv/kernel_path": "Kernel_path",
      "hv/initrd_path": "Initrd_path",
      "hv/boot_order": "HVM_boot_order",
      "hv/acpi": "HVM_ACPI",
      "hv/pae": "HVM_PAE",
      "hv/cdrom_image_path": "HVM_CDROM_image_path",
      "hv/nic_type": "HVM_NIC_type",
      "hv/disk_type": "HVM_Disk_type",
240
      "hv/vnc_bind_address": "VNC_bind_address",
241
      "serial_no": "SerialNo", "hypervisor": "Hypervisor",
242
      "hvparams": "Hypervisor_parameters",
Iustin Pop's avatar
Iustin Pop committed
243
244
      "be/memory": "Configured_memory",
      "be/vcpus": "VCPUs",
245
      "be/auto_balance": "Auto_balance",
Iustin Pop's avatar
Iustin Pop committed
246
247
248
      "disk.count": "Disks", "disk.sizes": "Disk_sizes",
      "nic.count": "NICs", "nic.ips": "NIC_IPs",
      "nic.bridges": "NIC_bridges", "nic.macs": "NIC_MACs",
249
      }
250
251
252
  else:
    headers = None

253
  unitfields = ["be/memory", "oper_ram", "sd(a|b)_size", "disk\.size/.*"]
254
  numfields = ["be/memory", "oper_ram", "sd(a|b)_size", "be/vcpus",
Iustin Pop's avatar
Iustin Pop committed
255
               "serial_no", "(disk|nic)\.count", "disk\.size/.*"]
256

Iustin Pop's avatar
Iustin Pop committed
257
258
  list_type_fields = ("tags", "disk.sizes",
                      "nic.macs", "nic.ips", "nic.bridges")
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
  # change raw values to nicer strings
  for row in output:
    for idx, field in enumerate(selected_fields):
      val = row[idx]
      if field == "snodes":
        val = ",".join(val) or "-"
      elif field == "admin_state":
        if val:
          val = "yes"
        else:
          val = "no"
      elif field == "oper_state":
        if val is None:
          val = "(node down)"
        elif val: # True
          val = "running"
        else:
          val = "stopped"
      elif field == "oper_ram":
        if val is None:
          val = "(node down)"
      elif field == "sda_size" or field == "sdb_size":
        if val is None:
          val = "N/A"
283
      elif field in list_type_fields:
Iustin Pop's avatar
Iustin Pop committed
284
        val = ",".join(str(item) for item in val)
285
286
      elif val is None:
        val = "-"
287
288
      row[idx] = str(val)

289
290
  data = GenerateTable(separator=opts.separator, headers=headers,
                       fields=selected_fields, unitfields=unitfields,
291
                       numfields=numfields, data=output, units=opts.units)
292
293

  for line in data:
294
    ToStdout(line)
Iustin Pop's avatar
Iustin Pop committed
295
296
297
298
299
300
301

  return 0


def AddInstance(opts, args):
  """Add an instance to the cluster.

302
303
304
305
306
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should contain only one element, the new instance name
  @rtype: int
  @return: the desired exit code
Iustin Pop's avatar
Iustin Pop committed
307
308
309
310

  """
  instance = args[0]

311
312
  (pnode, snode) = SplitNodeOption(opts.node)

313
314
315
316
  hypervisor = None
  hvparams = {}
  if opts.hypervisor:
    hypervisor, hvparams = opts.hypervisor
317

318
319
320
321
322
323
324
325
326
  if opts.nics:
    try:
      nic_max = max(int(nidx[0])+1 for nidx in opts.nics)
    except ValueError, err:
      raise errors.OpPrereqError("Invalid NIC index passed: %s" % str(err))
    nics = [{}] * nic_max
    for nidx, ndict in opts.nics.items():
      nidx = int(nidx)
      nics[nidx] = ndict
Iustin Pop's avatar
Iustin Pop committed
327
328
329
  elif opts.no_nics:
    # no nics
    nics = []
330
331
332
333
  else:
    # default of one nic, all auto
    nics = [{}]

Iustin Pop's avatar
Iustin Pop committed
334
335
336
337
338
  if opts.disk_template == constants.DT_DISKLESS:
    if opts.disks:
      raise errors.OpPrereqError("Diskless instance but disk"
                                 " information passed")
    disks = []
339
  else:
Iustin Pop's avatar
Iustin Pop committed
340
341
    if not opts.disks:
      raise errors.OpPrereqError("No disk information specified")
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
    try:
      disk_max = max(int(didx[0])+1 for didx in opts.disks)
    except ValueError, err:
      raise errors.OpPrereqError("Invalid disk index passed: %s" % str(err))
    disks = [{}] * disk_max
    for didx, ddict in opts.disks:
      didx = int(didx)
      if "size" not in ddict:
        raise errors.OpPrereqError("Missing size for disk %d" % didx)
      try:
        ddict["size"] = utils.ParseUnit(ddict["size"])
      except ValueError, err:
        raise errors.OpPrereqError("Invalid disk size for disk %d: %s" %
                                   (didx, err))
      disks[didx] = ddict

358
  utils.ForceDictType(opts.beparams, constants.BES_PARAMETER_TYPES)
359
  utils.ForceDictType(hvparams, constants.HVS_PARAMETER_TYPES)
360

Iustin Pop's avatar
Iustin Pop committed
361
  op = opcodes.OpCreateInstance(instance_name=instance,
362
                                disks=disks,
Iustin Pop's avatar
Iustin Pop committed
363
                                disk_template=opts.disk_template,
364
                                nics=nics,
Iustin Pop's avatar
Iustin Pop committed
365
                                mode=constants.INSTANCE_CREATE,
366
                                os_type=opts.os, pnode=pnode,
Iustin Pop's avatar
Iustin Pop committed
367
                                snode=snode,
368
                                start=opts.start, ip_check=opts.ip_check,
369
                                wait_for_sync=opts.wait_for_sync,
370
371
                                hypervisor=hypervisor,
                                hvparams=hvparams,
Iustin Pop's avatar
Iustin Pop committed
372
                                beparams=opts.beparams,
373
                                iallocator=opts.iallocator,
374
375
                                file_storage_dir=opts.file_storage_dir,
                                file_driver=opts.file_driver,
376
                                )
377

378
  SubmitOrSend(op, opts)
Iustin Pop's avatar
Iustin Pop committed
379
380
381
  return 0


382
def BatchCreate(opts, args):
383
384
385
386
387
388
  """Create instances using a definition file.

  This function reads a json file with instances defined
  in the form::

    {"instance-name":{
389
      "disk_size": [20480],
390
391
392
393
      "template": "drbd",
      "backend": {
        "memory": 512,
        "vcpus": 1 },
394
      "os": "debootstrap",
395
396
397
398
399
400
401
402
403
404
405
406
407
      "primary_node": "firstnode",
      "secondary_node": "secondnode",
      "iallocator": "dumb"}
    }

  Note that I{primary_node} and I{secondary_node} have precedence over
  I{iallocator}.

  @param opts: the command line options selected by the user
  @type args: list
  @param args: should contain one element, the json filename
  @rtype: int
  @return: the desired exit code
408
409

  """
410
  _DEFAULT_SPECS = {"disk_size": [20 * 1024],
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
                    "backend": {},
                    "iallocator": None,
                    "primary_node": None,
                    "secondary_node": None,
                    "ip": 'none',
                    "mac": 'auto',
                    "bridge": None,
                    "start": True,
                    "ip_check": True,
                    "hypervisor": None,
                    "file_storage_dir": None,
                    "file_driver": 'loop'}

  def _PopulateWithDefaults(spec):
    """Returns a new hash combined with default values."""
426
427
428
    mydict = _DEFAULT_SPECS.copy()
    mydict.update(spec)
    return mydict
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460

  def _Validate(spec):
    """Validate the instance specs."""
    # Validate fields required under any circumstances
    for required_field in ('os', 'template'):
      if required_field not in spec:
        raise errors.OpPrereqError('Required field "%s" is missing.' %
                                   required_field)
    # Validate special fields
    if spec['primary_node'] is not None:
      if (spec['template'] in constants.DTS_NET_MIRROR and
          spec['secondary_node'] is None):
        raise errors.OpPrereqError('Template requires secondary node, but'
                                   ' there was no secondary provided.')
    elif spec['iallocator'] is None:
      raise errors.OpPrereqError('You have to provide at least a primary_node'
                                 ' or an iallocator.')

    if (spec['hypervisor'] and
        not isinstance(spec['hypervisor'], dict)):
      raise errors.OpPrereqError('Hypervisor parameters must be a dict.')

  json_filename = args[0]
  fd = open(json_filename, 'r')
  try:
    instance_data = simplejson.load(fd)
  finally:
    fd.close()

  # Iterate over the instances and do:
  #  * Populate the specs with default value
  #  * Validate the instance specs
461
462
463
  i_names = utils.NiceSort(instance_data.keys())
  for name in i_names:
    specs = instance_data[name]
464
465
466
467
468
469
470
471
    specs = _PopulateWithDefaults(specs)
    _Validate(specs)

    hypervisor = None
    hvparams = {}
    if specs['hypervisor']:
      hypervisor, hvparams = specs['hypervisor'].iteritems()

472
473
474
475
476
477
478
479
480
481
482
483
    disks = []
    for elem in specs['disk_size']:
      try:
        size = utils.ParseUnit(elem)
      except ValueError, err:
        raise errors.OpPrereqError("Invalid disk size '%s' for"
                                   " instance %s: %s" %
                                   (elem, name, err))
      disks.append({"size": size})

    nic0 = {'ip': specs['ip'], 'bridge': specs['bridge'], 'mac': specs['mac']}

484
485
486
    utils.ForceDictType(specs['backend'], constants.BES_PARAMETER_TYPES)
    utils.ForceDictType(hvparams, constants.HVS_PARAMETER_TYPES)

487
    op = opcodes.OpCreateInstance(instance_name=name,
488
                                  disks=disks,
489
490
491
492
493
                                  disk_template=specs['template'],
                                  mode=constants.INSTANCE_CREATE,
                                  os_type=specs['os'],
                                  pnode=specs['primary_node'],
                                  snode=specs['secondary_node'],
494
                                  nics=[nic0],
495
496
497
498
499
500
501
502
503
504
                                  start=specs['start'],
                                  ip_check=specs['ip_check'],
                                  wait_for_sync=True,
                                  iallocator=specs['iallocator'],
                                  hypervisor=hypervisor,
                                  hvparams=hvparams,
                                  beparams=specs['backend'],
                                  file_storage_dir=specs['file_storage_dir'],
                                  file_driver=specs['file_driver'])

505
    ToStdout("%s: %s", name, cli.SendJob([op]))
506
507
508
509

  return 0


510
511
512
def ReinstallInstance(opts, args):
  """Reinstall an instance.

513
514
515
516
517
518
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should contain only one element, the name of the
      instance to be reinstalled
  @rtype: int
  @return: the desired exit code
519
520
521
522

  """
  instance_name = args[0]

523
524
525
526
527
  if opts.select_os is True:
    op = opcodes.OpDiagnoseOS(output_fields=["name", "valid"], names=[])
    result = SubmitOpCode(op)

    if not result:
528
      ToStdout("Can't get the OS list")
529
530
      return 1

531
    ToStdout("Available OS templates:")
532
533
534
    number = 0
    choices = []
    for entry in result:
535
      ToStdout("%3s: %s", number, entry[0])
536
537
538
539
540
541
542
543
      choices.append(("%s" % number, entry[0], entry[0]))
      number = number + 1

    choices.append(('x', 'exit', 'Exit gnt-instance reinstall'))
    selected = AskUser("Enter OS template name or number (or x to abort):",
                       choices)

    if selected == 'exit':
544
      ToStdout("User aborted reinstall, exiting")
545
546
      return 1

547
    os_name = selected
548
  else:
549
    os_name = opts.os
550

551
  if not opts.force:
552
553
    usertext = ("This will reinstall the instance %s and remove"
                " all data. Continue?") % instance_name
554
    if not AskUser(usertext):
555
556
      return 1

557
  op = opcodes.OpReinstallInstance(instance_name=instance_name,
558
                                   os_type=os_name)
559
  SubmitOrSend(op, opts)
560
561
562
563

  return 0


Iustin Pop's avatar
Iustin Pop committed
564
565
566
def RemoveInstance(opts, args):
  """Remove an instance.

567
568
569
570
571
572
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should contain only one element, the name of
      the instance to be removed
  @rtype: int
  @return: the desired exit code
Iustin Pop's avatar
Iustin Pop committed
573
574
575
576

  """
  instance_name = args[0]
  force = opts.force
577
  cl = GetClient()
Iustin Pop's avatar
Iustin Pop committed
578
579

  if not force:
580
581
    _EnsureInstancesExist(cl, [instance_name])

Iustin Pop's avatar
Iustin Pop committed
582
583
584
    usertext = ("This will remove the volumes of the instance %s"
                " (including mirrors), thus removing all the data"
                " of the instance. Continue?") % instance_name
585
    if not AskUser(usertext):
Iustin Pop's avatar
Iustin Pop committed
586
587
      return 1

Iustin Pop's avatar
Iustin Pop committed
588
589
  op = opcodes.OpRemoveInstance(instance_name=instance_name,
                                ignore_failures=opts.ignore_failures)
590
  SubmitOrSend(op, opts, cl=cl)
Iustin Pop's avatar
Iustin Pop committed
591
592
593
  return 0


594
def RenameInstance(opts, args):
Guido Trotter's avatar
Guido Trotter committed
595
  """Rename an instance.
596

597
598
599
600
601
602
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should contain two elements, the old and the
      new instance names
  @rtype: int
  @return: the desired exit code
603
604
605
606
607

  """
  op = opcodes.OpRenameInstance(instance_name=args[0],
                                new_name=args[1],
                                ignore_ip=opts.ignore_ip)
608
  SubmitOrSend(op, opts)
609
610
611
  return 0


Iustin Pop's avatar
Iustin Pop committed
612
613
614
615
def ActivateDisks(opts, args):
  """Activate an instance's disks.

  This serves two purposes:
616
617
    - it allows (as long as the instance is not running)
      mounting the disks and modifying them from the node
Iustin Pop's avatar
Iustin Pop committed
618
619
    - it repairs inactive secondary drbds

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

Iustin Pop's avatar
Iustin Pop committed
626
627
628
  """
  instance_name = args[0]
  op = opcodes.OpActivateInstanceDisks(instance_name=instance_name)
629
  disks_info = SubmitOrSend(op, opts)
Iustin Pop's avatar
Iustin Pop committed
630
  for host, iname, nname in disks_info:
631
    ToStdout("%s:%s:%s", host, iname, nname)
Iustin Pop's avatar
Iustin Pop committed
632
633
634
635
  return 0


def DeactivateDisks(opts, args):
636
  """Deactivate an instance's disks..
Iustin Pop's avatar
Iustin Pop committed
637
638
639
640

  This function takes the instance name, looks for its primary node
  and the tries to shutdown its block devices on that node.

641
642
643
644
645
646
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should contain only one element, the instance name
  @rtype: int
  @return: the desired exit code

Iustin Pop's avatar
Iustin Pop committed
647
648
649
  """
  instance_name = args[0]
  op = opcodes.OpDeactivateInstanceDisks(instance_name=instance_name)
650
  SubmitOrSend(op, opts)
Iustin Pop's avatar
Iustin Pop committed
651
652
653
  return 0


Iustin Pop's avatar
Iustin Pop committed
654
def GrowDisk(opts, args):
655
  """Grow an instance's disks.
Iustin Pop's avatar
Iustin Pop committed
656

657
658
659
660
661
662
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should contain two elements, the instance name
      whose disks we grow and the disk name, e.g. I{sda}
  @rtype: int
  @return: the desired exit code
Iustin Pop's avatar
Iustin Pop committed
663
664
665
666

  """
  instance = args[0]
  disk = args[1]
667
668
669
670
  try:
    disk = int(disk)
  except ValueError, err:
    raise errors.OpPrereqError("Invalid disk index: %s" % str(err))
Iustin Pop's avatar
Iustin Pop committed
671
  amount = utils.ParseUnit(args[2])
672
673
  op = opcodes.OpGrowDisk(instance_name=instance, disk=disk, amount=amount,
                          wait_for_sync=opts.wait_for_sync)
674
  SubmitOrSend(op, opts)
Iustin Pop's avatar
Iustin Pop committed
675
676
677
  return 0


Iustin Pop's avatar
Iustin Pop committed
678
def StartupInstance(opts, args):
679
  """Startup instances.
Iustin Pop's avatar
Iustin Pop committed
680

681
682
683
684
685
686
687
688
689
690
  Depending on the options given, this will start one or more
  instances.

  @param opts: the command line options selected by the user
  @type args: list
  @param args: the instance or node names based on which we
      create the final selection (in conjunction with the
      opts argument)
  @rtype: int
  @return: the desired exit code
Iustin Pop's avatar
Iustin Pop committed
691
692

  """
693
  cl = GetClient()
694
695
  if opts.multi_mode is None:
    opts.multi_mode = _SHUTDOWN_INSTANCES
696
  inames = _ExpandMultiNames(opts.multi_mode, args, client=cl)
697
698
  if not inames:
    raise errors.OpPrereqError("Selection filter does not match any instances")
699
700
701
702
  multi_on = opts.multi_mode != _SHUTDOWN_INSTANCES or len(inames) > 1
  if not (opts.force_multi or not multi_on
          or _ConfirmOperation(inames, "startup")):
    return 1
703
  jex = cli.JobExecutor(verbose=multi_on, cl=cl)
704
705
706
707
  for name in inames:
    op = opcodes.OpStartupInstance(instance_name=name,
                                   force=opts.force,
                                   extra_args=opts.extra_args)
708
709
    jex.QueueJob(name, op)
  jex.WaitOrShow(not opts.submit_only)
Iustin Pop's avatar
Iustin Pop committed
710
711
  return 0

712

713
def RebootInstance(opts, args):
714
715
716
717
  """Reboot instance(s).

  Depending on the parameters given, this will reboot one or more
  instances.
718

719
720
721
722
723
724
725
  @param opts: the command line options selected by the user
  @type args: list
  @param args: the instance or node names based on which we
      create the final selection (in conjunction with the
      opts argument)
  @rtype: int
  @return: the desired exit code
726
727

  """
728
  cl = GetClient()
729
730
  if opts.multi_mode is None:
    opts.multi_mode = _SHUTDOWN_INSTANCES
731
  inames = _ExpandMultiNames(opts.multi_mode, args, client=cl)
732
733
  if not inames:
    raise errors.OpPrereqError("Selection filter does not match any instances")
734
735
736
737
  multi_on = opts.multi_mode != _SHUTDOWN_INSTANCES or len(inames) > 1
  if not (opts.force_multi or not multi_on
          or _ConfirmOperation(inames, "reboot")):
    return 1
738
  jex = JobExecutor(verbose=multi_on, cl=cl)
739
740
741
742
  for name in inames:
    op = opcodes.OpRebootInstance(instance_name=name,
                                  reboot_type=opts.reboot_type,
                                  ignore_secondaries=opts.ignore_secondaries)
743
744
    jex.QueueJob(name, op)
  jex.WaitOrShow(not opts.submit_only)
745
  return 0
Iustin Pop's avatar
Iustin Pop committed
746

747

Iustin Pop's avatar
Iustin Pop committed
748
749
750
def ShutdownInstance(opts, args):
  """Shutdown an instance.

751
752
753
754
755
756
757
  @param opts: the command line options selected by the user
  @type args: list
  @param args: the instance or node names based on which we
      create the final selection (in conjunction with the
      opts argument)
  @rtype: int
  @return: the desired exit code
Iustin Pop's avatar
Iustin Pop committed
758
759

  """
760
  cl = GetClient()
761
762
  if opts.multi_mode is None:
    opts.multi_mode = _SHUTDOWN_INSTANCES
763
  inames = _ExpandMultiNames(opts.multi_mode, args, client=cl)
764
765
  if not inames:
    raise errors.OpPrereqError("Selection filter does not match any instances")
766
767
768
769
  multi_on = opts.multi_mode != _SHUTDOWN_INSTANCES or len(inames) > 1
  if not (opts.force_multi or not multi_on
          or _ConfirmOperation(inames, "shutdown")):
    return 1
770
771

  jex = cli.JobExecutor(verbose=multi_on, cl=cl)
772
773
  for name in inames:
    op = opcodes.OpShutdownInstance(instance_name=name)
774
775
    jex.QueueJob(name, op)
  jex.WaitOrShow(not opts.submit_only)
Iustin Pop's avatar
Iustin Pop committed
776
777
778
779
780
781
  return 0


def ReplaceDisks(opts, args):
  """Replace the disks of an instance

782
783
784
785
786
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should contain only one element, the instance name
  @rtype: int
  @return: the desired exit code
Iustin Pop's avatar
Iustin Pop committed
787
788
789

  """
  instance_name = args[0]
790
  new_2ndary = opts.new_secondary
791
  iallocator = opts.iallocator
792
  if opts.disks is None:
793
    disks = []
794
  else:
795
796
797
798
    try:
      disks = [int(i) for i in opts.disks.split(",")]
    except ValueError, err:
      raise errors.OpPrereqError("Invalid disk index passed: %s" % str(err))
799
800
801
802
803
804
  cnt = [opts.on_primary, opts.on_secondary,
         new_2ndary is not None, iallocator is not None].count(True)
  if cnt != 1:
    raise errors.OpPrereqError("One and only one of the -p, -s, -n and -i"
                               " options must be passed")
  elif opts.on_primary:
805
    mode = constants.REPLACE_DISK_PRI
806
  elif opts.on_secondary:
807
    mode = constants.REPLACE_DISK_SEC
808
809
810
  elif new_2ndary is not None or iallocator is not None:
    # replace secondary
    mode = constants.REPLACE_DISK_CHG
811
812

  op = opcodes.OpReplaceDisks(instance_name=args[0], disks=disks,
813
814
                              remote_node=new_2ndary, mode=mode,
                              iallocator=iallocator)
815
  SubmitOrSend(op, opts)
Iustin Pop's avatar
Iustin Pop committed
816
817
818
819
820
821
822
823
824
  return 0


def FailoverInstance(opts, args):
  """Failover an instance.

  The failover is done by shutting it down on its present node and
  starting it on the secondary.

825
826
827
828
829
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should contain only one element, the instance name
  @rtype: int
  @return: the desired exit code
Iustin Pop's avatar
Iustin Pop committed
830
831

  """
832
  cl = GetClient()
833
834
  instance_name = args[0]
  force = opts.force
Iustin Pop's avatar
Iustin Pop committed
835

836
  if not force:
837
838
    _EnsureInstancesExist(cl, [instance_name])

839
840
841
842
843
    usertext = ("Failover will happen to image %s."
                " This requires a shutdown of the instance. Continue?" %
                (instance_name,))
    if not AskUser(usertext):
      return 1
Iustin Pop's avatar
Iustin Pop committed
844

845
846
  op = opcodes.OpFailoverInstance(instance_name=instance_name,
                                  ignore_consistency=opts.ignore_consistency)
847
  SubmitOrSend(op, opts, cl=cl)
848
  return 0
Iustin Pop's avatar
Iustin Pop committed
849
850


851
852
853
854
855
def MigrateInstance(opts, args):
  """Migrate an instance.

  The migrate is done without shutdown.

Iustin Pop's avatar
Iustin Pop committed
856
857
858
859
860
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should contain only one element, the instance name
  @rtype: int
  @return: the desired exit code
861
862

  """
863
  cl = GetClient()
864
865
866
867
  instance_name = args[0]
  force = opts.force

  if not force:
868
869
    _EnsureInstancesExist(cl, [instance_name])

870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
    if opts.cleanup:
      usertext = ("Instance %s will be recovered from a failed migration."
                  " Note that the migration procedure (including cleanup)" %
                  (instance_name,))
    else:
      usertext = ("Instance %s will be migrated. Note that migration" %
                  (instance_name,))
    usertext += (" is **experimental** in this version."
                " This might impact the instance if anything goes wrong."
                " Continue?")
    if not AskUser(usertext):
      return 1

  op = opcodes.OpMigrateInstance(instance_name=instance_name, live=opts.live,
                                 cleanup=opts.cleanup)
885
  SubmitOpCode(op, cl=cl)
886
887
888
  return 0


Iustin Pop's avatar
Iustin Pop committed
889
890
891
def ConnectToInstanceConsole(opts, args):
  """Connect to the console of an instance.

892
893
894
895
896
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should contain only one element, the instance name
  @rtype: int
  @return: the desired exit code
Iustin Pop's avatar
Iustin Pop committed
897
898
899
900
901

  """
  instance_name = args[0]

  op = opcodes.OpConnectConsole(instance_name=instance_name)
902
  cmd = SubmitOpCode(op)
903
904

  if opts.show_command:
905
    ToStdout("%s", utils.ShellQuoteArgs(cmd))
906
907
908
909
  else:
    try:
      os.execvp(cmd[0], cmd)
    finally:
910
      ToStderr("Can't run console command %s with arguments:\n'%s'",
911
               cmd[0], " ".join(cmd))
912
      os._exit(1)
Iustin Pop's avatar
Iustin Pop committed
913
914


915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
def _FormatLogicalID(dev_type, logical_id):
  """Formats the logical_id of a disk.

  """
  if dev_type == constants.LD_DRBD8:
    node_a, node_b, port, minor_a, minor_b, key = logical_id
    data = [
      ("nodeA", "%s, minor=%s" % (node_a, minor_a)),
      ("nodeB", "%s, minor=%s" % (node_b, minor_b)),
      ("port", port),
      ("auth key", key),
      ]
  elif dev_type == constants.LD_LV:
    vg_name, lv_name = logical_id
    data = ["%s/%s" % (vg_name, lv_name)]
  else:
    data = [str(logical_id)]

  return data


def _FormatBlockDevInfo(idx, top_level, dev, static):
Iustin Pop's avatar
Iustin Pop committed
937
938
  """Show block device information.

939
  This is only used by L{ShowInstanceConfig}, but it's too big to be
Iustin Pop's avatar
Iustin Pop committed
940
941
  left for an inline definition.

942
943
944
945
  @type idx: int
  @param idx: the index of the current disk
  @type top_level: boolean
  @param top_level: if this a top-level disk?
946
947
948
949
950
  @type dev: dict
  @param dev: dictionary with disk information
  @type static: boolean
  @param static: wheter the device information doesn't contain
      runtime information but only static data
951
952
  @return: a list of either strings, tuples or lists
      (which should be formatted at a higher indent level)
953

Iustin Pop's avatar
Iustin Pop committed
954
  """
955
  def helper(dtype, status):
956
957
958
959
960
961
    """Format one line for physical device status.

    @type dtype: str
    @param dtype: a constant from the L{constants.LDS_BLOCK} set
    @type status: tuple
    @param status: a tuple as returned from L{backend.FindBlockDevice}
962
    @return: the string representing the status
963
964

    """
Iustin Pop's avatar
Iustin Pop committed
965
    if not status:
966
967
968
969
970
      return "not active"
    txt = ""
    (path, major, minor, syncp, estt, degr, ldisk) = status
    if major is None:
      major_string = "N/A"
Iustin Pop's avatar
Iustin Pop committed
971
    else:
972
      major_string = str(major)
973

974
975
976
977
978
979
980
981
982
983
984
    if minor is None:
      minor_string = "N/A"
    else:
      minor_string = str(minor)

    txt += ("%s (%s:%s)" % (path, major_string, minor_string))
    if dtype in (constants.LD_DRBD8, ):
      if syncp is not None:
        sync_text = "*RECOVERING* %5.2f%%," % syncp
        if estt:
          sync_text += " ETA %ds" % estt
985
        else:
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
1001
1002
1003
1004
1005
1006
1007
1008
1009
1010
1011
          sync_text += " ETA unknown"
      else:
        sync_text = "in sync"
      if degr:
        degr_text = "*DEGRADED*"
      else:
        degr_text = "ok"
      if ldisk:
        ldisk_text = " *MISSING DISK*"
      else:
        ldisk_text = ""
      txt += (" %s, status %s%s" % (sync_text, degr_text, ldisk_text))
    elif dtype == constants.LD_LV:
      if ldisk:
        ldisk_text = " *FAILED* (failed drive?)"
      else:
        ldisk_text = ""
      txt += ldisk_text
    return txt

  # the header
  if top_level:
    if dev["iv_name"] is not None:
      txt = dev["iv_name"]
    else:
      txt = "disk %d" % idx
Iustin Pop's avatar
Iustin Pop committed
1012
  else:
1013
1014
1015
1016
1017
    txt = "child %d" % idx
  d1 = ["- %s: %s" % (txt, dev["dev_type"])]
  data = []
  if top_level:
    data.append(("access mode", dev["mode"]))
Iustin Pop's avatar
Iustin Pop committed
1018
  if dev["logical_id"] is not None:
1019
1020
1021
1022
1023
1024
1025
1026
    try:
      l_id = _FormatLogicalID(dev["dev_type"], dev["logical_id"])
    except ValueError:
      l_id = [str(dev["logical_id"])]
    if len(l_id) == 1:
      data.append(("logical_id", l_id[0]))
    else:
      data.extend(l_id)
Iustin Pop's avatar
Iustin Pop committed
1027
  elif dev["physical_id"] is not None:
1028
1029
    data.append("physical_id:")
    data.append([dev["physical_id"]])
1030
  if not static:
1031
    data.append(("on primary", helper(dev["dev_type"], dev["pstatus"])))
1032
  if dev["sstatus"] and not static:
1033
    data.append(("on secondary", helper(dev["dev_type"], dev["sstatus"])))
Iustin Pop's avatar
Iustin Pop committed
1034
1035

  if dev["children"]:
1036
1037
1038
1039
1040
    data.append("child devices:")
    for c_idx, child in enumerate(dev["children"]):
      data.append(_FormatBlockDevInfo(c_idx, False, child, static))
  d1.append(data)
  return d1
Iustin Pop's avatar
Iustin Pop committed
1041
1042


1043
1044
1045
1046
1047
1048
1049
1050
1051
1052
1053
1054
1055
1056
1057
1058
1059
1060
1061
1062
1063
1064
1065
1066
1067
1068
1069
1070
def _FormatList(buf, data, indent_level):
  """Formats a list of data at a given indent level.

  If the element of the list is:
    - a string, it is simply formatted as is
    - a tuple, it will be split into key, value and the all the
      values in a list will be aligned all at the same start column
    - a list, will be recursively formatted

  @type buf: StringIO
  @param buf: the buffer into which we write the output
  @param data: the list to format
  @type indent_level: int
  @param indent_level: the indent level to format at

  """
  max_tlen = max([len(elem[0]) for elem in data
                 if isinstance(elem, tuple)] or [0])
  for elem in data:
    if isinstance(elem, basestring):
      buf.write("%*s%s\n" % (2*indent_level, "", elem))
    elif isinstance(elem, tuple):
      key, value = elem
      spacer = "%*s" % (max_tlen - len(key), "")
      buf.write("%*s%s:%s %s\n" % (2*indent_level, "", key, spacer, value))
    elif isinstance(elem, list):
      _FormatList(buf, elem, indent_level+1)

Iustin Pop's avatar
Iustin Pop committed
1071
1072
1073
def ShowInstanceConfig(opts, args):
  """Compute instance run-time status.

1074
1075
1076
1077
1078
1079
1080
  @param opts: the command line options selected by the user
  @type args: list
  @param args: either an empty list, and then we query all
      instances, or should contain a list of instance names
  @rtype: int
  @return: the desired exit code

Iustin Pop's avatar
Iustin Pop committed
1081
1082
  """
  retcode = 0
1083
  op = opcodes.OpQueryInstanceData(instances=args, static=opts.static)
Iustin Pop's avatar
Iustin Pop committed
1084
1085
  result = SubmitOpCode(op)
  if not result:
1086
    ToStdout("No instances.")
Iustin Pop's avatar
Iustin Pop committed
1087
1088
1089
1090
1091
1092
1093
    return 1

  buf = StringIO()
  retcode = 0
  for instance_name in result:
    instance = result[instance_name]
    buf.write("Instance name: %s\n" % instance["name"])
1094
1095
1096
1097
1098
1099
    buf.write("State: configured to be %s" % instance["config_state"])
    if not opts.static:
      buf.write(", actual state is %s" % instance["run_state"])
    buf.write("\n")
    ##buf.write("Considered for memory checks in cluster verify: %s\n" %
    ##          instance["auto_balance"])
Iustin Pop's avatar
Iustin Pop committed
1100
1101
1102
1103
    buf.write("  Nodes:\n")
    buf.write("    - primary: %s\n" % instance["pnode"])
    buf.write("    - secondaries: %s\n" % ", ".join(instance["snodes"]))
    buf.write("  Operating system: %s\n" % instance["os"])
1104
1105
    if instance.has_key("network_port"):
      buf.write("  Allocated network port: %s\n" % instance["network_port"])
1106
1107
1108
1109
1110
1111
1112
1113
1114
1115
1116
1117
1118
1119
1120
1121
1122
1123
1124
1125
1126
1127
    buf.write("  Hypervisor: %s\n" % instance["hypervisor"])
    if instance["hypervisor"] == constants.HT_XEN_PVM:
      hvattrs = ((constants.HV_KERNEL_PATH, "kernel path"),
                 (constants.HV_INITRD_PATH, "initrd path"))
    elif instance["hypervisor"] == constants.HT_XEN_HVM:
      hvattrs = ((constants.HV_BOOT_ORDER, "boot order"),
                 (constants.HV_ACPI, "ACPI"),
                 (constants.HV_PAE, "PAE"),
                 (constants.HV_CDROM_IMAGE_PATH, "virtual CDROM"),
                 (constants.HV_NIC_TYPE, "NIC type"),
                 (constants.HV_DISK_TYPE, "Disk type"),
                 (constants.HV_VNC_BIND_ADDRESS, "VNC bind address"),
                 )
      # custom console information for HVM
      vnc_bind_address = instance["hv_actual"][constants.HV_VNC_BIND_ADDRESS]
      if vnc_bind_address == constants.BIND_ADDRESS_GLOBAL:
        vnc_console_port = "%s:%s" % (instance["pnode"],
                                      instance["network_port"])
      elif vnc_bind_address == constants.LOCALHOST_IP_ADDRESS:
        vnc_console_port = "%s:%s on node %s" % (vnc_bind_address,
                                                 instance["network_port"],
                                                 instance["pnode"])
1128
      else:
1129
1130
1131
1132
1133
1134
1135
1136
1137
1138
1139
        vnc_console_port = "%s:%s" % (vnc_bind_address,
                                      instance["network_port"])
      buf.write("    - console connection: vnc to %s\n" % vnc_console_port)

    else:
      # auto-handle other hypervisor types
      hvattrs = [(key, key) for key in instance["hv_actual"]]

    for key, desc in hvattrs:
      if key in instance["hv_instance"]:
        val = instance["hv_instance"][key]
1140
      else:
1141
1142
        val = "default (%s)" % instance["hv_actual"][key]
      buf.write("    - %s: %s\n" % (desc, val))
Iustin Pop's avatar
Iustin Pop committed
1143
    buf.write("  Hardware:\n")
Iustin Pop's avatar
Iustin Pop committed
1144
1145
1146
1147
    buf.write("    - VCPUs: %d\n" %
              instance["be_actual"][constants.BE_VCPUS])
    buf.write("    - memory: %dMiB\n" %
              instance["be_actual"][constants.BE_MEMORY])
1148
1149
1150
1151
    buf.write("    - NICs:\n")
    for idx, (mac, ip, bridge) in enumerate(instance["nics"]):
      buf.write("      - nic/%d: MAC: %s, IP: %s, bridge: %s\n" %
                (idx, mac, ip, bridge))
1152
    buf.write("  Disks:\n")
Iustin Pop's avatar
Iustin Pop committed
1153

1154
1155
    for idx, device in enumerate(instance["disks"]):
      _FormatList(buf, _FormatBlockDevInfo(idx, True, device, opts.static), 2)
Iustin Pop's avatar
Iustin Pop committed
1156

1157
  ToStdout(buf.getvalue().rstrip('\n'))
Iustin Pop's avatar
Iustin Pop committed
1158
1159
1160
  return retcode


1161
def SetInstanceParams(opts, args):
Iustin Pop's avatar
Iustin Pop committed
1162
1163
1164
1165
  """Modifies an instance.

  All parameters take effect only at the next restart of the instance.

1166
1167
1168
1169
1170
  @param opts: the command line options selected by the user
  @type args: list
  @param args: should contain only one element, the instance name
  @rtype: int
  @return: the desired exit code
Iustin Pop's avatar
Iustin Pop committed
1171
1172

  """
1173
  if not (opts.nics or opts.disks or
1174
          opts.hypervisor or opts.beparams):
1175
    ToStderr("Please give at least one of the parameters.")
Iustin Pop's avatar
Iustin Pop committed
1176
1177
    return 1

1178
  for param in opts.beparams:
1179
1180
1181
    if isinstance(opts.beparams[param], basestring):
      if opts.beparams[param].lower() == "default":
        opts.beparams[param] = constants.VALUE_DEFAULT
1182
1183
1184

  utils.ForceDictType(opts.beparams, constants.BES_PARAMETER_TYPES,
                      allowed_values=[constants.VALUE_DEFAULT])
1185
1186

  for param in opts.hypervisor:
1187
1188
1189
    if isinstance(opts.hypervisor[param], basestring):
      if opts.hypervisor[param].lower() == "default":
        opts.hypervisor[param] = constants.VALUE_DEFAULT
1190
1191
1192

  utils.ForceDictType(opts.hypervisor, constants.HVS_PARAMETER_TYPES,
                      allowed_values=[constants.VALUE_DEFAULT])
1193

1194
1195
1196
1197
1198
1199
1200
1201
1202
1203
1204
1205
1206
1207
1208
1209
1210
1211
  for idx, (nic_op, nic_dict) in enumerate(opts.nics):
    try:
      nic_op = int(nic_op)
      opts.nics[idx] = (nic_op, nic_dict)
    except ValueError:
      pass

  for idx, (disk_op, disk_dict) in enumerate(opts.disks):
    try:
      disk_op = int(disk_op)
      opts.disks[idx] = (disk_op, disk_dict)
    except ValueError:
      pass
    if disk_op == constants.DDM_ADD:
      if 'size' not in disk_dict:
        raise errors.OpPrereqError("Missing required parameter 'size'")
      disk_dict['size'] = utils.ParseUnit(disk_dict['size'])

Iustin Pop's avatar
Iustin Pop committed
1212
  op = opcodes.OpSetInstanceParams(instance_name=args[0],
1213
1214
                                   nics=opts.nics,
                                   disks=opts.disks,
1215
                                   hvparams=opts.hypervisor,
Iustin Pop's avatar
Iustin Pop committed
1216
                                   beparams=opts.beparams,
1217
                                   force=opts.force)
1218

1219
1220
  # even if here we process the result, we allow submit only
  result = SubmitOrSend(op, opts)
Iustin Pop's avatar
Iustin Pop committed
1221
1222

  if result:
1223
    ToStdout("Modified instance %s", args[0])
Iustin Pop's avatar
Iustin Pop committed
1224
    for param, data in result:
1225
1226
1227
      ToStdout(" - %-5s -> %s", param, data)
    ToStdout("Please don't forget that these parameters take effect"
             " only at the next start of the instance.")
Iustin Pop's avatar
Iustin Pop committed
1228
1229
1230
1231
1232
1233
1234
  return 0


# options used in more than one cmd
node_opt = make_option("-n", "--node", dest="node", help="Target node",
                       metavar="<node>")

1235
os_opt = cli_option("-o", "--os-type", dest="os", help="What OS to run",
Michael Hanselmann's avatar
Michael Hanselmann committed
1236
                    metavar="<os>")
1237

1238
# multi-instance selection options
1239
1240
1241
1242
1243
m_force_multi = make_option("--force-multiple", dest="force_multi",
                            help="Do not ask for confirmation when more than"
                            " one instance is affected",
                            action="store_true", default=False)

1244
1245
1246
1247
1248
1249
1250
1251
1252
1253
1254
1255
1256
1257
1258
1259
1260
1261
1262
1263
1264
m_pri_node_opt = make_option("--primary", dest="multi_mode",
                             help="Filter by nodes (primary only)",
                             const=_SHUTDOWN_NODES_PRI, action="store_const")

m_sec_node_opt = make_option("--secondary", dest="multi_mode",
                             help="Filter by nodes (secondary only)",
                             const=_SHUTDOWN_NODES_SEC, action="store_const")

m_node_opt = make_option("--node", dest="multi_mode",
                         help="Filter by nodes (primary and secondary)",
                         const=_SHUTDOWN_NODES_BOTH, action="store_const")

m_clust_opt = make_option("--all", dest="multi_mode",
                          help="Select all instances in the cluster",
                          const=_SHUTDOWN_CLUSTER, action="store_const")

m_inst_opt = make_option("--instance", dest="multi_mode",
                         help="Filter by instance name [default]",
                         const=_SHUTDOWN_INSTANCES, action="store_const")


Iustin Pop's avatar
Iustin Pop committed
1265
1266
1267
# this is defined separately due to readability only
add_opts = [
  DEBUG_OPT,
</