diff options
Diffstat (limited to 'src/sbin')
| -rwxr-xr-x | src/sbin/bcfg2 | 22 |
1 files changed, 17 insertions, 5 deletions
diff --git a/src/sbin/bcfg2 b/src/sbin/bcfg2 index 5265601ab..3c97e8bab 100755 --- a/src/sbin/bcfg2 +++ b/src/sbin/bcfg2 @@ -6,7 +6,7 @@ __revision__ = '$Revision$' from ConfigParser import ConfigParser, NoSectionError, NoOptionError from lxml.etree import Element, XML, tostring, XMLSyntaxError -import getopt, logging, os, signal, sys, tempfile, time, traceback +import getopt, logging, os, signal, sys, tempfile, time, traceback, xmlrpclib import Bcfg2.Client.Proxy, Bcfg2.Logging def cb_sigint_handler(signum, frame): @@ -285,8 +285,12 @@ class Client: if self.setup['profile']: proxy.AssertProfile(self.setup['profile']) - - probe_data = proxy.GetProbes() + + try: + probe_data = proxy.GetProbes() + except xmlrpclib.Fault: + self.logger.error("Failed to download probes from bcfg2") + raise SystemExit, 1 times['probe_download'] = time.time() @@ -309,7 +313,11 @@ class Client: times['probe_upload'] = time.time() - rawconfig = proxy.GetConfig() + try: + rawconfig = proxy.GetConfig() + except xmlrpclib.Fault: + self.logger.error("Failed to download configuration from bcfg2") + raise SystemExit, 2 times['config_download'] = time.time() @@ -378,7 +386,11 @@ class Client: stats.append(timeinfo) feedback.append(stats) - proxy.RecvStats(tostring(feedback)) + try: + proxy.RecvStats(tostring(feedback)) + except xmlrpclib.Fault: + self.logger.error("Failed to upload configuration statistics") + raise SystemExit, 2 if __name__ == '__main__': signal.signal(signal.SIGINT, cb_sigint_handler) |