code.grnet.gr Git - ganeti-local/blob - scripts/gnt-backup

   1 #!/usr/bin/python
   2 #
   3
   4 # Copyright (C) 2006, 2007 Google Inc.
   5 #
   6 # This program is free software; you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation; either version 2 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful, but
  12 # WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 # General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program; if not, write to the Free Software
  18 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  19 # 02110-1301, USA.
  20
  21 """Backup related commands"""
  22
  23 # pylint: disable-msg=W0401,W0613,W0614,C0103
  24 # W0401: Wildcard import ganeti.cli
  25 # W0613: Unused argument, since all functions follow the same API
  26 # W0614: Unused import %s from wildcard import (since we need cli)
  27 # C0103: Invalid name gnt-backup
  28
  29 import sys
  30
  31 from ganeti.cli import *
  32 from ganeti import opcodes
  33 from ganeti import constants
  34
  35
  36 _VALUE_TRUE = "true"
  37
  38
  39 def PrintExportList(opts, args):
  40   """Prints a list of all the exported system images.
  41
  42   @param opts: the command line options selected by the user
  43   @type args: list
  44   @param args: should be an empty list
  45   @rtype: int
  46   @return: the desired exit code
  47
  48   """
  49   exports = GetClient().QueryExports(opts.nodes, False)
  50   retcode = 0
  51   for node in exports:
  52     ToStdout("Node: %s", node)
  53     ToStdout("Exports:")
  54     if isinstance(exports[node], list):
  55       for instance_name in exports[node]:
  56         ToStdout("\t%s", instance_name)
  57     else:
  58       ToStdout("  Could not get exports list")
  59       retcode = 1
  60   return retcode
  61
  62
  63 def ExportInstance(opts, args):
  64   """Export an instance to an image in the cluster.
  65
  66   @param opts: the command line options selected by the user
  67   @type args: list
  68   @param args: should contain only one element, the name
  69       of the instance to be exported
  70   @rtype: int
  71   @return: the desired exit code
  72
  73   """
  74   op = opcodes.OpExportInstance(instance_name=args[0],
  75                                 target_node=opts.node,
  76                                 shutdown=opts.shutdown,
  77                                 shutdown_timeout=opts.shutdown_timeout)
  78
  79   fin_resu, dlist = SubmitOpCode(op, opts=opts)
  80   if not isinstance(dlist, list):
  81     ToStderr("Cannot parse execution results")
  82     return 1
  83   # TODO: handle diskless instances
  84   if dlist.count(False) == 0:
  85     # all OK
  86     rcode = 0
  87   elif dlist.count(True) == 0:
  88     ToStderr("Error: No disks were backed up successfully."
  89              " The export doesn't have any valid data,"
  90              " it is recommended to retry the operation.")
  91     rcode = 1
  92   else:
  93     ToStderr("Partial export failure: %d disks backed up, %d disks failed.",
  94              dlist.count(True), dlist.count(False))
  95     rcode = 2
  96   if not fin_resu:
  97     rcode = 1
  98   return rcode
  99
 100 def ImportInstance(opts, args):
 101   """Add an instance to the cluster.
 102
 103   This is just a wrapper over GenericInstanceCreate.
 104
 105   """
 106   return GenericInstanceCreate(constants.INSTANCE_IMPORT, opts, args)
 107
 108
 109 def RemoveExport(opts, args):
 110   """Remove an export from the cluster.
 111
 112   @param opts: the command line options selected by the user
 113   @type args: list
 114   @param args: should contain only one element, the name of the
 115       instance whose backup should be removed
 116   @rtype: int
 117   @return: the desired exit code
 118
 119   """
 120   op = opcodes.OpRemoveExport(instance_name=args[0])
 121
 122   SubmitOpCode(op, opts=opts)
 123   return 0
 124
 125
 126 # this is defined separately due to readability only
 127 import_opts = [
 128   BACKEND_OPT,
 129   DISK_OPT,
 130   DISK_TEMPLATE_OPT,
 131   FILESTORE_DIR_OPT,
 132   FILESTORE_DRIVER_OPT,
 133   HYPERVISOR_OPT,
 134   IALLOCATOR_OPT,
 135   IDENTIFY_DEFAULTS_OPT,
 136   NET_OPT,
 137   NODE_PLACEMENT_OPT,
 138   NOIPCHECK_OPT,
 139   NONAMECHECK_OPT,
 140   NONICS_OPT,
 141   NWSYNC_OPT,
 142   OS_SIZE_OPT,
 143   SRC_DIR_OPT,
 144   SRC_NODE_OPT,
 145   SUBMIT_OPT,
 146   ]
 147
 148 commands = {
 149   'list': (
 150     PrintExportList, ARGS_NONE,
 151     [NODE_LIST_OPT],
 152     "", "Lists instance exports available in the ganeti cluster"),
 153   'export': (
 154     ExportInstance, ARGS_ONE_INSTANCE,
 155     [FORCE_OPT, SINGLE_NODE_OPT, NOSHUTDOWN_OPT, SHUTDOWN_TIMEOUT_OPT],
 156     "-n <target_node> [opts...] <name>",
 157     "Exports an instance to an image"),
 158   'import': (
 159     ImportInstance, ARGS_ONE_INSTANCE, import_opts,
 160     "[...] -t disk-type -n node[:secondary-node] <name>",
 161     "Imports an instance from an exported image"),
 162   'remove': (
 163     RemoveExport, [ArgUnknown(min=1, max=1)], [],
 164     "<name>", "Remove exports of named instance from the filesystem."),
 165   }
 166
 167
 168 if __name__ == '__main__':
 169   sys.exit(GenericMain(commands))