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'}}),
45 def __init__(self,server,dry_run,method,defaults):
49 self.defaults=defaults
50 def __call__ (self, *args):
52 print "dry_run:",self.method,
53 if len(args)>0 and type(args[0])==type({}) and args[0].has_key('AuthMethod'):
59 actual_method=getattr(self.server,self.method)
60 return actual_method(*args)
62 def __init__(self,url,dry_run=False):
63 self.apiserver = xmlrpclib.Server(url,allow_none=True)
65 for (method,defaults) in server_methods:
66 setattr(self,method,TestApiserver.Callable(self.apiserver,dry_run,method,defaults))
68 def set_dry_run (self, dry_run):
70 for (method,defaults) in server_methods:
71 getattr(self,method).dry_run = dry_run