Uploaded Test files

venv/Lib/site-packages/notebook/tree/tests/handlers.py

@@ -0,0 +1,77 @@
+"""Tornado handlers for the tree view."""
+
+# Copyright (c) Jupyter Development Team.
+# Distributed under the terms of the Modified BSD License.
+
+from tornado import web
+import os
+from ..base.handlers import IPythonHandler, path_regex
+from ..utils import url_path_join, url_escape
+
+
+class TreeHandler(IPythonHandler):
+    """Render the tree view, listing notebooks, etc."""
+
+    def generate_breadcrumbs(self, path):
+        breadcrumbs = [(url_path_join(self.base_url, 'tree'), '')]
+        parts = path.split('/')
+        for i in range(len(parts)):
+            if parts[i]:
+                link = url_path_join(self.base_url, 'tree',
+                    url_escape(url_path_join(*parts[:i+1])),
+                )
+                breadcrumbs.append((link, parts[i]))
+        return breadcrumbs
+
+    def generate_page_title(self, path):
+        comps = path.split('/')
+        if len(comps) > 3:
+            for i in range(len(comps)-2):
+                comps.pop(0)
+        page_title = url_path_join(*comps)
+        if page_title:
+            return page_title+'/'
+        else:
+            return 'Home'
+
+    @web.authenticated
+    def get(self, path=''):
+        path = path.strip('/')
+        cm = self.contents_manager
+        
+        if cm.dir_exists(path=path):
+            if cm.is_hidden(path) and not cm.allow_hidden:
+                self.log.info("Refusing to serve hidden directory, via 404 Error")
+                raise web.HTTPError(404)
+            breadcrumbs = self.generate_breadcrumbs(path)
+            page_title = self.generate_page_title(path)
+            self.write(self.render_template('tree.html',
+                page_title=page_title,
+                notebook_path=path,
+                breadcrumbs=breadcrumbs,
+                terminals_available=self.settings['terminals_available'],
+                server_root=self.settings['server_root_dir'],
+            ))
+        elif cm.file_exists(path):
+            # it's not a directory, we have redirecting to do
+            model = cm.get(path, content=False)
+            # redirect to /api/notebooks if it's a notebook, otherwise /api/files
+            service = 'notebooks' if model['type'] == 'notebook' else 'files'
+            url = url_path_join(
+                self.base_url, service, url_escape(path),
+            )
+            self.log.debug("Redirecting %s to %s", self.request.path, url)
+            self.redirect(url)
+        else:
+            raise web.HTTPError(404)
+
+
+#-----------------------------------------------------------------------------
+# URL to handler mappings
+#-----------------------------------------------------------------------------
+
+
+default_handlers = [
+    (r"/tree%s" % path_regex, TreeHandler),
+    (r"/tree", TreeHandler),
+    ]

venv/Lib/site-packages/notebook/tree/tests/test_tree_handler.py

@@ -0,0 +1,35 @@
+"""Test the /tree handlers"""
+import os
+import io
+from notebook.utils import url_path_join
+from nbformat import write
+from nbformat.v4 import new_notebook
+from urllib.parse import urlparse
+
+import requests
+
+from notebook.tests.launchnotebook import NotebookTestBase
+
+class TreeTest(NotebookTestBase):
+    def setUp(self):
+        nbdir = self.notebook_dir
+        d = os.path.join(nbdir, 'foo')
+        os.mkdir(d)
+
+        with io.open(os.path.join(d, 'bar.ipynb'), 'w', encoding='utf-8') as f:
+            nb = new_notebook()
+            write(nb, f, version=4)
+
+        with io.open(os.path.join(d, 'baz.txt'), 'w', encoding='utf-8') as f:
+            f.write(u'flamingo')
+
+        self.base_url()
+
+    def test_redirect(self):
+        r = self.request('GET', 'tree/foo/bar.ipynb')
+        self.assertEqual(r.url, self.base_url() + 'notebooks/foo/bar.ipynb')
+
+        r = self.request('GET', 'tree/foo/baz.txt', allow_redirects=False)
+        self.assertEqual(r.status_code, 302)
+        self.assertEqual(r.headers['Location'],
+             urlparse(url_path_join(self.base_url(), 'files/foo/baz.txt')).path)