From c9f9cf89e882e341c062a63b03b25794a345cd29 Mon Sep 17 00:00:00 2001
From: Andrej730 <azhilenkov@gmail.com>
Date: Thu, 19 Sep 2024 17:14:32 +0500
Subject: [PATCH] file -> import/export - add poll messages in case ifc is not
 loaded

---
 src/bonsai/bonsai/bim/module/cost/operator.py |  8 +++
 .../bonsai/bim/module/sequence/operator.py    | 49 +++++++++++++------
 2 files changed, 43 insertions(+), 14 deletions(-)

diff --git a/src/bonsai/bonsai/bim/module/cost/operator.py b/src/bonsai/bonsai/bim/module/cost/operator.py
index b541702f89..45b78c89a7 100644
--- a/src/bonsai/bonsai/bim/module/cost/operator.py
+++ b/src/bonsai/bonsai/bim/module/cost/operator.py
@@ -518,6 +518,14 @@ class ImportCostScheduleCsv(bpy.types.Operator, ImportHelper, tool.Ifc.Operator)
     filter_glob: bpy.props.StringProperty(default="*.csv", options={"HIDDEN"})
     is_schedule_of_rates: bpy.props.BoolProperty(name="Is Schedule Of Rates", default=False)
 
+    @classmethod
+    def poll(cls, context):
+        ifc_file = tool.Ifc.get()
+        if ifc_file is None:
+            cls.poll_message_set("No IFC file is loaded.")
+            return False
+        return True
+
     def _execute(self, context):
         core.import_cost_schedule_csv(tool.Cost, self.filepath, self.is_schedule_of_rates)
         return {"FINISHED"}
diff --git a/src/bonsai/bonsai/bim/module/sequence/operator.py b/src/bonsai/bonsai/bim/module/sequence/operator.py
index ba05a3a31d..2dc9d8014e 100644
--- a/src/bonsai/bonsai/bim/module/sequence/operator.py
+++ b/src/bonsai/bonsai/bim/module/sequence/operator.py
@@ -632,8 +632,11 @@ class ImportCSV(bpy.types.Operator, tool.Ifc.Operator, ImportHelper):
 
     @classmethod
     def poll(cls, context):
-        ifc_file = IfcStore.get_file()
-        return ifc_file is not None
+        ifc_file = tool.Ifc.get()
+        if ifc_file is None:
+            cls.poll_message_set("No IFC file is loaded.")
+            return False
+        return True
 
     def execute(self, context):
         from ifc4d.csv4d2ifc import Csv2Ifc
@@ -657,8 +660,11 @@ class ImportP6(bpy.types.Operator, ImportHelper):
 
     @classmethod
     def poll(cls, context):
-        ifc_file = IfcStore.get_file()
-        return ifc_file is not None
+        ifc_file = tool.Ifc.get()
+        if ifc_file is None:
+            cls.poll_message_set("No IFC file is loaded.")
+            return False
+        return True
 
     def execute(self, context):
         from ifc4d.p62ifc import P62Ifc
@@ -683,8 +689,11 @@ class ImportP6XER(bpy.types.Operator, ImportHelper):
 
     @classmethod
     def poll(cls, context):
-        ifc_file = IfcStore.get_file()
-        return ifc_file is not None
+        ifc_file = tool.Ifc.get()
+        if ifc_file is None:
+            cls.poll_message_set("No IFC file is loaded.")
+            return False
+        return True
 
     def execute(self, context):
         from ifc4d.p6xer2ifc import P6XER2Ifc
@@ -709,8 +718,11 @@ class ImportPP(bpy.types.Operator, ImportHelper):
 
     @classmethod
     def poll(cls, context):
-        ifc_file = IfcStore.get_file()
-        return ifc_file is not None
+        ifc_file = tool.Ifc.get()
+        if ifc_file is None:
+            cls.poll_message_set("No IFC file is loaded.")
+            return False
+        return True
 
     def execute(self, context):
         from ifc4d.pp2ifc import PP2Ifc
@@ -735,8 +747,11 @@ class ImportMSP(bpy.types.Operator, ImportHelper):
 
     @classmethod
     def poll(cls, context):
-        ifc_file = IfcStore.get_file()
-        return ifc_file is not None
+        ifc_file = tool.Ifc.get()
+        if ifc_file is None:
+            cls.poll_message_set("No IFC file is loaded.")
+            return False
+        return True
 
     def execute(self, context):
         from ifc4d.msp2ifc import MSP2Ifc
@@ -763,8 +778,11 @@ class ExportMSP(bpy.types.Operator, ImportHelper):
 
     @classmethod
     def poll(cls, context):
-        ifc_file = IfcStore.get_file()
-        return ifc_file is not None
+        ifc_file = tool.Ifc.get()
+        if ifc_file is None:
+            cls.poll_message_set("No IFC file is loaded.")
+            return False
+        return True
 
     def execute(self, context):
         from ifc4d.ifc2msp import Ifc2Msp
@@ -793,8 +811,11 @@ class ExportP6(bpy.types.Operator, ImportHelper):
 
     @classmethod
     def poll(cls, context):
-        ifc_file = IfcStore.get_file()
-        return ifc_file is not None
+        ifc_file = tool.Ifc.get()
+        if ifc_file is None:
+            cls.poll_message_set("No IFC file is loaded.")
+            return False
+        return True
 
     def execute(self, context):
         from ifc4d.ifc2p6 import Ifc2P6