Fix #1401. Modified custom psets are now immediately available without needing to restart Blender.

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

@@ -195,6 +195,7 @@ class EditPset(bpy.types.Operator):
                     "pset": self.file.by_id(pset_id),
                     "name": props.active_pset_name,
                     "properties": properties,
+                    "pset_template": blenderbim.bim.schema.ifc.psetqto.get_by_name(props.active_pset_name),
                 },
             )
         else:

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

@@ -1,7 +1,8 @@
 import bpy
+import blenderbim.bim.schema
 from blenderbim.bim.prop import Attribute
 from ifcopenshell.api.pset.data import Data
-import blenderbim.bim.schema
+from blenderbim.bim.ifc import IfcStore
 from bpy.types import PropertyGroup
 from bpy.props import (
     PointerProperty,
@@ -29,16 +30,17 @@ def purge():
 def getPsetNames(self, context):
     global psetnames
     obj = context.active_object
-    if "/" in obj.name:
-        ifc_class = obj.name.split("/")[0]
-        if ifc_class not in psetnames:
-            psets = blenderbim.bim.schema.ifc.psetqto.get_applicable(ifc_class, pset_only=True)
-            psetnames[ifc_class] = [(p.Name, p.Name, "") for p in psets]
-        assigned_names = [
-            Data.psets[p]["Name"] for p in Data.products[obj.BIMObjectProperties.ifc_definition_id]["psets"]
-        ]
-        return [p for p in psetnames[ifc_class] if p[0] not in assigned_names]
-    return []
+    if not obj.BIMObjectProperties.ifc_definition_id:
+        return []
+    element = IfcStore.get_file().by_id(obj.BIMObjectProperties.ifc_definition_id)
+    ifc_class = element.is_a()
+    if ifc_class not in psetnames:
+        psets = blenderbim.bim.schema.ifc.psetqto.get_applicable(ifc_class, pset_only=True)
+        psetnames[ifc_class] = [(p.Name, p.Name, "") for p in psets]
+    assigned_names = [
+        Data.psets[p]["Name"] for p in Data.products[obj.BIMObjectProperties.ifc_definition_id]["psets"]
+    ]
+    return [p for p in psetnames[ifc_class] if p[0] not in assigned_names]
 
 
 def getMaterialPsetNames(self, context):

src/blenderbim/blenderbim/bim/module/pset_template/operator.py

@@ -1,6 +1,8 @@
 import bpy
 import ifcopenshell
 import ifcopenshell.api
+import blenderbim.bim.schema
+import blenderbim.bim.handler
 from blenderbim.bim.module.pset_template.prop import updatePsetTemplateFiles, updatePsetTemplates
 from ifcopenshell.api.pset_template.data import Data
 from blenderbim.bim.ifc import IfcStore
@@ -165,6 +167,8 @@ class SavePsetTemplateFile(bpy.types.Operator):
 
     def execute(self, context):
         IfcStore.pset_template_file.write(IfcStore.pset_template_path)
+        blenderbim.bim.handler.purge_module_data()
+        blenderbim.bim.schema.reload()
         return {"FINISHED"}
 
 

src/blenderbim/blenderbim/bim/schema.py

@@ -77,3 +77,7 @@ class IfcSchema:
 
 
 ifc = IfcSchema()
+
+def reload():
+    global ifc
+    ifc = IfcSchema()

src/ifcopenshell-python/ifcopenshell/api/pset/edit_pset.py

@@ -5,7 +5,7 @@ import ifcopenshell.util.pset
 class Usecase:
     def __init__(self, file, **settings):
         self.file = file
-        self.settings = {"pset": None, "name": None, "properties": {}}
+        self.settings = {"pset": None, "name": None, "properties": {}, "pset_template": None}
         for key, value in settings.items():
             self.settings[key] = value
 
@@ -21,9 +21,12 @@ class Usecase:
             self.settings["pset"].Name = self.settings["name"]
 
     def load_pset_template(self):
-        # TODO: add IFC2X3 PsetQto template support
-        self.psetqto = ifcopenshell.util.pset.get_template("IFC4")
-        self.pset_template = self.psetqto.get_by_name(self.settings["pset"].Name)
+        if self.settings["pset_template"]:
+            self.pset_template = self.settings["pset_template"]
+        else:
+            # TODO: add IFC2X3 PsetQto template support
+            self.psetqto = ifcopenshell.util.pset.get_template("IFC4")
+            self.pset_template = self.psetqto.get_by_name(self.settings["pset"].Name)
 
     def update_existing_properties(self):
         for prop in self.get_properties():