Fix #1786. Swap out sub-operators with refactored core calls.

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

@@ -31,6 +31,7 @@ import blenderbim.core.context
 import blenderbim.core.owner
 from blenderbim.bim.ifc import IfcStore
 from blenderbim.bim import import_ifc
+from blenderbim.bim import export_ifc
 
 
 class CreateProject(bpy.types.Operator):
@@ -729,3 +730,79 @@ class LoadLink(bpy.types.Operator):
         link = context.scene.BIMProjectProperties.links.get(self.filepath)
         link.is_loaded = True
         return {"FINISHED"}
+
+
+class ExportIFC(bpy.types.Operator):
+    bl_idname = "export_ifc.bim"
+    bl_label = "Export IFC"
+    bl_options = {"REGISTER", "UNDO"}
+    filename_ext = ".ifc"
+    filter_glob: bpy.props.StringProperty(default="*.ifc;*.ifczip;*.ifcxml;*.ifcjson", options={"HIDDEN"})
+    filepath: bpy.props.StringProperty(subtype="FILE_PATH")
+    json_version: bpy.props.EnumProperty(items=[("4", "4", ""), ("5a", "5a", "")], name="IFC JSON Version")
+    json_compact: bpy.props.BoolProperty(name="Export Compact IFCJSON", default=False)
+
+    def invoke(self, context, event):
+        if not IfcStore.get_file():
+            self.report({"ERROR"}, "No IFC project is available for export - create or import a project first.")
+            return {"FINISHED"}
+        if context.scene.BIMProperties.ifc_file:
+            self.filepath = context.scene.BIMProperties.ifc_file
+            return self.execute(context)
+        if not self.filepath:
+            self.filepath = bpy.path.ensure_ext(bpy.data.filepath, ".ifc")
+        WindowManager = context.window_manager
+        WindowManager.fileselect_add(self)
+        return {"RUNNING_MODAL"}
+
+    def execute(self, context):
+        return IfcStore.execute_ifc_operator(self, context)
+
+    def _execute(self, context):
+        start = time.time()
+        logger = logging.getLogger("ExportIFC")
+        path_log = os.path.join(context.scene.BIMProperties.data_dir, "process.log")
+        if not os.access(context.scene.BIMProperties.data_dir, os.W_OK):
+            path_log = os.path.join(tempfile.mkdtemp(), "process.log")
+        logging.basicConfig(
+            filename=path_log,
+            filemode="a",
+            level=logging.DEBUG,
+        )
+        extension = self.filepath.split(".")[-1]
+        if extension == "ifczip":
+            output_file = bpy.path.ensure_ext(self.filepath, ".ifczip")
+        elif extension == "ifcjson":
+            output_file = bpy.path.ensure_ext(self.filepath, ".ifcjson")
+        else:
+            output_file = bpy.path.ensure_ext(self.filepath, ".ifc")
+
+        settings = export_ifc.IfcExportSettings.factory(context, output_file, logger)
+        settings.json_version = self.json_version
+        settings.json_compact = self.json_compact
+
+        ifc_exporter = export_ifc.IfcExporter(settings)
+        settings.logger.info("Starting export")
+        ifc_exporter.export()
+        settings.logger.info("Export finished in {:.2f} seconds".format(time.time() - start))
+        print("Export finished in {:.2f} seconds".format(time.time() - start))
+        scene = context.scene
+        if not scene.DocProperties.ifc_files:
+            new = scene.DocProperties.ifc_files.add()
+            new.name = output_file
+        if not scene.BIMProperties.ifc_file:
+            scene.BIMProperties.ifc_file = output_file
+        if bpy.data.is_saved and bpy.data.is_dirty and bpy.data.filepath:
+            bpy.ops.wm.save_mainfile(filepath=bpy.data.filepath)
+        blenderbim.bim.handler.purge_module_data()
+        return {"FINISHED"}
+
+
+class ImportIFC(bpy.types.Operator):
+    bl_idname = "import_ifc.bim"
+    bl_label = "Import IFC"
+    bl_options = {"REGISTER", "UNDO"}
+
+    def execute(self, context):
+        bpy.ops.bim.load_project("INVOKE_DEFAULT")
+        return {"FINISHED"}