Write docstring instead of comment for initialization function

Author: PiOverFour

NodeToPython/compositor/operator.py

@@ -58,8 +58,8 @@ def _create_scene(self, indent_level: int):
 
     def _initialize_compositor_node_tree(self, ntp_nt, nt_name):
         #initialize node group
-        self._write(f"#initialize {nt_name} node group", self._outer_indent_level)
         self._write(f"def {ntp_nt.var}_node_group():", self._outer_indent_level)
+        self._write(f'"""Initialize {nt_name} node group"""')
 
         if ntp_nt.node_tree == self._base_node_tree and self.is_scene:
             self._write(f"{ntp_nt.var} = {SCENE}.node_tree")

NodeToPython/geometry/operator.py

@@ -116,8 +116,8 @@ def _process_node_tree(self, node_tree: GeometryNodeTree) -> None:
         self._node_tree_vars[node_tree] = nt_var
 
         #initialize node group
-        self._write(f"#initialize {nt_var} node group", self._outer_indent_level)
         self._write(f"def {nt_var}_node_group():", self._outer_indent_level)
+        self._write(f'"""Initialize {nt_var} node group"""')
         self._write(f"{nt_var} = bpy.data.node_groups.new("
                     f"type = \'GeometryNodeTree\', "
                     f"name = {str_to_py_str(node_tree.name)})\n")

NodeToPython/shader/operator.py

@@ -42,8 +42,8 @@ def _initialize_shader_node_tree(self, ntp_node_tree: NTP_NodeTree,
             variable to use
         nt_name (str): name to use for the node tree
         """
-        self._write(f"#initialize {nt_name} node group", self._outer_indent_level)
-        self._write(f"def {ntp_node_tree.var}_node_group():\n", self._outer_indent_level)
+        self._write(f"def {ntp_node_tree.var}_node_group():", self._outer_indent_level)
+        self._write(f'"""Initialize {nt_name} node group"""')
 
         if ntp_node_tree.node_tree == self._base_node_tree:
             self._write(f"{ntp_node_tree.var} = {MAT_VAR}.node_tree")