You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
135 lines
3.6 KiB
135 lines
3.6 KiB
"""EasyEngine 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 ee.core import exc
|
|
from ee.cli.ext.ee_outputhandler import EEOutputHandler
|
|
|
|
# Application default. Should update config/ee.conf to reflect any
|
|
# changes, or additions here.
|
|
defaults = init_defaults('ee')
|
|
|
|
# All internal/external plugin configurations are loaded from here
|
|
defaults['ee']['plugin_config_dir'] = '/etc/ee/plugins.d'
|
|
|
|
# External plugins (generally, do not ship with application code)
|
|
defaults['ee']['plugin_dir'] = '/var/lib/ee/plugins'
|
|
|
|
# External templates (generally, do not ship with application code)
|
|
defaults['ee']['template_dir'] = '/var/lib/ee/templates'
|
|
|
|
|
|
class EEArgHandler(ArgParseArgumentHandler):
|
|
class Meta:
|
|
label = 'ee_args_handler'
|
|
|
|
def error(self, message):
|
|
super(EEArgHandler, self).error("unknown args")
|
|
|
|
|
|
class EEApp(foundation.CementApp):
|
|
class Meta:
|
|
label = 'ee'
|
|
|
|
config_defaults = defaults
|
|
|
|
# All built-in application bootstrapping (always run)
|
|
bootstrap = 'ee.cli.bootstrap'
|
|
|
|
# Optional plugin bootstrapping (only run if plugin is enabled)
|
|
plugin_bootstrap = 'ee.cli.plugins'
|
|
|
|
# Internal templates (ship with application code)
|
|
template_module = 'ee.cli.templates'
|
|
|
|
# Internal plugins (ship with application code)
|
|
plugin_bootstrap = 'ee.cli.plugins'
|
|
|
|
extensions = ['mustache']
|
|
|
|
# default output handler
|
|
output_handler = EEOutputHandler
|
|
|
|
arg_handler = EEArgHandler
|
|
|
|
debug = TOGGLE_DEBUG
|
|
|
|
|
|
class EETestApp(EEApp):
|
|
"""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 = EEApp()
|
|
|
|
|
|
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("\nOnly root or sudo user can run this EasyEngine\n")
|
|
app.close(1)
|
|
|
|
# Setup the application
|
|
app.setup()
|
|
|
|
# Dump all arguments into ee log
|
|
app.log.debug(sys.argv)
|
|
|
|
# Run the application
|
|
app.run()
|
|
except exc.EEError 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 = EEApp(**kw)
|
|
return app
|
|
|
|
if __name__ == '__main__':
|
|
main()
|
|
|