bcfg2-admin: set appropriate return values on error

9 files changed, 28 insertions, 60 deletions

diff --git a/misc/bcfg2.spec b/misc/bcfg2.spec
index f7289f1dd..10d913d7c 100644
--- a/misc/bcfg2.spec
+++ b/misc/bcfg2.spec
@@ -99,8 +99,6 @@ Requires:         bcfg2 = %{version}
 Requires:         python-ssl
 %endif
 Requires:         python-lxml >= 1.2.1
-%if "%{_vendor}" == "redhat"
-%endif
 %if 0%{?suse_version}
 Requires:         python-pyinotify
 Requires:         python-python-daemon
@@ -151,7 +149,7 @@ Group:            System Tools
 Requires:         bcfg2 = %{version}
 Requires:         bcfg2-server = %{version}
 
-# cherrypy 3.2.3 actually doesn't exist yet, but 3.2.2 has bugs that
+# cherrypy 3.3 actually doesn't exist yet, but 3.2 has bugs that
 # prevent it from working:
 # https://bitbucket.org/cherrypy/cherrypy/issue/1154/assertionerror-in-recv-when-ssl-is-enabled
 Requires:         python-cherrypy > 3.3
@@ -297,6 +295,8 @@ This package includes the Bcfg2 reports web frontend.
 %{?pythonpath: export PYTHONPATH="%{pythonpath}"}
 %{__python}%{pythonversion} setup.py build_sphinx
 
+sed -i "s/apache2/httpd/g" misc/apache/bcfg2.conf
+
 %install
 rm -rf %{buildroot}
 %{__python}%{pythonversion} setup.py install --root=%{buildroot} --record=INSTALLED_FILES --prefix=/usr
