def module_execute(module_name, entry_name, skull_txn=None, data=None, skull_txndata=None):
if module_name is None or isinstance(module_name, types.StringType) is False:
return
if entry_name is None or isinstance(entry_name, types.StringType) is False:
return
user_module = UserModuleTables.get(module_name)
if user_module is None:
Logger.fatal('module_execute', 'Cannot find user module: {}'.format(module_name),
'please check the module has contained correct entries')
return
entry_func = user_module['entries'][entry_name]
if entry_name == MODULE_INIT_FUNCNAME:
config = user_module['config']
skull_module_executor.run_module_init(entry_func, config)
elif entry_name == MODULE_RELEASE_FUNCNAME:
skull_module_executor.run_module_release(entry_func)
elif entry_name == MODULE_RUN_FUNCNAME:
return skull_module_executor.run_module_run(entry_func, skull_txn)
elif entry_name == MODULE_UNPACK_FUNCNAME:
return skull_module_executor.run_module_unpack(entry_func, skull_txn, data)
elif entry_name == MODULE_PACK_FUNCNAME:
skull_module_executor.run_module_pack(entry_func, skull_txn, skull_txndata)
else:
Logger.error('module_execute', 'Unknown entry name for execution: {}'.format(entry_name),
'Please check the engine code, it shouldn\'t be')
raise
def module_load_config(module_name, config_file_name):
if module_name is None or isinstance(module_name, types.StringType) is False:
return
if config_file_name is None or isinstance(config_file_name, types.StringType) is False:
return
user_module = UserModuleTables.get(module_name)
if user_module is None:
Logger.fatal('load_config', 'Cannot find user module: {}'.format(module_name),
'please check the module has contained correct entries')
raise
# Loading the config yaml file
yaml_file = None
try:
yaml_file = file(config_file_name, 'r')
except Exception as e:
Logger.fatal('load_config',
'Cannot load user module {} config {} : {}'.format(module_name, config_file_name, str(e)),
'please check the config content format')
raise
conf_yml_obj = yaml.load(yaml_file)
user_module['config'] = conf_yml_obj
user_module['config_name'] = config_file_name
yaml_file.close()
def module_load(module_name):
#print sys.path
#print "module name: %s" % module_name
full_name = 'skull.modules.' + module_name + '.module'
# Create Global Environment Vars
global EnvVars
if EnvVars is None:
EnvVars = {
'working_dir': os.getcwd()
}
# User Module Table
userModule = {
'env' : EnvVars,
'name' : module_name,
'fullname' : full_name,
'module_obj' : None,
'config' : None,
'config_name:': '',
'entries' : {}
}
# 1. Load user module
try:
#print "Loading user module: %s" % full_name
uModule = __import__(full_name)
except Exception as e:
Logger.fatal('module_load', 'Cannot load user module {}: {}'.format(module_name, str(e)),
'please check the module whether exist')
raise
userModule['module_obj'] = uModule
# 2. Loader user module entries into userModule
# 2.1 Load module_init
if _load_user_entry(module_name, MODULE_INIT_FUNCNAME, uModule, userModule, True) is None:
return False
# 2.2 Load module_release
if _load_user_entry(module_name, MODULE_RELEASE_FUNCNAME, uModule, userModule, True) is None:
return False
# 2.3 Load module_run
if _load_user_entry(module_name, MODULE_RUN_FUNCNAME, uModule, userModule, True) is None:
return False
# 2.4 Load module_pack
_load_user_entry(module_name, MODULE_PACK_FUNCNAME, uModule, userModule, False)
# 2.5 Load module_unpack
_load_user_entry(module_name, MODULE_UNPACK_FUNCNAME, uModule, userModule, False)
# 3. Assembe user module into Global Module Table
UserModuleTables[module_name] = userModule
return True
def module_init(config):
print "py module init"
Logger.trace('py module init: trace test')
Logger.debug('py module init: debug test')
Logger.info('1', 'py module init: info test')
Logger.warn('2', 'py module init: warn test', 'no suggestion')
Logger.error('3', 'py module init: error test', 'no solution')
Logger.fatal('4', 'py module init: fatal test', 'no solution')
return
def _load_user_entry(moduleName, entryName, uModule, userModule, isPrintError):
pkg_modules = uModule.__dict__.get('modules')
pkg_module = pkg_modules.__dict__.get(moduleName)
entry_module = pkg_module.__dict__.get('module')
func = entry_module.__dict__.get(entryName)
if isinstance(func, types.FunctionType):
userModule['entries'][entryName] = func
return func
else:
if isPrintError:
Logger.fatal('user_entry', 'Not found {} entry'.format(entryName),
'Please check the source file')
return None
