use vtkmodules import instead of vtk (#918)

Author: darrellenns

cadquery/occ_impl/assembly.py

@@ -10,7 +10,11 @@
 from OCP.TopLoc import TopLoc_Location
 from OCP.Quantity import Quantity_ColorRGBA
 
-from vtk import vtkActor, vtkPolyDataMapper as vtkMapper, vtkRenderer
+from vtkmodules.vtkRenderingCore import (
+    vtkActor,
+    vtkPolyDataMapper as vtkMapper,
+    vtkRenderer,
+)
 
 from .geom import Location
 from .shapes import Shape, Compound

cadquery/occ_impl/exporters/assembly.py

@@ -3,7 +3,8 @@
 from tempfile import TemporaryDirectory
 from shutil import make_archive
 
-from vtk import vtkJSONSceneExporter, vtkRenderer, vtkRenderWindow, vtkVRMLExporter
+from vtkmodules.vtkIOExport import vtkJSONSceneExporter, vtkVRMLExporter
+from vtkmodules.vtkRenderingCore import vtkRenderer, vtkRenderWindow
 
 from OCP.XSControl import XSControl_WorkSession
 from OCP.STEPCAFControl import STEPCAFControl_Writer

cadquery/occ_impl/exporters/vtk.py

@@ -1,4 +1,4 @@
-from vtk import vtkXMLPolyDataWriter
+from vtkmodules.vtkIOXML import vtkXMLPolyDataWriter
 from ..shapes import Shape
 
 

cadquery/occ_impl/shapes.py

@@ -17,11 +17,8 @@
 
 from io import BytesIO
 
-from vtk import (
-    vtkPolyData,
-    vtkTriangleFilter,
-    vtkPolyDataNormals,
-)
+from vtkmodules.vtkCommonDataModel import vtkPolyData
+from vtkmodules.vtkFiltersCore import vtkTriangleFilter, vtkPolyDataNormals
 
 from .geom import Vector, BoundBox, Plane, Location, Matrix
 

mypy.ini

@@ -22,7 +22,7 @@ ignore_missing_imports = True
 [mypy-nlopt.*]
 ignore_missing_imports = True
 
-[mypy-vtk.*]
+[mypy-vtkmodules.*]
 ignore_missing_imports = True
 
 [mypy-docutils.*]