Comments: add whitespace at line beginning, capitalize first word

Author: PiOverFour

NodeToPython/compositor/operator.py

@@ -63,7 +63,7 @@ def _initialize_compositor_node_tree(self, ntp_nt, nt_name):
 
         if ntp_nt.node_tree == self._base_node_tree and self.is_scene:
             self._write(f"{ntp_nt.var} = {SCENE}.node_tree")
-            self._write(f"#start with a clean node tree")
+            self._write(f"# Start with a clean node tree")
             self._write(f"for {NODE} in {ntp_nt.var}.nodes:")
             self._write(f"{ntp_nt.var}.nodes.remove({NODE})", self._inner_indent_level + 1)
         else:
@@ -187,7 +187,7 @@ def _process_node_tree(self, node_tree: CompositorNodeTree):
             self._tree_interface_settings(ntp_nt)
 
         #initialize nodes
-        self._write(f"#initialize {nt_var} nodes")
+        self._write(f"# Initialize {nt_var} nodes")
 
         for node in node_tree.nodes:
             self._process_node(node, ntp_nt)

NodeToPython/geometry/operator.py

@@ -71,7 +71,7 @@ def _process_zones(self, zone_inputs: list[GeometryNode]) -> None:
                 zone_input_var = self._node_vars[zone_input]
                 zone_output_var = self._node_vars[zone_output]
 
-                self._write(f"#Process zone input {zone_input.name}")
+                self._write(f"# Process zone input {zone_input.name}")
                 self._write(f"{zone_input_var}.pair_with_output"
                             f"({zone_output_var})")
 
@@ -132,7 +132,7 @@ def _process_node_tree(self, node_tree: GeometryNodeTree) -> None:
             self._tree_interface_settings(ntp_nt)
 
         #initialize nodes
-        self._write(f"#initialize {nt_var} nodes")
+        self._write(f"# Initialize {nt_var} nodes\n")
         for node in node_tree.nodes:
             self._process_node(node, ntp_nt)
 

NodeToPython/ntp_operator.py

@@ -323,7 +323,7 @@ def _create_node(self, node: Node, node_tree_var: str) -> str:
         node_var (str): variable name for the node
         """
 
-        self._write(f"#node {node.name}")
+        self._write(f"# Node {node.name}")
 
         node_var = self._create_var(node.name)
         self._node_vars[node] = node_var
@@ -530,11 +530,11 @@ def _group_io_settings(self, node: Node,
                 io_sockets = node.inputs
                 io_socket_interfaces = node_tree.outputs
 
-            self._write(f"#{node_tree_var} {io}s")
+            self._write(f"# {node_tree_var} {io}s")
             for i, inout in enumerate(io_sockets):
                 if inout.bl_idname == 'NodeSocketVirtual':
                     continue
-                self._write(f"#{io} {inout.name}")
+                self._write(f"# {io.capitalize()} {inout.name}")
                 idname = enum_to_py_str(inout.bl_idname)
                 name = str_to_py_str(inout.name)
                 self._write(f"{node_tree_var}.{io}s.new({idname}, {name})")
@@ -642,7 +642,7 @@ def _create_socket(self, socket: NodeTreeInterfaceSocket,
             ntp_nt (NTP_NodeTree): owner of the socket
             """
 
-            self._write(f"#Socket {socket.name}")
+            self._write(f"# Socket {socket.name}")
             # initialization
             socket_var = self._create_var(socket.name + "_socket") 
             name = str_to_py_str(socket.name)
