diff options
-rw-r--r-- | ant_landscape/add_mesh_ant_landscape.py | 8 | ||||
-rw-r--r-- | ant_landscape/ant_functions.py | 5 |
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): |