Disallow adding in out animations if no animation is present

Griperis · Feb 8, 2025 · 6b4890e · 6b4890e
1 parent 6bb2531
commit 6b4890e
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/data_vis/geonodes/animations.py b/data_vis/geonodes/animations.py
@@ -107,6 +107,13 @@ class DV_AddInAnimation(DV_AnimationOperator):
         items=((AnimationNames.GROW_FROM_ZERO, "Grow", "Grow from zero"),),
     )
 
+    @classmethod
+    def poll(cls, context: bpy.types.Context) -> bool:
+        if not super().poll(context):
+            return False
+
+        return get_action(context.active_object) is not None
+
     def execute(self, context: bpy.types.Context):
         obj = context.active_object
         if is_in_present(obj):
@@ -165,6 +172,13 @@ class DV_AddOutAnimation(DV_AnimationOperator):
         items=((AnimationNames.GROW_TO_ZERO, "Shrink", "Shrink to zero"),),
     )
 
+    @classmethod
+    def poll(cls, context: bpy.types.Context) -> bool:
+        if not super().poll(context):
+            return False
+
+        return get_action(context.active_object) is not None
+
     def execute(self, context: bpy.types.Context):
         obj = context.active_object
         if is_out_present(context.active_object):