"""find a notebook, given its fully qualified name and an optional path"""

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

"""Module Loader for IPython 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)

disp = "importing IPython notebook from "

disp += "<a href='./{}' target='_blank'>{}</a>".format(path, path[:-6])

display_html(disp, raw=True)

#print disp

# load the notebook object

with io.open(path, 'r', encoding='utf-8') as f:

nb = current.read(f, 'json')

# 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

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.worksheets[0].cells:

if cell.cell_type == 'code' and cell.language == 'python':

# transform the input to executable Python

code = self.shell.input_transformer_manager.transform_cell(cell.input)

# run the code in themodule

exec code in mod.__dict__

finally:

self.shell.user_ns = save_user_ns

"""Module finder that locates IPython 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)