diff options
Diffstat (limited to 'src/sbin')
| -rwxr-xr-x | src/sbin/bcfg2 | 15 |
1 files changed, 8 insertions, 7 deletions
diff --git a/src/sbin/bcfg2 b/src/sbin/bcfg2 index 97b99ab5c..148bc9204 100755 --- a/src/sbin/bcfg2 +++ b/src/sbin/bcfg2 @@ -14,7 +14,6 @@ from time import time from sys import exc_info from traceback import extract_tb -import xmlrpclib import Bcfg2.Client.Proxy def cb_sigint_handler(signum, frame): @@ -308,9 +307,12 @@ class Client: % (self.setup['file'])) else: # retrieve config from server - proxy = Bcfg2.Client.Proxy.SafeProxy('bcfg2') + proxy = Bcfg2.Client.Proxy.bcfg2() + + if self.setup['profile']: + proxy.AssertProfile(self.setup['profile']) - probe_data = proxy.run_method("GetProbes", ()) + probe_data = proxy.GetProbes() times['probe_download'] = time() @@ -329,12 +331,11 @@ class Client: self.critical_error("executing probes") # upload probe responses - proxy.run_method("RecvProbeData", (probe_info, )) + proxy.RecvProbeData(probe_info) times['probe_upload'] = time() - rawconfig = proxy.run_method("GetConfig", (self.setup['image'], - self.setup['profile'])) + rawconfig = proxy.GetConfig() times['config_download'] = time() @@ -402,7 +403,7 @@ class Client: stats.append(timeinfo) feedback.append(stats) - proxy.run_method("RecvStats", (tostring(feedback),)) + proxy.RecvStats(tostring(feedback)) if __name__ == '__main__': signal(SIGINT, cb_sigint_handler) |