Uploaded Test files

venv/Lib/site-packages/notebook/bundler/handlers.py

@@ -0,0 +1,88 @@
+"""Tornado handler for bundling notebooks."""
+
+# Copyright (c) Jupyter Development Team.
+# Distributed under the terms of the Modified BSD License.
+
+from ipython_genutils.importstring import import_item
+from tornado import web, gen
+
+from notebook.utils import maybe_future, url2path
+from notebook.base.handlers import IPythonHandler
+from notebook.services.config import ConfigManager
+
+from . import tools
+
+
+class BundlerHandler(IPythonHandler):
+    def initialize(self):
+        """Make tools module available on the handler instance for compatibility
+        with existing bundler API and ease of reference."""
+        self.tools = tools
+    
+    def get_bundler(self, bundler_id):
+        """
+        Get bundler metadata from config given a bundler ID.
+        
+        Parameters
+        ----------
+        bundler_id: str
+            Unique bundler ID within the notebook/bundlerextensions config section
+        
+        Returns
+        -------
+        dict
+            Bundler metadata with label, group, and module_name attributes
+        
+        
+        Raises
+        ------
+        KeyError
+            If the bundler ID is unknown
+        """
+        cm = ConfigManager()
+        return cm.get('notebook').get('bundlerextensions', {})[bundler_id]
+
+    @web.authenticated
+    @gen.coroutine
+    def get(self, path):
+        """Bundle the given notebook.
+        
+        Parameters
+        ----------
+        path: str
+            Path to the notebook (path parameter)
+        bundler: str
+            Bundler ID to use (query parameter)
+        """
+        bundler_id = self.get_query_argument('bundler')
+        model = self.contents_manager.get(path=url2path(path))
+
+        try:
+            bundler = self.get_bundler(bundler_id)
+        except KeyError as e:
+            raise web.HTTPError(400, 'Bundler %s not enabled' %
+                                bundler_id) from e
+
+        module_name = bundler['module_name']
+        try:
+            # no-op in python3, decode error in python2
+            module_name = str(module_name)
+        except UnicodeEncodeError:
+            # Encode unicode as utf-8 in python2 else import_item fails
+            module_name = module_name.encode('utf-8')
+        
+        try:
+            bundler_mod = import_item(module_name)
+        except ImportError as e:
+            raise web.HTTPError(500, 'Could not import bundler %s ' %
+                                bundler_id) from e
+
+        # Let the bundler respond in any way it sees fit and assume it will
+        # finish the request
+        yield maybe_future(bundler_mod.bundle(self, model))
+
+_bundler_id_regex = r'(?P<bundler_id>[A-Za-z0-9_]+)'
+
+default_handlers = [
+    (r"/bundle/(.*)", BundlerHandler)
+]