@@ -333,7 +333,6 @@ cp -r tools/* %{buildroot}%{_defaultdocdir}/bcfg2-server-%{version}
 cp -r build/sphinx/html/* %{buildroot}%{_defaultdocdir}/bcfg2-doc-%{version}
 
 %{__install} -d %{buildroot}%{apache_conf}/conf.d
-sed -i "s/apache2/httpd/g" misc/apache/bcfg2.conf
 %{__install} -m 644 misc/apache/bcfg2.conf %{buildroot}%{apache_conf}/conf.d/wsgi_bcfg2.conf
 
 %{__mkdir_p} %{buildroot}%{_localstatedir}/cache/%{name}
diff --git a/src/lib/Bcfg2/Server/Admin/Client.py b/src/lib/Bcfg2/Server/Admin/Client.py
index 570e993ed..187ccfd71 100644
--- a/src/lib/Bcfg2/Server/Admin/Client.py
+++ b/src/lib/Bcfg2/Server/Admin/Client.py
@@ -18,19 +18,15 @@ class Client(Bcfg2.Server.Admin.MetadataCore):
             try:
                 self.metadata.add_client(args[1])
             except MetadataConsistencyError:
-                err = sys.exc_info()[1]
-                print("Error in adding client: %s" % err)
-                raise SystemExit(1)
+                self.errExit("Error in adding client: %s" % sys.exc_info()[1])
         elif args[0] in ['delete', 'remove', 'del', 'rm']:
             try:
                 self.metadata.remove_client(args[1])
             except MetadataConsistencyError:
-                err = sys.exc_info()[1]
-                print("Error in deleting client: %s" % err)
-                raise SystemExit(1)
+                self.errExit("Error in deleting client: %s" %
+                             sys.exc_info()[1])
         elif args[0] in ['list', 'ls']:
             for client in self.metadata.list_clients():
                 print(client)
         else:
-            print("No command specified")
-            raise SystemExit(1)
+            self.errExit("No command specified")
diff --git a/src/lib/Bcfg2/Server/Admin/Compare.py b/src/lib/Bcfg2/Server/Admin/Compare.py
index c56dd0a8f..e3648a6d0 100644
--- a/src/lib/Bcfg2/Server/Admin/Compare.py
+++ b/src/lib/Bcfg2/Server/Admin/Compare.py
@@ -145,5 +145,4 @@ class Compare(Bcfg2.Server.Admin.Mode):
             (old, new) = args
             return self.compareSpecifications(new, old)
         except IndexError:
-            print(self.__call__.__doc__)
-            raise SystemExit(1)
+            self.errExit(self.__call__.__doc__)
diff --git a/src/lib/Bcfg2/Server/Admin/Minestruct.py b/src/lib/Bcfg2/Server/Admin/Minestruct.py
index 93e42305c..21a56de23 100644
--- a/src/lib/Bcfg2/Server/Admin/Minestruct.py
+++ b/src/lib/Bcfg2/Server/Admin/Minestruct.py
@@ -21,8 +21,7 @@ class Minestruct(Bcfg2.Server.Admin.StructureMode):
         try:
             (opts, args) = getopt.getopt(args, 'f:g:h')
         except:
-            self.log.error(self.__doc__)
-            raise SystemExit(1)
+            self.errExit(self.__doc__)
 
         client = args[0]
         output = sys.stdout
@@ -33,8 +32,7 @@ class Minestruct(Bcfg2.Server.Admin.StructureMode):
                 try:
                     output = open(optarg, 'w')
                 except IOError:
-                    self.log.error("Failed to open file: %s" % (optarg))
-                    raise SystemExit(1)
+                    self.errExit("Failed to open file: %s" % (optarg))
             elif opt == '-g':
                 groups = optarg.split(':')
 
@@ -44,9 +42,8 @@ class Minestruct(Bcfg2.Server.Admin.StructureMode):
                 for item in source.GetExtra(client):
                     extra.add(item)
         except:
-            self.log.error("Failed to find extra entry info for client %s" %
-                           client)
-            raise SystemExit(1)
+            self.errExit("Failed to find extra entry info for client %s" %
+                         client)
         root = lxml.etree.Element("Base")
         self.log.info("Found %d extra entries" % (len(extra)))
         add_point = root
diff --git a/src/lib/Bcfg2/Server/Admin/Pull.py b/src/lib/Bcfg2/Server/Admin/Pull.py
index 8001425df..cc0dced82 100644
--- a/src/lib/Bcfg2/Server/Admin/Pull.py
+++ b/src/lib/Bcfg2/Server/Admin/Pull.py
@@ -33,8 +33,7 @@ class Pull(Bcfg2.Server.Admin.MetadataCore):
         try:
             opts, gargs = getopt.getopt(args, 'vfIs')
         except:
-            print(self.__doc__)
-            raise SystemExit(1)
+            self.errExit(self.__doc__)
         for opt in opts:
             if opt[0] == '-v':
                 self.log = True
diff --git a/src/lib/Bcfg2/Server/Admin/Reports.py b/src/lib/Bcfg2/Server/Admin/Reports.py
index 6e313e84b..849df8025 100644
--- a/src/lib/Bcfg2/Server/Admin/Reports.py
+++ b/src/lib/Bcfg2/Server/Admin/Reports.py
@@ -79,8 +79,7 @@ class Reports(Bcfg2.Server.Admin.Mode):
 
     def __call__(self, args):
         if len(args) == 0 or args[0] == '-h':
-            print(self.__usage__)
-            raise SystemExit(0)
+            self.errExit(self.__usage__)
 
         # FIXME - dry run
 
@@ -101,9 +100,7 @@ class Reports(Bcfg2.Server.Admin.Mode):
                 management.call_command("syncdb", verbosity=vrb)
                 management.call_command("migrate", verbosity=vrb)
             except:
-                print("Update failed: %s" %
-                      traceback.format_exc().splitlines()[-1])
-                raise SystemExit(1)
+                self.errExit("Update failed: %s" % sys.exc_info()[1])
         elif args[0] == 'purge':
             expired = False
             client = None
@@ -124,22 +121,20 @@ class Reports(Bcfg2.Server.Admin.Mode):
                         maxdate = datetime.datetime.now() - \
                             datetime.timedelta(days=int(args[i + 1]))
                     except:
-                        self.log.error("Invalid number of days: %s" %
-                                       args[i + 1])
-                        raise SystemExit(-1)
+                        self.errExit("Invalid number of days: %s" %
+                                     args[i + 1])
                     i = i + 1
                 elif args[i] == '--expired':
                     expired = True
                 i = i + 1
             if expired:
                 if state:
-                    self.log.error("--state is not valid with --expired")
-                    raise SystemExit(-1)
+                    self.errExit("--state is not valid with --expired")
                 self.purge_expired(maxdate)
             else:
                 self.purge(client, maxdate, state)
         else:
-            print("Unknown command: %s" % args[0])
+            self.errExit("Unknown command: %s" % args[0])
 
     @transaction.commit_on_success
     def scrub(self):
@@ -155,8 +150,7 @@ class Reports(Bcfg2.Server.Admin.Mode):
                     (start_count - cls.objects.count(), cls.__class__.__name__))
             except:
                 print("Failed to prune %s: %s" %
-                      (cls.__class__.__name__,
-                       traceback.format_exc().splitlines()[-1]))
+                      (cls.__class__.__name__, sys.exc_info()[1]))
 
     def django_command_proxy(self, command):
         '''Call a django command'''
@@ -180,8 +174,7 @@ class Reports(Bcfg2.Server.Admin.Mode):
                 cobj = Client.objects.get(name=client)
                 ipurge = ipurge.filter(client=cobj)
             except Client.DoesNotExist:
-                self.log.error("Client %s not in database" % client)
-                raise SystemExit(-1)
+                self.errExit("Client %s not in database" % client)
             self.log.debug("Filtering by client: %s" % client)
 
         if maxdate:
diff --git a/src/lib/Bcfg2/Server/Admin/Syncdb.py b/src/lib/Bcfg2/Server/Admin/Syncdb.py
index 4ba840b86..53cfd1bec 100644
--- a/src/lib/Bcfg2/Server/Admin/Syncdb.py
+++ b/src/lib/Bcfg2/Server/Admin/Syncdb.py
@@ -22,10 +22,7 @@ class Syncdb(Bcfg2.Server.Admin.Mode):
             call_command("syncdb", interactive=False, verbosity=0)
             self._database_available = True
         except ImproperlyConfigured:
-            err = sys.exc_info()[1]
-            self.log.error("Django configuration problem: %s" % err)
-            raise SystemExit(1)
+            self.errExit("Django configuration problem: %s" %
+                         sys.exc_info()[1])
         except:
-            err = sys.exc_info()[1]
-            self.log.error("Database update failed: %s" % err)
-            raise SystemExit(1)
+            self.errExit("Database update failed: %s" % sys.exc_info()[1])
diff --git a/src/lib/Bcfg2/Server/Admin/Xcmd.py b/src/lib/Bcfg2/Server/Admin/Xcmd.py
index be556bed4..e3173e20a 100644
--- a/src/lib/Bcfg2/Server/Admin/Xcmd.py
+++ b/src/lib/Bcfg2/Server/Admin/Xcmd.py
@@ -31,27 +31,15 @@ class Xcmd(Bcfg2.Server.Admin.Mode):
                                            ca=setup['ca'],
                                            timeout=setup['timeout'])
         if len(setup['args']) == 0:
-            print("Usage: xcmd <xmlrpc method> <optional arguments>")
-            return
+            self.errExit("Usage: xcmd <xmlrpc method> <optional arguments>")
         cmd = setup['args'][0]
         args = ()
         if len(setup['args']) > 1:
             args = tuple(setup['args'][1:])
         try:
             data = getattr(proxy, cmd)(*args)
-        except xmlrpclib.Fault:
-            flt = sys.exc_info()[1]
-            if flt.faultCode == 7:
-                print("Unknown method %s" % cmd)
-                return
-            elif flt.faultCode == 20:
-                return
-            else:
-                raise
         except Bcfg2.Proxy.ProxyError:
-            err = sys.exc_info()[1]
-            print("Proxy Error: %s" % err)
-            return
+            self.errExit("Proxy Error: %s" % sys.exc_info()[1])
 
         if data is not None:
             print(data)
diff --git a/src/sbin/bcfg2-admin b/src/sbin/bcfg2-admin
index 31e49c00b..14d188342 100755
--- a/src/sbin/bcfg2-admin
+++ b/src/sbin/bcfg2-admin
@@ -83,7 +83,7 @@ def main():
             raise SystemExit(1)
         mode = mode_cls(setup)
         try:
-            mode(setup['args'][1:])
+            return mode(setup['args'][1:])
         finally:
             mode.shutdown()
     else:
@@ -93,6 +93,6 @@ def main():
 
 if __name__ == '__main__':
     try:
-        main()
+        sys.exit(main())
     except KeyboardInterrupt:
         raise SystemExit(1)