Update hybridmanager.py

This is the very same example as for danielfrg/s3contents#163
viaduct-ai · May 4, 2023 · 4727348 · 4727348
1 parent 789403f
commit 4727348
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/hybridcontents/hybridmanager.py b/hybridcontents/hybridmanager.py
@@ -4,7 +4,17 @@
 from six import iteritems
 from tornado.web import HTTPError
 
-from notebook.services.contents.manager import ContentsManager
+# Ref. https://github.com/jupyter/nbdime/blob/c82362344e596efdc4f54c927d90338940e0fa41/nbdime/webapp/nb_server_extension.py#L16-L33
+# This allows solving conflicts between the class import from either notebook or jupyter_server
+try:
+    from notebook.services.contents.manager import ContentsManager
+except ModuleNotFoundError:
+    pass
+
+try:
+    from jupyter_server.services.contents.manager import ContentsManager
+except ModuleNotFoundError:
+    pass
 
 from traitlets import Dict