Refactored

wo/cli/main.py

@@ -0,0 +1,135 @@
+"""WordOps main application entry point."""
+import sys
+import os
+
+# this has to happen after you import sys, but before you import anything
+# from Cement "source: https://github.com/datafolklabs/cement/issues/290"
+if '--debug' in sys.argv:
+    sys.argv.remove('--debug')
+    TOGGLE_DEBUG = True
+else:
+    TOGGLE_DEBUG = False
+
+from cement.core import foundation
+from cement.utils.misc import init_defaults
+from cement.core.exc import FrameworkError, CaughtSignal
+from cement.ext.ext_argparse import ArgParseArgumentHandler
+from wo.core import exc
+from wo.cli.ext.wo_outputhandler import WOOutputHandler
+
+# Application default.  Should update config/wo.conf to reflect any
+# changes, or additions here.
+defaults = init_defaults('wo')
+
+# All internal/external plugin configurations are loaded from here
+defaults['wo']['plugin_config_dir'] = '/etc/wo/plugins.d'
+
+# External plugins (generally, do not ship with application code)
+defaults['wo']['plugin_dir'] = '/var/lib/wo/plugins'
+
+# External templates (generally, do not ship with application code)
+defaults['wo']['template_dir'] = '/var/lib/wo/templates'
+
+
+class WOArgHandler(ArgParseArgumentHandler):
+    class Meta:
+        label = 'wo_args_handler'
+
+    def error(self, message):
+        super(WOArgHandler, self).error("unknown args")
+
+
+class WOApp(foundation.CementApp):
+    class Meta:
+        label = 'wo'
+
+        config_defaults = defaults
+
+        # All built-in application bootstrapping (always run)
+        bootstrap = 'wo.cli.bootstrap'
+
+        # Optional plugin bootstrapping (only run if plugin is enabled)
+        plugin_bootstrap = 'wo.cli.plugins'
+
+        # Internal templates (ship with application code)
+        template_module = 'wo.cli.templates'
+
+        # Internal plugins (ship with application code)
+        plugin_bootstrap = 'wo.cli.plugins'
+
+        extensions = ['mustache']
+
+        # default output handler
+        output_handler = WOOutputHandler
+
+        arg_handler = WOArgHandler
+
+        debug = TOGGLE_DEBUG
+
+
+class WOTestApp(WOApp):
+    """A test app that is better suited for testing."""
+    class Meta:
+        argv = []
+        config_files = []
+
+
+# Define the applicaiton object outside of main, as some libraries might wish
+# to import it as a global (rather than passing it into another class/func)
+app = WOApp()
+
+
+def main():
+    try:
+        global sys
+        # Default our exit status to 0 (non-error)
+        code = 0
+
+        # if not root...kick out
+        if not os.geteuid() == 0:
+            print("\nNon-privileged users cant use WordOps. Switch to root or invoke sudo.\n")
+            app.close(1)
+
+        # Setup the application
+        app.setup()
+
+        # Dump all arguments into wo log
+        app.log.debug(sys.argv)
+
+        # Run the application
+        app.run()
+    except exc.WOError as e:
+        # Catch our application errors and exit 1 (error)
+        code = 1
+        print(e)
+    except FrameworkError as e:
+        # Catch framework errors and exit 1 (error)
+        code = 1
+        print(e)
+    except CaughtSignal as e:
+        # Default Cement signals are SIGINT and SIGTERM, exit 0 (non-error)
+        code = 0
+        print(e)
+    except Exception as e:
+        code = 1
+        print(e)
+    finally:
+        # Print an exception (if it occurred) and --debug was passed
+        if app.debug:
+            import sys
+            import traceback
+
+            exc_type, exc_value, exc_traceback = sys.exc_info()
+            if exc_traceback is not None:
+                traceback.print_exc()
+
+        # # Close the application
+    app.close(code)
+
+
+def get_test_app(**kw):
+    app = WOApp(**kw)
+    return app
+
+if __name__ == '__main__':
+    main()