2 # wrapper to xmlrpc server, that support dry-run commands
3 # we dont want to have to depend on PLCAPI, so:
6 server_methods = [ ('GetNodes' , []),
8 ('SetNodePlainBootstrapfs', True),
10 ('UpdateNode' , True),
11 ('AddInterface' , True),
12 ('GetInterfaces' , True),
14 ('AddTagType' , True),
15 ('AddInterfaceTag' , True),
16 ('GetBootMedium' , True),
17 ('GetNodeGroups' , True),
18 ('AddNodeGroup' , True),
19 ('DeleteNodeGroup', True),
20 ('GetNodeTags', True),
22 ('AddInitScript' , True),
23 ('DeleteInitScript', True),
24 ('GetInitScripts', True),
26 ('AddSiteAddress' , True),
27 ('DeleteSite' , True),
28 ('DeleteSlice' , True),
30 ('AddPersonToSlice' , True),
31 ('AddSliceToNodes' , True),
32 ('AddSliceTag' , True),
34 ('UpdatePerson' , True),
35 ('AddRoleToPerson' , True),
36 ('AddPersonToSite' , True),
37 ('DeletePerson' , True),
38 ('AddPersonKey' , True),
39 ('GetPlcRelease', {'build': {'target-arch':'i386'}}),
41 ('GetLeaseGranularity', 180),
47 def __init__(self,server,dry_run,method,defaults):
51 self.defaults=defaults
52 def __call__ (self, *args):
54 print "dry_run:",self.method,
55 if len(args)>0 and type(args[0])==type({}) and args[0].has_key('AuthMethod'):
61 actual_method=getattr(self.server,self.method)
62 return actual_method(*args)
64 def __init__(self,url,dry_run=False):
65 self.apiserver = xmlrpclib.Server(url,allow_none=True)
67 for (method,defaults) in server_methods:
68 setattr(self,method,TestApiserver.Callable(self.apiserver,dry_run,method,defaults))
70 def set_dry_run (self, dry_run):
72 for (method,defaults) in server_methods:
73 getattr(self,method).dry_run = dry_run