2 # wrapper to xmlrpc server, that support dry-run commands
3 # we dont want to have to depend on PLCAPI, so:
6 # the default value is for the dry run mode
7 server_methods = [ ('GetNodes' , []),
9 ('SetNodePlainBootstrapfs', True),
10 ('DeleteNode' , True),
11 ('UpdateNode' , True),
12 ('AddInterface' , True),
13 ('GetInterfaces' , True),
15 ('AddTagType' , True),
16 ('AddInterfaceTag' , True),
17 ('GetBootMedium' , True),
18 ('GetNodeGroups' , True),
19 ('AddNodeGroup' , True),
20 ('DeleteNodeGroup', True),
21 ('GetNodeTags', True),
23 ('AddInitScript' , True),
24 ('DeleteInitScript', True),
25 ('GetInitScripts', True),
27 ('AddSiteAddress' , True),
28 ('DeleteSite' , True),
29 ('DeleteSlice' , True),
31 ('AddPersonToSlice' , True),
32 ('AddSliceToNodes' , True),
33 ('AddSliceTag' , True),
35 ('UpdatePerson' , True),
36 ('AddRoleToPerson' , True),
37 ('AddPersonToSite' , True),
38 ('DeletePerson' , True),
39 ('AddPersonKey' , True),
40 ('GetPlcRelease', {'build': {'target-arch':'i386'}}),
42 ('GetLeaseGranularity', 180),
45 ('DeleteLeases',True),
50 def __init__(self,server,dry_run,method,defaults):
54 self.defaults=defaults
55 def __call__ (self, *args):
57 print "dry_run:",self.method,
58 if len(args)>0 and type(args[0])==type({}) and args[0].has_key('AuthMethod'):
64 actual_method=getattr(self.server,self.method)
65 return actual_method(*args)
67 def __init__(self,url,dry_run=False):
68 self.apiserver = xmlrpclib.Server(url,allow_none=True)
70 for (method,defaults) in server_methods:
71 setattr(self,method,TestApiserver.Callable(self.apiserver,dry_run,method,defaults))
73 def set_dry_run (self, dry_run):
75 for (method,defaults) in server_methods:
76 getattr(self,method).dry_run = dry_run