typing

src/bonsai/bonsai/bim/module/profile/ui.py

@@ -48,6 +48,7 @@ class BIM_PT_profiles(Panel):
         self.props = tool.Profile.get_profile_props()
 
         active_profile = None
+        assert self.layout
         if self.props.is_editing and (active_profile := tool.Profile.get_active_profile_ui()):
             preview_collection = ProfileData.preview_collection
             box = self.layout.box()

src/bonsai/bonsai/bim/module/style/operator.py

@@ -25,6 +25,7 @@ import ifcopenshell.api
 import ifcopenshell.api.style
 import ifcopenshell.util.representation
 import ifcopenshell.util.unit
+from bonsai.bim.module.style.prop import ColourRgb
 from bpy_extras.io_utils import ImportHelper
 from pathlib import Path
 from mathutils import Vector
@@ -739,9 +740,10 @@ class EnableEditingSurfaceStyle(bpy.types.Operator):
         callback = None
         if self.ifc_class == "IfcSurfaceStyleLighting":
 
-            def callback(attribute_name, _, data):
+            def callback(attribute_name: str, _: object, data: dict[str, Any]) -> None:
                 assert attributes
                 color = attributes.add()
+                assert isinstance(color, ColourRgb)
                 color.name = attribute_name
                 color_value = data[attribute_name]
                 if color_value is None:

src/bonsai/bonsai/bim/module/style/prop.py

@@ -78,6 +78,7 @@ class Style(PropertyGroup):
     )
 
     if TYPE_CHECKING:
+        name: str
         ifc_definition_id: int
         total_elements: int
         style_classes: bpy.types.bpy_prop_collection_idprop[StrProperty]

src/bonsai/bonsai/bim/module/style/ui.py

@@ -46,6 +46,7 @@ class BIM_PT_styles(Panel):
             StylesData.load()
 
         self.props = tool.Style.get_style_props()
+        assert self.layout
 
         if not self.props.is_editing:
             row = self.layout.row(align=True)

src/bonsai/bonsai/bim/operator.py

@@ -909,7 +909,7 @@ class FetchObjectPassport(bpy.types.Operator):
         context.active_object.data = bpy.data.meshes[reference.name]
 
 
-def update_enum_property_search_prop(self, context):
+def update_enum_property_search_prop(self: "BIM_OT_enum_property_search", context: bpy.types.Context) -> None:
     for i, prop in enumerate(self.collection_names):
         if prop.name == self.dummy_name:
             setattr(context.data, self.prop_name, self.collection_identifiers[i].name)
@@ -933,6 +933,7 @@ class BIM_OT_enum_property_search(bpy.types.Operator):
     bl_label = "Search"
     bl_description = "Search For Property"
     bl_options = {"REGISTER", "UNDO"}
+
     first_launch: bpy.props.BoolProperty(default=True, options={"SKIP_SAVE"})
     dummy_name: bpy.props.StringProperty(name="Property", update=update_enum_property_search_prop)
     collection_names: bpy.props.CollectionProperty(type=StrProperty)
@@ -1334,6 +1335,7 @@ class BIM_OT_attribute_search_values(bpy.types.Operator):
     bl_label = "Search Attribute Values"
     bl_description = "Search for attribute values within a collection"
     bl_options = {"REGISTER", "UNDO"}
+
     first_launch: bpy.props.BoolProperty(default=True, options={"SKIP_SAVE"})
     attribute_name: bpy.props.StringProperty(name="Attribute Name")
     attribute_ifc_class: bpy.props.StringProperty(name="Attribute IFC Class")
@@ -1352,7 +1354,13 @@ class BIM_OT_attribute_search_values(bpy.types.Operator):
     collection_values: bpy.props.CollectionProperty(type=StrProperty, options={"SKIP_SAVE"})
 
     if TYPE_CHECKING:
+        first_launch: bool
+        attribute_name: str
+        attribute_ifc_class: str
+        data_path: str
         data_type: AttributeSearchDataType
+        search_value: str
+        collection_values: bpy.types.bpy_prop_collection_idprop[StrProperty]
 
     @staticmethod
     def resolve_data_path(data_path: str) -> tuple[str, "Attribute"]:
@@ -1406,7 +1414,7 @@ class BIM_OT_attribute_search_values(bpy.types.Operator):
             results_are_suggestions=True,
         )
 
-    def execute(self, context):
+    def execute(self, context) -> "set[rna_enums.OperatorReturnItems]":
         return {"FINISHED"}
 
 
@@ -1416,7 +1424,7 @@ class BIM_OT_attribute_add_subitem(bpy.types.Operator):
     bl_description = "Add subitem to the current attribute"
     bl_options = {"REGISTER", "UNDO"}
 
-    data_path: bpy.props.StringProperty()
+    data_path: bpy.props.StringProperty()  # pyright: ignore[reportRedeclaration]
     """Full data path."""
 
     if TYPE_CHECKING:
@@ -1435,9 +1443,9 @@ class BIM_OT_attribute_remove_subitem(bpy.types.Operator):
     bl_description = "Add subitem to the current attribute"
     bl_options = {"REGISTER", "UNDO"}
 
-    data_path: bpy.props.StringProperty()
+    data_path: bpy.props.StringProperty()  # pyright: ignore[reportRedeclaration]
     """Full data path."""
-    index: bpy.props.IntProperty()
+    index: bpy.props.IntProperty()  # pyright: ignore[reportRedeclaration]
 
     if TYPE_CHECKING:
         data_path: str

src/bonsai/bonsai/bim/prop.py

@@ -621,6 +621,7 @@ class PsetQto(PropertyGroup):
     is_editable: BoolProperty(name="Is Editable")
 
     if TYPE_CHECKING:
+        name: str
         properties: bpy.types.bpy_prop_collection_idprop[Attribute]
         is_expanded: bool
         is_editable: bool