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())

---

本文转自疯吻IT博客园博客，原文链接：http://www.cnblogs.com/fengwenit/p/6422381.html，如需转载请自行联系原作者