@@ -755,7 +755,7 @@ def _create_panel(self, panel: NodeTreeInterfacePanel,
             ntp_nt (NTP_NodeTree): owner of the socket
             """
 
-            self._write(f"#Panel {panel.name}")
+            self._write(f"# Panel {panel.name}")
 
             panel_var = self._create_var(panel.name + "_panel")
             panel_dict[panel] = panel_var
@@ -840,7 +840,7 @@ def _tree_interface_settings(self, ntp_nt: NTP_NodeTree) -> None:
             ntp_nt (NTP_NodeTree): the node tree to set the interface for
             """
 
-            self._write(f"#{ntp_nt.var} interface")
+            self._write(f"# {ntp_nt.var} interface\n")
             panel_dict: dict[NodeTreeInterfacePanel, str] = {}
             items_processed: set[NodeTreeInterfaceItem] = set()
 
@@ -930,7 +930,7 @@ def _set_input_defaults(self, node: Node) -> None:
                 else:
                     default_val = input.default_value
                 if default_val is not None:
-                    self._write(f"#{input.identifier}")
+                    self._write(f"# {input.identifier}")
                     self._write(f"{socket_var}.default_value = {default_val}")
         self._write("", 0)
 
@@ -1028,7 +1028,7 @@ def _color_ramp_settings(self, node: Node, color_ramp_name: str) -> None:
         self._write("", 0)
 
         # key points
-        self._write(f"#initialize color ramp elements")
+        self._write(f"# Initialize color ramp elements")
         self._write((f"{ramp_str}.elements.remove"
                     f"({ramp_str}.elements[0])"))
         for i, element in enumerate(color_ramp.elements):
@@ -1062,7 +1062,7 @@ def _curve_mapping_settings(self, node: Node,
         node_var = self._node_vars[node]
 
         # mapping settings
-        self._write(f"#mapping settings")
+        self._write(f"# Mapping settings")
         mapping_var = f"{node_var}.{curve_mapping_name}"
 
         # extend
@@ -1098,7 +1098,7 @@ def _curve_mapping_settings(self, node: Node,
             self._create_curve_map(node, i, curve, curve_mapping_name)
 
         # update curve
-        self._write(f"#update curve after changes")
+        self._write(f"# Update curve after changes")
         self._write(f"{mapping_var}.update()")
 
     def _create_curve_map(self, node: Node, i: int, curve: bpy.types.CurveMap,
@@ -1114,7 +1114,7 @@ def _create_curve_map(self, node: Node, i: int, curve: bpy.types.CurveMap,
         """
         node_var = self._node_vars[node]
 
-        self._write(f"#curve {i}")
+        self._write(f"# Curve {i}")
         curve_i_var = self._create_var(f"{node_var}_curve_{i}")
         self._write(f"{curve_i_var} = "
                     f"{node_var}.{curve_mapping_name}.curves[{i}]")
@@ -1217,7 +1217,7 @@ def _load_image(self, img: bpy.types.Image, img_var: str) -> None:
         img_str = img_to_py_str(img)
 
         # TODO: convert to special variables
-        self._write(f"#load image {img_str}")
+        self._write(f"# Load image {img_str}")
         self._write(f"{BASE_DIR} = "
                     f"os.path.dirname(os.path.abspath(__file__))")
         self._write(f"{IMAGE_PATH} = "
@@ -1227,7 +1227,7 @@ def _load_image(self, img: bpy.types.Image, img_var: str) -> None:
                     f"({IMAGE_PATH}, check_existing = True)")
 
         # copy image settings
-        self._write(f"#set image settings")
+        self._write(f"# Set image settings")
 
         # source
         source = enum_to_py_str(img.source)
@@ -1415,7 +1415,7 @@ def _set_parents(self, node_tree: NodeTree) -> None:
         for node in node_tree.nodes:
             if node is not None and node.parent is not None:
                 if not parent_comment:
-                    self._write(f"#Set parents")
+                    self._write(f"# Set parents")
                     parent_comment = True
                 node_var = self._node_vars[node]
                 parent_var = self._node_vars[node.parent]
@@ -1430,7 +1430,7 @@ def _set_locations(self, node_tree: NodeTree) -> None:
         node_tree (NodeTree): node tree we're obtaining nodes from
         """
 
-        self._write(f"#Set locations")
+        self._write(f"# Set locations")
         for node in node_tree.nodes:
             node_var = self._node_vars[node]
             self._write(f"{node_var}.location "
@@ -1447,7 +1447,7 @@ def _set_dimensions(self, node_tree: NodeTree) -> None:
         if not self._should_set_dimensions:
             return
 
-        self._write(f"#Set dimensions")
+        self._write(f"# Set dimensions")
         for node in node_tree.nodes:
             node_var = self._node_vars[node]
             self._write(f"{node_var}.width, {node_var}.height "
@@ -1466,7 +1466,7 @@ def _init_links(self, node_tree: NodeTree) -> None:
 
         links = node_tree.links
         if links:
-            self._write(f"#initialize {nt_var} links")
+            self._write(f"# Initialize {nt_var} links\n")
             if hasattr(links[0], "multi_input_sort_id"):
                 # generate links in the correct order for multi input sockets
                 links = sorted(links, key=lambda link: link.multi_input_sort_id)
@@ -1495,7 +1495,7 @@ def _init_links(self, node_tree: NodeTree) -> None:
                     output_idx = i
                     break
 
-            self._write(f"#{in_node_var}.{input_socket.name} "
+            self._write(f"# {in_node_var}.{input_socket.name} "
                         f"-> {out_node_var}.{output_socket.name}")
             self._write(f"{nt_var}.links.new({in_node_var}"
                         f".outputs[{input_idx}], "

NodeToPython/shader/operator.py

@@ -47,7 +47,7 @@ def _initialize_shader_node_tree(self, ntp_node_tree: NTP_NodeTree,
 
         if ntp_node_tree.node_tree == self._base_node_tree:
             self._write(f"{ntp_node_tree.var} = {MAT_VAR}.node_tree")
-            self._write(f"#start with a clean node tree")
+            self._write(f"# Start with a clean node tree")
             self._write(f"for {NODE} in {ntp_node_tree.var}.nodes:")
             self._write(f"{ntp_node_tree.var}.nodes.remove({NODE})", self._inner_indent_level + 1)
         else:
@@ -108,7 +108,7 @@ def _process_node_tree(self, node_tree: ShaderNodeTree) -> None:
             self._tree_interface_settings(ntp_nt)
 
         #initialize nodes
-        self._write(f"#initialize {nt_var} nodes")
+        self._write(f"# Initialize {nt_var} nodes")
 
         for node in node_tree.nodes:
             self._process_node(node, ntp_nt)