gnt-backup 10.3 KB
Newer Older
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)

26
27
28
29
30
import sys

from ganeti.cli import *
from ganeti import opcodes
from ganeti import constants
31
32
from ganeti import errors
from ganeti import utils
33

34

35
36
_VALUE_TRUE = "true"

37

38
39
40
def PrintExportList(opts, args):
  """Prints a list of all the exported system images.

41
42
43
44
45
  @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
46
47

  """
48
  exports = GetClient().QueryExports(opts.nodes, False)
49
  retcode = 0
50
  for node in exports:
51
52
    ToStdout("Node: %s", node)
    ToStdout("Exports:")
53
54
    if isinstance(exports[node], list):
      for instance_name in exports[node]:
55
        ToStdout("\t%s", instance_name)
56
    else:
57
      ToStdout("  Could not get exports list")
58
59
      retcode = 1
  return retcode
60
61
62
63
64


def ExportInstance(opts, args):
  """Export an instance to an image in the cluster.

65
66
67
68
69
70
  @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 exported
  @rtype: int
  @return: the desired exit code
71
72
73
74
75
76

  """
  op = opcodes.OpExportInstance(instance_name=args[0],
                                target_node=opts.node,
                                shutdown=opts.shutdown)

Iustin Pop's avatar
Iustin Pop committed
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
  fin_resu, dlist = SubmitOpCode(op)
  if not isinstance(dlist, list):
    ToStderr("Cannot parse execution results")
    return 1
  tot_dsk = len(dlist)
  # TODO: handle diskless instances
  if dlist.count(False) == 0:
    # all OK
    rcode = 0
  elif dlist.count(True) == 0:
    ToStderr("Error: No disks were backed up successfully."
             " The export doesn't have any valid data,"
             " it is recommended to retry the operation.")
    rcode = 1
  else:
    ToStderr("Partial export failure: %d disks backed up, %d disks failed.",
             dlist.count(True), dlist.count(False))
    rcode = 2
  if not fin_resu:
    rcode = 1
  return rcode
98

99
100
101
def ImportInstance(opts, args):
  """Add an instance to the cluster.

102
103
104
105
106
  @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
107
108
109
110

  """
  instance = args[0]

111
112
  (pnode, snode) = SplitNodeOption(opts.node)

113
114
115
116
117
  hypervisor = None
  hvparams = {}
  if opts.hypervisor:
    hypervisor, hvparams = opts.hypervisor

118
119
120
121
122
123
  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
Iustin Pop's avatar
Iustin Pop committed
124
    for nidx, ndict in opts.nics:
125
      nidx = int(nidx)
126
127
128
      if not isinstance(ndict, dict):
        msg = "Invalid nic/%d value: expected dict, got %s" % (nidx, ndict)
        raise errors.OpPrereqError(msg)
129
130
131
132
133
134
135
136
137
      nics[nidx] = ndict
  elif opts.no_nics:
    # no nics
    nics = []
  else:
    # default of one nic, all auto
    nics = [{}]

  if opts.disk_template == constants.DT_DISKLESS:
138
    if opts.disks or opts.sd_size is not None:
139
140
141
142
      raise errors.OpPrereqError("Diskless instance but disk"
                                 " information passed")
    disks = []
  else:
143
    if not opts.disks and not opts.sd_size:
144
      raise errors.OpPrereqError("No disk information specified")
145
146
147
148
149
    if opts.disks and opts.sd_size is not None:
      raise errors.OpPrereqError("Please use either the '--disk' or"
                                 " '-s' option")
    if opts.sd_size is not None:
      opts.disks = [(0, {"size": opts.sd_size})]
150
151
152
153
154
155
156
    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)
157
158
159
160
      if not isinstance(ddict, dict):
        msg = "Invalid disk/%d value: expected dict, got %s" % (didx, ddict)
        raise errors.OpPrereqError(msg)
      elif "size" not in ddict:
161
162
163
164
165
166
167
168
        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

169
  utils.ForceDictType(opts.beparams, constants.BES_PARAMETER_TYPES)
170
  utils.ForceDictType(hvparams, constants.HVS_PARAMETER_TYPES)
171

Iustin Pop's avatar
Iustin Pop committed
172
  op = opcodes.OpCreateInstance(instance_name=instance,
173
                                disk_template=opts.disk_template,
174
175
                                disks=disks,
                                nics=nics,
176
                                mode=constants.INSTANCE_IMPORT,
177
                                pnode=pnode, snode=snode,
Iustin Pop's avatar
Iustin Pop committed
178
                                ip_check=opts.ip_check,
179
                                start=False,
180
                                src_node=opts.src_node, src_path=opts.src_dir,
181
                                wait_for_sync=opts.wait_for_sync,
182
183
                                file_storage_dir=opts.file_storage_dir,
                                file_driver=opts.file_driver,
184
                                iallocator=opts.iallocator,
185
186
187
                                hypervisor=hypervisor,
                                hvparams=hvparams,
                                beparams=opts.beparams)
