fix replacing material with material profile

myoualid · myoualid · commit a8f1a28ac258 · 2023-06-13T14:04:28.000+01:00
diff --git a/src/blenderbim/blenderbim/tool/material.py b/src/blenderbim/blenderbim/tool/material.py
@@ -212,11 +212,14 @@ def is_a_flow_segment(cls, element):
     def replace_material_with_material_profile(cls, element):
         old_material = cls.get_material(element, should_inherit=False)
         old_inherited_material = cls.get_material(element, should_inherit=True)
-        blenderbim.core.material.unassign_material(tool.Ifc, tool.Material, objects=[tool.Ifc.get_object(element)])
-        tool.Ifc.run("material.assign_material", product=element, type="IfcMaterialProfileSet")
-        assinged_material = cls.get_material(element)
         material = old_material if old_material and old_material.is_a("IfcMaterial") else None
         if not material and old_inherited_material:
             material = old_inherited_material if old_inherited_material.is_a("IfcMaterial") else None
+        if not material:
+            material = tool.Ifc.get().by_type("IfcMaterial")[0]
+        else:
+            blenderbim.core.material.unassign_material(tool.Ifc, tool.Material, objects=[tool.Ifc.get_object(element)])
+        tool.Ifc.run("material.assign_material", product=element, type="IfcMaterialProfileSet", material=material)
+        assinged_material = cls.get_material(element)
         material_profile = tool.Ifc.run("material.add_profile", profile_set=assinged_material, material=material)
         return material_profile
