1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
|
"""
Command line handling for Cobbler.
Copyright 2007, Red Hat, Inc
Michael DeHaan <mdehaan@redhat.com>
This software may be freely redistributed under the terms of the GNU
general public license.
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., 675 Mass Ave, Cambridge, MA 02139, USA.
"""
import optparse
from cexceptions import *
from rhpl.translate import _, N_, textdomain, utf8
#=============================================================
class FunctionLoader:
"""
The F'n Loader controls processing of cobbler commands.
"""
def __init__(self):
"""
When constructed the loader has no functions.
"""
self.functions = {}
def add_func(self, obj):
"""
Adds a CobblerFunction instance to the loader.
"""
self.functions[obj.command_name()] = obj
def run(self, args):
"""
Runs a command line sequence through the loader.
"""
# if no args given, show all loaded fns
if len(args) == 1:
return self.show_options()
called_name = args[1]
# also show avail options if command name is bogus
if not called_name in self.functions.keys():
return self.show_options()
fn = self.functions[called_name]
# some functions require args, if none given, show subcommands
if len(args) == 2:
no_args_rc = fn.no_args_handler()
if no_args_rc:
return True
# finally let the object parse its own args
loaded_ok = fn.parse_args(args)
if not loaded_ok:
raise CX(_("Invalid arguments"))
return fn.run()
def show_options(self):
"""
Prints out all loaded functions.
"""
print "usage:"
print "======"
for name in self.functions.keys():
print " cobbler %s ... | --help" % name
#=============================================================
class CobblerFunction:
def __init__(self,api):
"""
Constructor requires a Cobbler API handle.
"""
self.api = api
def no_args_handler(self):
"""
Called when no additional args are given to a command. False implies
this is ok, returning True indicates an error condition.
"""
return False
def command_name(self):
"""
The name of the command, as to be entered by users.
"""
return "unspecified"
def subcommands(self):
"""
The names of any subcommands, such as "add", "edit", etc
"""
return [ ]
def run(self):
"""
Called after arguments are parsed. Return True for success.
"""
return True
def add_options(self, parser, args):
"""
Used by subclasses to add options. See subclasses for examples.
"""
pass
def parse_args(self,args):
"""
Processes arguments, called prior to run ... do not override.
"""
accum = ""
for x in args[1:]:
if not x.startswith("-"):
accum = accum + "%s " % x
else:
break
p = optparse.OptionParser(usage="cobbler %s [ARGS]" % accum)
self.add_options(p, args)
if len(args) > 2:
for x in args[2:]:
if x.startswith("-"):
break
if x not in self.subcommands():
raise CX(_("Argument (%s) not recognized") % x)
(self.options, self.args) = p.parse_args(args)
return True
def object_manipulator_start(self,new_fn,collect_fn):
"""
Boilerplate for objects that offer add/edit/delete/remove/copy functionality.
"""
if "add" in self.args:
obj = new_fn()
else:
if not self.options.name:
raise CX(_("name is required"))
if "delete" in self.args:
collect_fn().remove(self.options.name, with_delete=True)
return None
obj = collect_fn().find(self.options.name)
if not "copy" in self.args and not "rename" in self.args and self.options.name:
obj.set_name(self.options.name)
return obj
def object_manipulator_finish(self,obj,collect_fn):
"""
Boilerplate for objects that offer add/edit/delete/remove/copy functionality.
"""
if "copy" in self.args or "rename" in self.args:
if self.options.newname:
obj = obj.make_clone()
obj.set_name(self.options.newname)
else:
raise CX(_("--newname is required"))
rc = collect_fn().add(obj, with_copy=True)
if "rename" in self.args:
return collect_fn().remove(self.options.name, with_delete=True)
return rc
def no_args_handler(self):
"""
Used to accept/reject/explain subcommands. Do not override.
"""
subs = self.subcommands()
if len(subs) == 0:
return False
for x in subs:
print " cobbler %s %s --help" % (self.command_name(), x)
return True # stop here
|
