Skip to content

Commit e18c01a

Browse files
committed
feat: set image sockets in scripts when possible
1 parent ea5819f commit e18c01a

File tree

1 file changed

+12
-4
lines changed

1 file changed

+12
-4
lines changed

NodeToPython/ntp_operator.py

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -423,10 +423,15 @@ def _set_settings_defaults(self, node: Node) -> None:
423423
elif st == ST.NODE_TREE:
424424
self._node_tree_settings(node, attr_name)
425425
elif st == ST.IMAGE:
426-
if self._addon_dir is not None and attr is not None:
426+
if attr is None:
427+
continue
428+
if self._addon_dir is not None:
427429
if attr.source in {'FILE', 'GENERATED', 'TILED'}:
428430
if self._save_image(attr):
429431
self._load_image(attr, f"{node_var}.{attr_name}")
432+
else:
433+
self._set_if_in_blend_file(attr, setting_str, "images")
434+
430435
elif st == ST.IMAGE_USER:
431436
self._image_user_settings(attr, f"{node_var}.{attr_name}")
432437
elif st == ST.SIM_OUTPUT_ITEMS:
@@ -845,9 +850,12 @@ def _set_input_defaults(self, node: Node) -> None:
845850
# images
846851
elif input.bl_idname == 'NodeSocketImage':
847852
img = input.default_value
848-
if img is not None and self._addon_dir != None: # write in a better way
849-
if self._save_image(img):
850-
self._load_image(img, f"{socket_var}.default_value")
853+
if img is not None:
854+
if self._addon_dir != None: # write in a better way
855+
if self._save_image(img):
856+
self._load_image(img, f"{socket_var}.default_value")
857+
else:
858+
self._in_file_inputs(input, socket_var, "images")
851859
default_val = None
852860

853861
# materials

0 commit comments

Comments
 (0)