188

189
190
191
192
  SubmitOpCode(op)
  return 0


193
194
195
def RemoveExport(opts, args):
  """Remove an export from the cluster.

196
197
198
199
200
201
  @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 whose backup should be removed
  @rtype: int
  @return: the desired exit code
202
203
204
205
206
207
208
209
210

  """
  instance = args[0]
  op = opcodes.OpRemoveExport(instance_name=args[0])

  SubmitOpCode(op)
  return 0


211
212
213
# this is defined separately due to readability only
import_opts = [
  DEBUG_OPT,
214
215
  cli_option("-n", "--node", dest="node",
             help="Target node and optional secondary node",
216
217
             metavar="<pnode>[:<snode>]",
             completion_suggest=OPT_COMPL_INST_ADD_NODES),
218
219
220
  cli_option("-B", "--backend", dest="beparams",
             type="keyval", default={},
             help="Backend parameters"),
221
  DISK_TEMPLATE_OPT,
222
  cli_option("--disk", help="Disk information",
223
224
225
             default=[], dest="disks",
             action="append",
             type="identkeyval"),
226
227
228
229
  cli_option("-s", "--os-size", dest="sd_size", help="Disk size for a"
             " single-disk configuration, when not using the --disk option,"
             " in MiB unless a suffix is used",
             default=None, type="unit", metavar="<size>"),
230
  cli_option("--net", help="NIC information",
231
232
233
             default=[], dest="nics",
             action="append",
             type="identkeyval"),
Iustin Pop's avatar
Iustin Pop committed
234
  NONICS_OPT,
235
  NWSYNC_OPT,
236
  cli_option("--src-node", dest="src_node", help="Source node",
237
238
             metavar="<node>",
             completion_suggest=OPT_COMPL_ONE_NODE),
239
240
241
242
243
244
245
  cli_option("--src-dir", dest="src_dir", help="Source directory",
             metavar="<dir>"),
  cli_option("--no-ip-check", dest="ip_check", default=True,
             action="store_false", help="Don't check that the instance's IP"
             " is alive"),
  cli_option("-I", "--iallocator", metavar="<NAME>",
             help="Select nodes for the instance automatically using the"
246
247
             " <NAME> iallocator plugin", default=None, type="string",
             completion_suggest=OPT_COMPL_ONE_IALLOCATOR),
248
249
250
251
252
  cli_option("--file-storage-dir", dest="file_storage_dir",
             help="Relative path under default cluster-wide file storage dir"
             " to store file-based disks", default=None,
             metavar="<DIR>"),
  cli_option("--file-driver", dest="file_driver", help="Driver to use"
253
254
             " for image files", default="loop", metavar="<DRIVER>",
             choices=list(constants.FILE_DRIVER)),
255
  cli_option("-H", "--hypervisor", dest="hypervisor",
256
257
258
             help="Hypervisor and hypervisor options, in the format"
             " hypervisor:option=value,option=value,...", default=None,
             type="identkeyval"),
259
260
261
  ]

commands = {
262
  'list': (PrintExportList, ARGS_NONE,
263
           [DEBUG_OPT,
264
265
            cli_option("--node", dest="nodes", default=[], action="append",
                       help="List only backups stored on this node"
266
267
                            " (can be used multiple times)",
                       completion_suggest=OPT_COMPL_ONE_NODE),
268
            ],
269
           "", "Lists instance exports available in the ganeti cluster"),
270
  'export': (ExportInstance, ARGS_ONE_INSTANCE,
271
             [DEBUG_OPT, FORCE_OPT,
272
              cli_option("-n", "--node", dest="node", help="Target node",
273
274
                         metavar="<node>",
                         completion_suggest=OPT_COMPL_ONE_NODE),
275
276
277
              cli_option("","--noshutdown", dest="shutdown",
                         action="store_false", default=True,
                         help="Don't shutdown the instance (unsafe)"), ],
278
             "-n <target_node> [opts...] <name>",
279
             "Exports an instance to an image"),
280
  'import': (ImportInstance, ARGS_ONE_INSTANCE, import_opts,
281
282
             ("[...] -t disk-type -n node[:secondary-node]"
              " <name>"),
283
             "Imports an instance from an exported image"),
284
  'remove': (RemoveExport, [ArgUnknown(min=1, max=1)],
285
             [DEBUG_OPT],
286
             "<name>",
287
             "Remove exports of named instance from the filesystem."),
288
289
  }

290

291
if __name__ == '__main__':
292
  sys.exit(GenericMain(commands))