module_loader.py

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92

# few functions that make it possible to import functions # from jupyter notebooks as from modules; # source: http://jupyter-notebook.readthedocs.io/en/latest/examples/Notebook/Importing%20Notebooks.html   import io, os, sys, types   from IPython import get_ipython from nbformat import read from IPython.core.interactiveshell import InteractiveShell   def find_notebook(fullname, path=None):     """find a notebook, given its fully qualified name and an optional path     This turns "foo.bar" into "foo/bar.ipynb"     and tries turning "Foo_Bar" into "Foo Bar" if Foo_Bar     does not exist.     """     name = fullname.rsplit('.', 1)[-1]     if not path:         path = ['']     for d in path:         nb_path = os.path.join(d, name + ".ipynb")         if os.path.isfile(nb_path):             return nb_path         # let import Notebook_Name find "Notebook Name.ipynb"         nb_path = nb_path.replace("_", " ")         if os.path.isfile(nb_path):             return nb_path   class NotebookLoader(object):     """Module Loader for Jupyter Notebooks"""     def __init__(self, path=None):         self.shell = InteractiveShell.instance()         self.path = path       def load_module(self, fullname):         """import a notebook as a module"""         path = find_notebook(fullname, self.path)           #print ("importing Jupyter notebook from %s" % path)           # load the notebook object         with io.open(path, 'r', encoding='utf-8') as f:             nb = read(f, 4)             # create the module and add it to sys.modules         # if name in sys.modules:         #    return sys.modules[name]         mod = types.ModuleType(fullname)         mod.__file__ = path         mod.__loader__ = self         mod.__dict__['get_ipython'] = get_ipython         sys.modules[fullname] = mod           # extra work to ensure that magics that would affect the user_ns         # actually affect the notebook module's ns         save_user_ns = self.shell.user_ns         self.shell.user_ns = mod.__dict__           try:             for cell in nb.cells:                 if cell.cell_type == 'code':                     # transform the input to executable Python                     code = self.shell.input_transformer_manager.transform_cell(cell.source)                     # run the code in themodule                     exec(code, mod.__dict__)         finally:             self.shell.user_ns = save_user_ns         return mod   class NotebookFinder(object):     """Module finder that locates Jupyter Notebooks"""     def __init__(self):         self.loaders = {}       def find_module(self, fullname, path=None):         nb_path = find_notebook(fullname, path)         if not nb_path:             return           key = path         if path:             # lists aren't hashable             key = os.path.sep.join(path)           if key not in self.loaders:             self.loaders[key] = NotebookLoader(path)         return self.loaders[key]       sys.meta_path.append(NotebookFinder())

---

如果您觉得阅读本文对您有帮助，请点一下“推荐”按钮，您的“推荐”将是我最大的写作动力！欢迎各位转载，但必须在文章页面明显位置给出作者和原文连接，否则保留追究法律责任的权利。