daemon.GenericMain: Don't use list of multithreaded daemons
[ganeti-local] / 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
22 # pylint: disable-msg=W0401,W0614
23 # W0401: Wildcard import ganeti.cli
24 # W0614: Unused import %s from wildcard import (since we need cli)
25
26 import sys
27
28 from ganeti.cli import *
29 from ganeti import opcodes
30 from ganeti import constants
31 from ganeti import errors
32 from ganeti import utils
33
34
35 _VALUE_TRUE = "true"
36
37
38 def PrintExportList(opts, args):
39   """Prints a list of all the exported system images.
40
41   @param opts: the command line options selected by the user
42   @type args: list
43   @param args: should be an empty list
44   @rtype: int
45   @return: the desired exit code
46
47   """
48   exports = GetClient().QueryExports(opts.nodes, False)
49   retcode = 0
50   for node in exports:
51     ToStdout("Node: %s", node)
52     ToStdout("Exports:")
53     if isinstance(exports[node], list):
54       for instance_name in exports[node]:
55         ToStdout("\t%s", instance_name)
56     else:
57       ToStdout("  Could not get exports list")
58       retcode = 1
59   return retcode
60
61
62 def ExportInstance(opts, args):
63   """Export an instance to an image in the cluster.
64
65   @param opts: the command line options selected by the user
66   @type args: list
67   @param args: should contain only one element, the name
68       of the instance to be exported
69   @rtype: int
70   @return: the desired exit code
71
72   """
73   op = opcodes.OpExportInstance(instance_name=args[0],
74                                 target_node=opts.node,
75                                 shutdown=opts.shutdown,
76                                 shutdown_timeout=opts.shutdown_timeout)
77
78   fin_resu, dlist = SubmitOpCode(op)
79   if not isinstance(dlist, list):
80     ToStderr("Cannot parse execution results")
81     return 1
82   tot_dsk = len(dlist)
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   instance = args[0]
121   op = opcodes.OpRemoveExport(instance_name=args[0])
122
123   SubmitOpCode(op)
124   return 0
125
126
127 # this is defined separately due to readability only
128 import_opts = [
129   NODE_PLACEMENT_OPT,
130   BACKEND_OPT,
131   DISK_TEMPLATE_OPT,
132   DISK_OPT,
133   OS_SIZE_OPT,
134   NET_OPT,
135   NONICS_OPT,
136   NWSYNC_OPT,
137   SRC_DIR_OPT,
138   SRC_NODE_OPT,
139   NOIPCHECK_OPT,
140   IALLOCATOR_OPT,
141   FILESTORE_DIR_OPT,
142   FILESTORE_DRIVER_OPT,
143   HYPERVISOR_OPT,
144   SUBMIT_OPT,
145   ]
146
147 commands = {
148   'list': (
149     PrintExportList, ARGS_NONE,
150     [NODE_LIST_OPT],
151     "", "Lists instance exports available in the ganeti cluster"),
152   'export': (
153     ExportInstance, ARGS_ONE_INSTANCE,
154     [FORCE_OPT, SINGLE_NODE_OPT, NOSHUTDOWN_OPT, SHUTDOWN_TIMEOUT_OPT],
155     "-n <target_node> [opts...] <name>",
156     "Exports an instance to an image"),
157   'import': (
158     ImportInstance, ARGS_ONE_INSTANCE, import_opts,
159     "[...] -t disk-type -n node[:secondary-node] <name>",
160     "Imports an instance from an exported image"),
161   'remove': (
162     RemoveExport, [ArgUnknown(min=1, max=1)], [],
163     "<name>", "Remove exports of named instance from the filesystem."),
164   }
165
166
167 if __name__ == '__main__':
168   sys.exit(GenericMain(commands))