Do not install init script in PREFIX/bin.
[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 import sys
23 from optparse import make_option
24
25 from ganeti.cli import *
26 from ganeti import cmdlib
27 from ganeti import opcodes
28 from ganeti import constants
29
30 def PrintExportList(opts, args):
31   """Prints a list of all the exported system images.
32
33   Args:
34    opts - class with options as members (should be empty)
35    args - should be empty
36
37   Returns:
38     nothing
39
40   """
41   op = opcodes.OpQueryExports(nodes=opts.nodes)
42   exports = SubmitOpCode(op)
43   for node in exports:
44     print ("Node: %s" % node)
45     print ("Exports:")
46     for instance_name in exports[node]:
47       print ("\t%s" % instance_name)
48
49
50 def ExportInstance(opts, args):
51   """Export an instance to an image in the cluster.
52
53   Args:
54    opts - class with options as members
55    args - list with a single element, the instance name
56
57   Returns:
58     1 in case of error, 0 otherwise
59
60   """
61   op = opcodes.OpExportInstance(instance_name=args[0],
62                                 target_node=opts.node,
63                                 shutdown=opts.shutdown)
64
65   SubmitOpCode(op)
66
67 def ImportInstance(opts, args):
68   """Add an instance to the cluster.
69
70   Args:
71    opts - class with options as members
72    args - list with a single element, the new instance name
73   Opts used:
74    memory - amount of memory to allocate to instance (MiB)
75    size - amount of disk space to allocate to instance (MiB)
76    os - which OS to run on instance
77    node - node to run new instance on
78    src_node - node containing the export
79    src_dir - directory on the old node with the export in it
80
81   Returns:
82     1 in case of error, 0 otherwise
83
84   """
85   instance = args[0]
86
87   op = opcodes.OpCreateInstance(instance_name=instance, mem_size=opts.mem,
88                                 disk_size=opts.size, swap_size=opts.swap,
89                                 disk_template=opts.disk_template,
90                                 mode=constants.INSTANCE_IMPORT,
91                                 pnode=opts.node, snode=opts.snode,
92                                 vcpus=opts.vcpus, ip_check=opts.ip_check,
93                                 ip=opts.ip, bridge=opts.bridge, start=False,
94                                 src_node=opts.src_node, src_path=opts.src_dir,
95                                 wait_for_sync=opts.wait_for_sync)
96   SubmitOpCode(op)
97   return 0
98
99
100 # options used in more than one cmd
101 node_opt = make_option("-n", "--node", dest="node", help="Target node",
102                        metavar="<node>")
103
104 # this is defined separately due to readability only
105 import_opts = [
106   DEBUG_OPT,
107   node_opt,
108   cli_option("-s", "--os-size", dest="size", help="Disk size",
109              default=20 * 1024, type="unit", metavar="<size>"),
110   cli_option("--swap-size", dest="swap", help="Swap size",
111              default=4 * 1024, type="unit", metavar="<size>"),
112   cli_option("-m", "--memory", dest="mem", help="Memory size",
113              default=128, type="unit", metavar="<mem>"),
114   make_option("-p", "--cpu", dest="vcpus", help="Number of virtual CPUs",
115               default=1, type="int", metavar="<PROC>"),
116   make_option("-t", "--disk-template", dest="disk_template",
117               help="Custom disk setup (diskless, plain, local_raid1 or"
118               " remote_raid1)", default=None, metavar="TEMPL"),
119   make_option("-i", "--ip", dest="ip",
120               help="IP address ('none' [default], 'auto', or specify address)",
121               default='none', type="string", metavar="<ADDRESS>"),
122   make_option("--no-wait-for-sync", dest="wait_for_sync", default=True,
123               action="store_false", help="Don't wait for sync (DANGEROUS!)"),
124   make_option("--secondary-node", dest="snode",
125               help="Secondary node for remote_raid1 disk layout",
126               metavar="<node>"),
127   make_option("-b", "--bridge", dest="bridge",
128               help="Bridge to connect this instance to",
129               default=None, metavar="<bridge>"),
130   make_option("--src-node", dest="src_node", help="Source node",
131               metavar="<node>"),
132   make_option("--src-dir", dest="src_dir", help="Source directory",
133               metavar="<dir>"),
134   make_option("--no-ip-check", dest="ip_check", default=True,
135               action="store_false", help="Don't check that the instance's IP"
136               " is alive"),
137   ]
138
139 commands = {
140   'list': (PrintExportList, ARGS_NONE,
141            [DEBUG_OPT,
142             make_option("--nodes", dest="nodes", default=[], action="append",
143                         help="List only backups stored on these nodes"),
144             ],
145            "", "Lists instance exports available in the ganeti cluster"),
146   'export': (ExportInstance, ARGS_ONE,
147              [node_opt, DEBUG_OPT, FORCE_OPT,
148               make_option("","--noshutdown", dest="shutdown",
149                           action="store_false", default=True,
150                           help="Don't shutdown the instance (unsafe)"), ],
151              "[opts...] <name>",
152              "Exports an instance to an image"),
153   'import': (ImportInstance, ARGS_ONE, import_opts, "[opts...] <name>",
154              "Imports an instance from an exported image"),
155   }
156
157 if __name__ == '__main__':
158   sys.exit(GenericMain(commands))