fix T53624: A.N.T. Landscape should not be operating in edit mode

2 files changed, 12 insertions, 1 deletions

diff --git a/ant_landscape/add_mesh_ant_landscape.py b/ant_landscape/add_mesh_ant_landscape.py
index 5d25cf24..6a36c42f 100644
--- a/ant_landscape/add_mesh_ant_landscape.py
+++ b/ant_landscape/add_mesh_ant_landscape.py
@@ -586,6 +586,14 @@ class AntAddLandscape(bpy.types.Operator):
             description="Automatic refresh"
             )
 
+    @classmethod
+    def poll(self, context):
+        ob = context.object
+        if ob is not None:
+            if ob.mode == 'EDIT':
+                return False
+        return True
+
     def draw(self, context):
         draw_ant_refresh(self, context)
         draw_ant_main(self, context, generate=True)
diff --git a/ant_landscape/ant_functions.py b/ant_landscape/ant_functions.py
index 3b515933..3c0f2c34 100644
--- a/ant_landscape/ant_functions.py
+++ b/ant_landscape/ant_functions.py
@@ -193,7 +193,10 @@ class AntLandscapeRegenerate(bpy.types.Operator):
 
     @classmethod
     def poll(cls, context):
-        return bpy.context.active_object.ant_landscape
+        ob = bpy.context.active_object
+        if ob.mode == 'EDIT':
+            return False
+        return ob.ant_landscape
 
 
     def execute(self, context):