From 07b91f35dd7ea997c7135ec97cecbae30bd45334 Mon Sep 17 00:00:00 2001 From: Damien Picard Date: Sun, 17 Dec 2023 22:40:18 +0100 Subject: [PATCH] Node Wrangler: do not rely on image name to detect Viewer Node image Instead, use the fact that Viewer Nodes have 'VIEWER' for a source and exactly 0 render_slots, contrary to Render Results, which have at least 1. --- node_wrangler/operators.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/node_wrangler/operators.py b/node_wrangler/operators.py index 9c5f2f1c..8dd924b4 100644 --- a/node_wrangler/operators.py +++ b/node_wrangler/operators.py @@ -2557,7 +2557,9 @@ class NWSaveViewer(bpy.types.Operator, ExportHelper): def poll(cls, context): return (nw_check(context) and nw_check_space_type(cls, context, 'CompositorNodeTree') - and "Viewer Node" in [i.name for i in bpy.data.images] + and any(img.source == 'VIEWER' + and img.render_slots == 0 + for img in bpy.data.images) and sum(bpy.data.images["Viewer Node"].size) > 0) # False if not connected or connected but no image def execute(self, context): -- 2.11.4.GIT