Statistics
| Branch: | Tag: | Revision:

root / lib / client / gnt_group.py @ 66e884e1

History | View | Annotate | Download (4.5 kB)

1
#
2
#
3

    
4
# Copyright (C) 2010 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
"""Node group related commands"""
22

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

    
27
from ganeti.cli import *
28
from ganeti import compat
29
from ganeti import opcodes
30
from ganeti import utils
31

    
32

    
33
#: default list of fields for L{ListGroups}
34
_LIST_DEF_FIELDS = ["name", "node_cnt", "pinst_cnt"]
35

    
36

    
37
#: headers (and full field list) for L{ListGroups}
38
_LIST_HEADERS = {
39
  "name": "Group", "uuid": "UUID",
40
  "node_cnt": "Nodes", "node_list": "NodeList",
41
  "pinst_cnt": "Instances", "pinst_list": "InstanceList",
42
  "ctime": "CTime", "mtime": "MTime", "serial_no": "SerialNo",
43
}
44

    
45

    
46
def AddGroup(opts, args):
47
  """Add a node group to the cluster.
48

49
  @param opts: the command line options selected by the user
50
  @type args: list
51
  @param args: a list of length 1 with the name of the group to create
52
  @rtype: int
53
  @return: the desired exit code
54

55
  """
56
  (group_name,) = args
57
  op = opcodes.OpAddGroup(group_name=group_name)
58
  SubmitOpCode(op, opts=opts)
59

    
60

    
61
def ListGroups(opts, args):
62
  """List node groups and their properties.
63

64
  @param opts: the command line options selected by the user
65
  @type args: list
66
  @param args: groups to list, or empty for all
67
  @rtype: int
68
  @return: the desired exit code
69

70
  """
71
  desired_fields = ParseFields(opts.output, _LIST_DEF_FIELDS)
72

    
73
  output = GetClient().QueryGroups(args, desired_fields, opts.do_locking)
74

    
75
  if opts.no_headers:
76
    headers = None
77
  else:
78
    headers = _LIST_HEADERS
79

    
80
  int_type_fields = frozenset(["node_cnt", "pinst_cnt", "serial_no"])
81
  list_type_fields = frozenset(["node_list", "pinst_list"])
82
  date_type_fields = frozenset(["mtime", "ctime"])
83

    
84
  for row in output:
85
    for idx, field in enumerate(desired_fields):
86
      val = row[idx]
87

    
88
      if field in list_type_fields:
89
        val = ",".join(val)
90
      elif opts.roman_integers and field in int_type_fields:
91
        val = compat.TryToRoman(val)
92
      elif field in date_type_fields:
93
        val = utils.FormatTime(val)
94
      elif val is None:
95
        val = "?"
96

    
97
      row[idx] = str(val)
98

    
99
  data = GenerateTable(separator=opts.separator, headers=headers,
100
                       fields=desired_fields, data=output)
101

    
102
  for line in data:
103
    ToStdout(line)
104

    
105
  return 0
106

    
107

    
108
def RemoveGroup(opts, args):
109
  """Remove a node group from the cluster.
110

111
  @param opts: the command line options selected by the user
112
  @type args: list
113
  @param args: a list of length 1 with the name of the group to remove
114
  @rtype: int
115
  @return: the desired exit code
116

117
  """
118
  (group_name,) = args
119
  op = opcodes.OpRemoveGroup(group_name=group_name)
120
  SubmitOpCode(op, opts=opts)
121

    
122

    
123
def RenameGroup(opts, args):
124
  """Rename a node group.
125

126
  @param opts: the command line options selected by the user
127
  @type args: list
128
  @param args: a list of length 2, [old_name, new_name]
129
  @rtype: int
130
  @return: the desired exit code
131

132
  """
133
  old_name, new_name = args
134
  op = opcodes.OpRenameGroup(old_name=old_name, new_name=new_name)
135
  SubmitOpCode(op, opts=opts)
136

    
137

    
138
commands = {
139
  "add": (
140
    AddGroup, ARGS_ONE_GROUP, [DRY_RUN_OPT],
141
    "<group_name>", "Add a new node group to the cluster"),
142
  "list": (
143
    ListGroups, ARGS_MANY_GROUPS,
144
    [NOHDR_OPT, SEP_OPT, FIELDS_OPT, SYNC_OPT, ROMAN_OPT],
145
    "[<group_name>...]",
146
    "Lists the node groups in the cluster. The available fields are (see"
147
    " the man page for details): %s. The default list is (in order): %s." %
148
    (utils.CommaJoin(_LIST_HEADERS), utils.CommaJoin(_LIST_DEF_FIELDS))),
149
  "remove": (
150
    RemoveGroup, ARGS_ONE_GROUP, [DRY_RUN_OPT],
151
    "[--dry-run] <group_name>",
152
    "Remove an (empty) node group from the cluster"),
153
  "rename": (
154
    RenameGroup, [ArgGroup(min=2, max=2)], [DRY_RUN_OPT],
155
    "[--dry-run] <old_name> <new_name>", "Rename a node group"),
156
}
157

    
158

    
159
def Main():
160
  return GenericMain(commands)