"""WordOps main application entry point.""" import os import sys from cement.core.exc import CaughtSignal, FrameworkError from cement.core.foundation import CementApp from cement.ext.ext_argparse import ArgParseArgumentHandler from cement.utils.misc import init_defaults from wo.cli.ext.wo_outputhandler import WOOutputHandler from wo.core import exc # 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 # 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(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 exit_on_close = True 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 CaughtSignal as e: # Default Cement signals are SIGINT and SIGTERM, exit 0 (non-error) code = 0 print(e) except FrameworkError as e: # Catch framework errors and exit 1 (error) code = 1 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()