provide method to cast labware defs and types to typeddict

shared-data/python/opentrons_shared_data/labware/types.py

@@ -5,7 +5,7 @@
 """
 from typing import Dict, List, NewType, Union
 from typing_extensions import Literal, TypedDict, NotRequired
-
+from .labware_definition import WellSegment as WellSegmentDef
 
 LabwareUri = NewType("LabwareUri", str)
 
@@ -176,10 +176,21 @@ class RoundedRectangularSegment(TypedDict):
     SphericalSegment,
 ]
 
+@staticmethod
+def ToWellSegmentDict(segment: Union[WellSegment,WellSegmentDef]) -> WellSegment:
+    if isinstance(segment, WellSegmentDef):
+        return typing.cast(WellSegment, segment.model_dump(exclude_none=True, exclude_unset=True))
+    return segment
+
 
 class InnerWellGeometry(TypedDict):
     sections: List[WellSegment]
 
+@staticmethod
+def ToInnerWellGeometryDict(inner_well_geometry: Union[InnerWellGeometry,InnerWellGeometryDef]) -> InnerWellGeometry:
+    return InnerWellGeometry([ToWellSegmentDict(section) for section in inner_well_geometry["sections"]])
+
+
 
 class LabwareDefinition(TypedDict):
     schemaVersion: Literal[2]