Object enhancements

NodeToPython/export/compositor/exporter.py

@@ -33,9 +33,11 @@ def __init__(
     def _create_scene(self):
         indent_level = self._get_obj_creation_indent()
 
+        scene: bpy.types.Scene = self._node_tree_info._obj
+
         #TODO: wrap in more general unique name util function
         self._write(f"# Generate unique scene name", indent_level)
-        self._write(f"{BASE_NAME} = {str_to_py_str(self._node_tree_info._obj.name)}",
+        self._write(f"{BASE_NAME} = {str_to_py_str(scene.name)}",
                     indent_level)
         self._write(f"{END_NAME} = {BASE_NAME}", indent_level)
         self._write(f"if bpy.data.scenes.get({END_NAME}) is not None:", indent_level)
@@ -55,6 +57,46 @@ def _create_scene(self):
         self._write(f"{SCENE}.name = {END_NAME}", indent_level)
         self._write(f"{SCENE}.use_fake_user = True", indent_level)
         self._write(f"bpy.context.window.scene = {SCENE}", indent_level)
+
+        regular_attrs = [
+            "audio_doppler_factor",
+            "audio_doppler_speed",
+            "audio_volume",
+            "use_audio",
+            "use_audio_scrub",
+            "use_gravity"
+        ]
+        enum_attrs = [
+            "audio_distance_model",
+            "sync_mode"
+        ]
+        vec3_attrs = [
+            "gravity"
+        ]
+        str_attrs = [
+            "use_stamp_note"
+        ]
+
+        for attr in regular_attrs:
+            self._write(
+                f"{self._obj_var}.{attr} = {getattr(scene, attr)}",
+                indent_level
+            )
+        for attr in enum_attrs:
+            self._write(
+                f"{self._obj_var}.{attr} = {enum_to_py_str(getattr(scene, attr))}",
+                indent_level
+            )
+        for attr in vec3_attrs:
+            self._write(
+                f"{self._obj_var}.{attr} = {vec3_to_py_str(getattr(scene, attr))}",
+                indent_level
+            )
+        for attr in str_attrs:
+            self._write(
+                f"{self._obj_var}.{attr} = {str_to_py_str(getattr(scene, attr))}",
+                indent_level
+            )
         self._write("", 0)
 
     # NodeTreeExporter interface

NodeToPython/export/node_tree_exporter.py

@@ -273,8 +273,10 @@ def _set_node_tree_properties(self, node_tree: bpy.types.NodeTree) -> None:
         if bpy.app.version >= (4, 2, 0):
             color_tag_str = enum_to_py_str(node_tree.color_tag)
             self._write(f"{nt_var}.color_tag = {color_tag_str}")
+
             desc_str = str_to_py_str(node_tree.description)
             self._write(f"{nt_var}.description = {desc_str}")
+
         if bpy.app.version >= (4, 3, 0):
             default_width = node_tree.default_group_node_width
             self._write(f"{nt_var}.default_group_node_width = {default_width}")

NodeToPython/export/ntp_operator.py

@@ -398,7 +398,6 @@ def dfs(nt: bpy.types.NodeTree) -> None:
                     print(f"Library {lib_path} didn't seem essential, copying node groups")
 
             if nt not in self._visited:
-                print(f"Visiting {nt.name}")
                 self._visited.add(nt)
                 if nt not in self._node_trees:
                     self._node_trees[nt] = NodeTreeInfo()