Merge pull request #21 from yacchin1205/fix/base-dir

Fix base directory

nbsearch/server.py

@@ -19,6 +19,12 @@
 ]
 
 
+def _get_root_dir(nb_server_app):
+    if hasattr(nb_server_app, 'root_dir'):
+        return nb_server_app.root_dir
+    return nb_server_app.notebook_dir
+
+
 def get_api_handlers(parent_app, base_dir):
     db = NBSearchDB(parent=parent_app)
 
@@ -34,9 +40,11 @@ def get_api_handlers(parent_app, base_dir):
 
 def register_routes(nb_server_app, web_app):
     from notebook.utils import url_path_join
-    api_handlers = get_api_handlers(nb_server_app, nb_server_app.notebook_dir)
+    base_dir = _get_root_dir(nb_server_app)
+    nb_server_app.log.info(f'nbsearch extension: base_dir={base_dir}')
+    api_handlers = get_api_handlers(nb_server_app, base_dir)
 
-    nbsearchignore = os.path.join(nb_server_app.notebook_dir, '.nbsearchignore')
+    nbsearchignore = os.path.join(base_dir, '.nbsearchignore')
     if not os.path.exists(nbsearchignore):
         with open(nbsearchignore, 'w') as f:
             f.write(f'''# Generated by nbsearch