From edd893fef8f9e2b5096e9f54a1664c7a6d171515 Mon Sep 17 00:00:00 2001
From: Greg <gregzzmail@gmail.com>
Date: Thu, 11 Aug 2016 12:44:17 +0200
Subject: [PATCH] Node Wrangler: Some more checks for Save This Image
---
node_wrangler.py | 8 +++++++-
1 file changed, 7 insertions(+), 1 deletion(-)
diff --git a/node_wrangler.py b/node_wrangler.py
index b6a7b4377..6b0659513 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
--
GitLab