Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-09-16 21:42:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

typing

Browse Source
This commit is contained in:
Andrej730
2024-10-18 14:35:04 +05:00
parent dbd64ebb0e
commit e073f11e25
2 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/bonsai/bonsai/bim/module/pset_template/operator.py
+10 -7
View File
@@ -33,19 +33,22 @@ class PsetTemplateOperator:
@final @final
def execute(self, context): def execute(self, context):
IfcStore.begin_transaction(self) IfcStore.begin_transaction(self)
IfcStore.pset_template_file.begin_transaction() template_file = IfcStore.pset_template_file
assert template_file
self.template_file = template_file
template_file.begin_transaction()
result = self._execute(context) result = self._execute(context)
IfcStore.pset_template_file.end_transaction() template_file.end_transaction()
IfcStore.add_transaction_operation(self) IfcStore.add_transaction_operation(self)
IfcStore.end_transaction(self) IfcStore.end_transaction(self)
bonsai.bim.handler.refresh_ui_data() bonsai.bim.handler.refresh_ui_data()
return {"FINISHED"} return {"FINISHED"}
def rollback(self, data): def rollback(self, data):
IfcStore.pset_template_file.undo() self.template_file.undo()
def commit(self, data): def commit(self, data):
IfcStore.pset_template_file.redo() self.template_file.redo()
def _execute(self, context: bpy.types.Context) -> None: def _execute(self, context: bpy.types.Context) -> None:
tool.Ifc.Operator._execute(self, context) tool.Ifc.Operator._execute(self, context)
@@ -102,10 +105,10 @@ class RemovePsetTemplate(bpy.types.Operator, PsetTemplateOperator):
bpy.ops.bim.disable_editing_pset_template() bpy.ops.bim.disable_editing_pset_template()
ifcopenshell.api.run( ifcopenshell.api.run(
"pset_template.remove_pset_template", "pset_template.remove_pset_template",
IfcStore.pset_template_file, self.template_file,
**{"pset_template": IfcStore.pset_template_file.by_id(int(props.pset_templates))} **{"pset_template": self.template_file.by_id(int(props.pset_templates))}
) )
IfcStore.pset_template_file.write(IfcStore.pset_template_path) self.template_file.write(IfcStore.pset_template_path)
bonsai.bim.handler.refresh_ui_data() bonsai.bim.handler.refresh_ui_data()
bonsai.bim.schema.reload(tool.Ifc.get().schema) bonsai.bim.schema.reload(tool.Ifc.get().schema)
src/bonsai/bonsai/tool/blender.py
+2
View File
@@ -559,6 +559,8 @@ class Blender(bonsai.core.tool.Blender):
def get_enum_safe(cls, props: bpy.types.PropertyGroup, prop_name: str) -> Union[str, None]: def get_enum_safe(cls, props: bpy.types.PropertyGroup, prop_name: str) -> Union[str, None]:
"""method created for readibility and to avoid console warnings like """method created for readibility and to avoid console warnings like
`pyrna_enum_to_py: current value '17' matches no enum in 'BIMModelProperties', '', 'relating_type_id'` `pyrna_enum_to_py: current value '17' matches no enum in 'BIMModelProperties', '', 'relating_type_id'`
:return: Enum property value as a string or None if current enum value is invalid.
""" """
# Yes, accessing items through annotations is a bit hacky # Yes, accessing items through annotations is a bit hacky
# but it's the only way to get the dynamic enum items # but it's the only way to get the dynamic enum items