typing

src/blenderbim/blenderbim/bim/module/pset/data.py

@@ -19,6 +19,7 @@
 import bpy
 import ifcopenshell
 import ifcopenshell.util.doc
+import ifcopenshell.util.element
 import blenderbim.tool as tool
 import blenderbim.bim.schema
 

src/blenderbim/blenderbim/bim/module/pset/prop.py

@@ -20,6 +20,7 @@ import bpy
 import blenderbim.bim.schema
 import ifcopenshell
 import ifcopenshell.util.attribute
+import ifcopenshell.util.doc
 import ifcopenshell.util.element
 import blenderbim.tool as tool
 from blenderbim.bim.prop import Attribute, StrProperty

src/blenderbim/blenderbim/core/qto.py

@@ -16,19 +16,28 @@
 # You should have received a copy of the GNU General Public License
 # along with BlenderBIM Add-on.  If not, see <http://www.gnu.org/licenses/>.
 
+from __future__ import annotations
+from typing import TYPE_CHECKING, Optional
 
-def calculate_circle_radius(qto, obj=None):
+if TYPE_CHECKING:
+    import bpy
+    import ifcopenshell
+    import blenderbim.tool as tool
+    from blenderbim.bim.module.pset.qto_calculator import QtoCalculator
+
+
+def calculate_circle_radius(qto: tool.Qto, obj: bpy.types.Object) -> float:
     result = qto.get_radius_of_selected_vertices(obj)
     qto.set_qto_result(result)
     return result
 
 
-def assign_objects_base_qto(ifc, qto, selected_objects):
+def assign_objects_base_qto(ifc: tool.Ifc, qto: tool.Qto, selected_objects: list[bpy.types.Object]) -> None:
     for obj in selected_objects:
         assign_object_base_qto(ifc, qto, obj)
 
 
-def assign_object_base_qto(ifc, qto, obj):
+def assign_object_base_qto(ifc: tool.Ifc, qto: tool.Qto, obj: bpy.types.Object) -> None:
     product = ifc.get_entity(obj)
     if not product:
         return
@@ -42,13 +51,17 @@ def assign_object_base_qto(ifc, qto, obj):
     )
 
 
-def calculate_all_quantities(ifc, cost, qto, calculator, selected_objects):
+def calculate_all_quantities(
+    ifc: tool.Ifc, cost: tool.Cost, qto: tool.Qto, calculator: QtoCalculator, selected_objects: list[bpy.types.Object]
+) -> None:
     if selected_objects:
         for obj in selected_objects:
             calculate_object_base_quantities(ifc, cost, qto, calculator, obj)
 
 
-def calculate_object_base_quantities(ifc, cost, qto, calculator, obj):
+def calculate_object_base_quantities(
+    ifc: tool.Ifc, cost: tool.Cost, qto: tool.Qto, calculator: QtoCalculator, obj: bpy.types.Object
+) -> None:
     product = ifc.get_entity(obj)
     if not product:
         return

src/blenderbim/test/tool/test_qto.py

@@ -19,6 +19,7 @@
 import bpy
 import ifcopenshell
 import ifcopenshell.api
+import ifcopenshell.util.pset
 import test.bim.bootstrap
 import blenderbim.core.tool
 import blenderbim.core.root