Node Wrangler: Some more checks for Save This Image

1 files changed, 7 insertions, 1 deletions

diff --git a/node_wrangler.py b/node_wrangler.py
index b6a7b437..6b065951 100644
--- a/node_wrangler.py
+++ b/node_wrangler.py
@@ -3192,7 +3192,13 @@ class NWSaveViewer(bpy.types.Operator, ExportHelper):
 
     @classmethod
     def poll(cls, context):
-        return nw_check(context) and context.space_data.tree_type == 'CompositorNodeTree' and "Viewer Node" in [i.name for i in bpy.data.images]
+        valid = False
+        if nw_check(context):
+            if context.space_data.tree_type == 'CompositorNodeTree':
+                if "Viewer Node" in [i.name for i in bpy.data.images]:
+                    if sum(bpy.data.images["Viewer Node"].size) > 0:  # False if not connected or connected but no image
+                        valid = True
+        return valid
 
     def execute(self, context):
         fp = self.filepath