make repr/display conditional on ipywidgets version

Author: vidartf

pythreejs/_base/Three.py

@@ -1,6 +1,13 @@
 from ipywidgets import Widget, widget_serialization
 from traitlets import Unicode
 
+try:
+    from importlib.metadata import version
+except ImportError:
+    def version(pkg):
+        import pkg_resources
+        return pkg_resources.get_distribution(pkg).version
+
 from .._package import npm_pkg_name
 from .._version import EXTENSION_SPEC_VERSION
 
@@ -39,28 +46,29 @@ def _on_ret_val(self, method_name, ret_val):
         """Message callback used internally for logging exec returns"""
         self.log.info('%s() -> %s' % (method_name, ret_val))
 
-    def _repr_mimebundle_(self, **kwargs):
-        if self._previewable:
-            from .renderable import Preview
-            plaintext = repr(self)
-            if len(plaintext) > 110:
-                plaintext = plaintext[:110] + '…'
-            preview = Preview(self)
-            return {
-                'text/plain': plaintext,
-                'application/vnd.jupyter.widget-view+json': {
-                    "version_major": 2,
-                    "version_minor": 0,
-                    "model_id": preview._model_id
+    if version("ipywidget") >= (8, 0, 0):
+        def _repr_mimebundle_(self, **kwargs):
+            if self._previewable:
+                from .renderable import Preview
+                plaintext = repr(self)
+                if len(plaintext) > 110:
+                    plaintext = plaintext[:110] + '…'
+                preview = Preview(self)
+                return {
+                    'text/plain': plaintext,
+                    'application/vnd.jupyter.widget-view+json': {
+                        "version_major": 2,
+                        "version_minor": 0,
+                        "model_id": preview._model_id
+                    }
                 }
-            }
-        else:
-            return super(ThreeWidget, self)._repr_mimebundle_(**kwargs)
-
-    def _ipython_display_(self, **kwargs):
-        if self._previewable:
-            from IPython.display import display
-            from .renderable import Preview
-            return display(Preview(self), **kwargs)
-        else:
-            return super(ThreeWidget, self)._ipython_display_(**kwargs)
+            else:
+                return super(ThreeWidget, self)._repr_mimebundle_(**kwargs)
+    else:
+        def _ipython_display_(self, **kwargs):
+            if self._previewable:
+                from IPython.display import display
+                from .renderable import Preview
+                return display(Preview(self), **kwargs)
+            else:
+                return super(ThreeWidget, self)._ipython_display_(**kwargs)