def init(formatter):
try:
action_loader.load_actions()
except LoadError as e:
print_error_list(formatter, "action import: ", e)
try:
template.load_customtests()
except LoadError as e:
print_error_list(formatter, "customtests import: ", e)
try:
template.load_customfilters()
except LoadError as e:
print_error_list(formatter, "customfilters import: ", e)
if option.config_file:
formatter.debug("config file: " + option.config_file)
config_loader = ConfigLoader(option.config_file)
config_loader.setup_klass(action_loader.get_all_action_map())
env = {"formatter": formatter}
env.update(action_loader.get_all_action_map())
return env
def init(formatter):
action_loader.load_actions()
for error in action_loader.load_error_list:
formatter.debug("action import: " + unicode(error))
if option.config_file:
formatter.debug("config file: " + option.config_file)
config_loader = ConfigLoader(option.config_file)
config_loader.apply_config_to_klass(action_loader.get_all_action_map())
env = {"formatter": formatter}
env.update(action_loader.get_all_action_map())
return env
def main():
action_loader.load_actions()
global_env = {}
global_env.update(action_loader.get_all_action_map())
if option.config_file:
print("using config file: " + option.config_file)
config_loader = ConfigLoader(option.config_file)
config_loader.apply_config_to_klass(action_loader.get_all_action_map())
for target in option.targets:
try:
print ""
print "[running]", target
scenario = scenario_loader.load_file(target)
scenario.run(global_env)
except ActionException as e:
print "--------------------------"
print(e.stack_trace)
print(e.original)
print "--------------------------"
self.scenario.finished_action_count += 1
except ActionException as e:
print(e.original.__class__.__name__)
print(e.original)
self.scenario.finished_action_count += 1
except Exception as e:
print(e.__class__.__name__)
print(e)
sys.stdout.write("\n")
def set_param_complete(self, action_key):
if action_key in self.action_param_complete:
for param_name in self.action_param_complete[action_key]:
setattr(CUI, "do_" + param_name + ': ', lambda x=0: x)
def delete_param_complete(self, action_key):
if action_key in self.action_param_complete:
for param_name in self.action_param_complete[action_key]:
delattr(CUI, "do_" + param_name + ': ')
def _complete_action(self, *args):
return [a for a in self.action_complete if a.startswith(args[1].split()[1])]
def run(global_env):
CUI(global_env).cmdloop()
if __name__ == '__main__':
action_loader.load_actions()
run({})
