def test_broken_register_catalog_hook(self, tmp_path):
pattern = "Expected an instance of `DataCatalog`, got `NoneType` instead."
with KedroSession.create(MOCK_PACKAGE_NAME, tmp_path) as session:
context = session.load_context()
with pytest.raises(KedroContextError, match=re.escape(pattern)):
_ = context.catalog
def test_broken_register_config_loader_hook(self, tmp_path):
pattern = "Expected an instance of `ConfigLoader`, got `NoneType` instead."
with pytest.raises(KedroContextError, match=re.escape(pattern)):
KedroSession.create(MOCK_PACKAGE_NAME, tmp_path)
def reload_kedro(path,
line=None,
env: str = None,
extra_params: Dict[str, Any] = None):
"""Line magic which reloads all Kedro default variables."""
global startup_error
global context
global catalog
global session
try:
import kedro.config.default_logger
from kedro.framework.hooks import get_hook_manager
from kedro.framework.session import KedroSession
from kedro.framework.session.session import _activate_session
from kedro.framework.cli.jupyter import collect_line_magic
except ImportError:
logging.error(
"Kedro appears not to be installed in your current environment "
"or your current IPython session was not started in a valid Kedro project."
)
raise
try:
path = path or project_path
# clear hook manager
hook_manager = get_hook_manager()
name_plugin_pairs = hook_manager.list_name_plugin()
for name, plugin in name_plugin_pairs:
hook_manager.unregister(name=name, plugin=plugin)
# remove cached user modules
metadata = _get_project_metadata(path)
to_remove = [
mod for mod in sys.modules if mod.startswith(metadata.package_name)
]
# `del` is used instead of `reload()` because: If the new version of a module does not
# define a name that was defined by the old version, the old definition remains.
for module in to_remove:
del sys.modules[module]
session = KedroSession.create(metadata.package_name,
path,
env=env,
extra_params=extra_params)
_activate_session(session, force=True)
logging.debug("Loading the context from %s", str(path))
context = session.load_context()
catalog = context.catalog
logging.info("** Kedro project %s", str(metadata.project_name))
logging.info(
"Defined global variable `context`, `session` and `catalog`")
for line_magic in collect_line_magic():
register_line_magic(needs_local_scope(line_magic))
logging.info("Registered line magic `%s`", line_magic.__name__)
except Exception as err:
startup_error = err
logging.exception("Kedro's ipython session startup script failed:\n%s",
str(err))
raise err
def mock_session_with_hooks(tmp_path, mock_settings_import, logging_hooks): # pylint: disable=unused-argument
logging_hooks.queue_listener.start()
yield KedroSession.create(MOCK_PACKAGE_NAME, tmp_path)
logging_hooks.queue_listener.stop()
def run_package():
# Entry point for running a Kedro project packaged with `kedro package`
# using `python -m <project_package>.run` command.
package_name = Path(__file__).resolve().parent.name
with KedroSession.create(package_name) as session:
session.run()