Add fix for pyvista#3224

dcbr · Dec 28, 2022 · 0782d3b · 0782d3b
1 parent 49f45ef
commit 0782d3b
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/pyvista/_vtk.py b/pyvista/_vtk.py
@@ -445,7 +445,20 @@
         vtkSplineWidget,
         vtkTexturedButtonRepresentation2D,
     )
-    from vtkmodules.vtkPythonContext2D import vtkPythonItem
+
+    try:
+        from vtkmodules.vtkPythonContext2D import vtkPythonItem
+    except ImportError:  # pragma: no cover
+        # `vtkmodules.vtkPythonContext2D` is unavailable in some versions of `vtk` (see #3224)
+        class vtkPythonItem:  # type: ignore
+            """Empty placeholder."""
+
+            def __init__(self):  # pragma: no cover
+                """Raise version error on init."""
+                from pyvista.core.errors import VTKVersionError
+
+                raise VTKVersionError('Chart backgrounds require the vtkPythonContext2D module')
+
     from vtkmodules.vtkRenderingAnnotation import (
         vtkAnnotatedCubeActor,
         vtkAxesActor,