From 847b5d226ee192193af8cb920235064ac757e51d Mon Sep 17 00:00:00 2001
From: Andrej730 <azhilenkov@gmail.com>
Date: Wed, 21 Aug 2024 17:03:41 +0500
Subject: [PATCH] bim.add_bcf_comment - poll message

---
 src/bonsai/bonsai/bim/module/bcf/operator.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/bonsai/bonsai/bim/module/bcf/operator.py b/src/bonsai/bonsai/bim/module/bcf/operator.py
index fd39e63edf..433735b789 100644
--- a/src/bonsai/bonsai/bim/module/bcf/operator.py
+++ b/src/bonsai/bonsai/bim/module/bcf/operator.py
@@ -1106,9 +1106,20 @@ class AddBcfComment(bpy.types.Operator):
     def poll(cls, context):
         props = context.scene.BCFProperties
         if not props.comment:
+            cls.poll_message_set("No comment to add.")
             return False
-        if props.has_related_viewpoint and not bcf_prop.getBcfViewpoints(None, context):
+
+        topic = props.active_topic
+        if not topic:
+            cls.poll_message_set("No topic is active.")
             return False
+
+        if props.has_related_viewpoint:
+            topic = props.topics[topic.name]
+            viewpoint = tool.Blender.get_enum_safe(topic, "viewpoints")
+            if not viewpoint:
+                cls.poll_message_set("No viewpoint is active to add a comment with viewpoint.")
+                return False
         return True
 
     def execute(self, context):