def _setup_hooks(self, package, global_context, mods): if package.use_backends: if len(package.use_backends) > 1: raise ValueError("Only up to one backend supported for now") else: assert global_context.backend is None global_context.backend = load_backend( package.use_backends[0])() startup_hooks = find_startup_hooks(mods) option_hooks = find_options_hooks(mods) shutdown_hooks = find_shutdown_hooks(mods) if startup_hooks: # FIXME: there should be an error or a warning if startup defined in # mods beyond the first one startup_hooks[0](global_context) if global_context.backend: global_context.backend.register_command_contexts(global_context) for command in find_command_hooks(mods): global_context.register_command(command.name, command) if global_context.backend: global_context.backend.register_options_contexts(global_context) if option_hooks: # FIXME: there should be an error or a warning if shutdown defined in # mods beyond the first one option_hooks[0](global_context) # FIXME: this registered options for new commands registered in hook. It # should be made all in one place (hook and non-hook) for cmd_name in global_context.command_names(public_only=False): if not global_context.is_options_context_registered(cmd_name): # FIXME: this should be supported in global context directly # (redundant with bentomakerlib) cmd = global_context.retrieve_command(cmd_name) context = OptionsContext.from_command(cmd) if not global_context.is_options_context_registered(cmd_name): global_context.register_options_context(cmd_name, context) for cmd_name in global_context.command_names(): for hook in find_pre_hooks(mods, cmd_name): global_context.add_pre_hook(hook, cmd_name) for hook in find_post_hooks(mods, cmd_name): global_context.add_post_hook(hook, cmd_name)
def _setup_hooks(self, package, global_context, mods): if package.use_backends: if len(package.use_backends) > 1: raise ValueError("Only up to one backend supported for now") else: assert global_context.backend is None global_context.backend = load_backend(package.use_backends[0])() startup_hooks = find_startup_hooks(mods) option_hooks = find_options_hooks(mods) shutdown_hooks = find_shutdown_hooks(mods) if startup_hooks: # FIXME: there should be an error or a warning if startup defined in # mods beyond the first one startup_hooks[0](global_context) if global_context.backend: global_context.backend.register_command_contexts(global_context) for command in find_command_hooks(mods): global_context.register_command(command.name, command) if global_context.backend: global_context.backend.register_options_contexts(global_context) if option_hooks: # FIXME: there should be an error or a warning if shutdown defined in # mods beyond the first one option_hooks[0](global_context) # FIXME: this registered options for new commands registered in hook. It # should be made all in one place (hook and non-hook) for cmd_name in global_context.command_names(public_only=False): if not global_context.is_options_context_registered(cmd_name): # FIXME: this should be supported in global context directly # (redundant with bentomakerlib) cmd = global_context.retrieve_command(cmd_name) context = OptionsContext.from_command(cmd) if not global_context.is_options_context_registered(cmd_name): global_context.register_options_context(cmd_name, context) for cmd_name in global_context.command_names(): for hook in find_pre_hooks(mods, cmd_name): global_context.add_pre_hook(hook, cmd_name) for hook in find_post_hooks(mods, cmd_name): global_context.add_post_hook(hook, cmd_name)
def _wrapped_main(global_context, popts, run_node, top_node, build_node): # Some commands work without a bento description file (convert, help) # FIXME: this should not be called here then - clearly separate commands # which require bento.info from the ones who do not bento_info_node = top_node.find_node(BENTO_SCRIPT) if bento_info_node is not None: db_node = build_node.make_node(DB_FILE) cached_package = CachedPackage(db_node) package = cached_package.get_package(bento_info_node) package_options = cached_package.get_options(bento_info_node) if package.use_backends: if len(package.use_backends) > 1: raise ValueError("Only up to one backend supported for now") else: assert global_context.backend is None global_context.backend = load_backend( package.use_backends[0])() global_context.register_package_options(package_options) mods = set_main(package, top_node, build_node) else: warnings.warn("No %r file in current directory - only generic options " "will be displayed" % BENTO_SCRIPT) cached_package = None package_options = None mods = [] startup_hooks = find_startup_hooks(mods) option_hooks = find_options_hooks(mods) shutdown_hooks = find_shutdown_hooks(mods) if startup_hooks: # FIXME: there should be an error or a warning if startup defined in # mods beyond the first one startup_hooks[0](global_context) if global_context.backend: global_context.backend.register_command_contexts(global_context) for command in find_command_hooks(mods): global_context.register_command(command.name, command) register_stuff(global_context) for cmd_name in global_context.command_names(): register_options(global_context, cmd_name) if global_context.backend: global_context.backend.register_options_contexts(global_context) if option_hooks: # FIXME: there should be an error or a warning if shutdown defined in # mods beyond the first one option_hooks[0](global_context) # FIXME: this registered options for new commands registered in hook. It # should be made all in one place (hook and non-hook) for cmd_name in global_context.command_names(public_only=False): if not global_context.is_options_context_registered(cmd_name): register_options(global_context, cmd_name) for cmd_name in global_context.command_names(): for hook in find_pre_hooks(mods, cmd_name): global_context.add_pre_hook(hook, cmd_name) for hook in find_post_hooks(mods, cmd_name): global_context.add_post_hook(hook, cmd_name) try: return _main(global_context, cached_package, popts, run_node, top_node, build_node) finally: if shutdown_hooks: shutdown_hooks[0](global_context)
def _wrapped_main(global_context, popts, run_node, top_node, build_node): # Some commands work without a bento description file (convert, help) # FIXME: this should not be called here then - clearly separate commands # which require bento.info from the ones who do not bento_info_node = top_node.find_node(BENTO_SCRIPT) if bento_info_node is not None: db_node = build_node.make_node(DB_FILE) cached_package = CachedPackage(db_node) package = cached_package.get_package(bento_info_node) package_options = cached_package.get_options(bento_info_node) if package.use_backends: if len(package.use_backends) > 1: raise ValueError("Only up to one backend supported for now") else: assert global_context.backend is None global_context.backend = load_backend(package.use_backends[0])() global_context.register_package_options(package_options) mods = set_main(package, top_node, build_node) else: warnings.warn("No %r file in current directory - only generic options " "will be displayed" % BENTO_SCRIPT) cached_package = None package_options = None mods = [] startup_hooks = find_startup_hooks(mods) option_hooks = find_options_hooks(mods) shutdown_hooks = find_shutdown_hooks(mods) if startup_hooks: # FIXME: there should be an error or a warning if startup defined in # mods beyond the first one startup_hooks[0](global_context) if global_context.backend: global_context.backend.register_command_contexts(global_context) for command in find_command_hooks(mods): global_context.register_command(command.name, command) register_stuff(global_context) for cmd_name in global_context.command_names(): register_options(global_context, cmd_name) if global_context.backend: global_context.backend.register_options_contexts(global_context) if option_hooks: # FIXME: there should be an error or a warning if shutdown defined in # mods beyond the first one option_hooks[0](global_context) # FIXME: this registered options for new commands registered in hook. It # should be made all in one place (hook and non-hook) for cmd_name in global_context.command_names(public_only=False): if not global_context.is_options_context_registered(cmd_name): register_options(global_context, cmd_name) for cmd_name in global_context.command_names(): for hook in find_pre_hooks(mods, cmd_name): global_context.add_pre_hook(hook, cmd_name) for hook in find_post_hooks(mods, cmd_name): global_context.add_post_hook(hook, cmd_name) try: return _main(global_context, cached_package, popts, run_node, top_node, build_node) finally: if shutdown_hooks: shutdown_hooks[0](global_context)