WIP

Author: nutti

src/gen_modfile/gen_node_inputs_outputs_for_node_class.py

@@ -26,8 +26,8 @@
 #
 ##############################################################################
 
-# ruff: noqa: UP006, UP032, UP035, PTH103, PTH113, PTH118, PTH120,
-# ruff: noqa: PTH123, SIM115
+# ruff: noqa: UP006, UP032, UP035, PTH103, PTH118, PTH120,
+# ruff: noqa: PTH123
 
 import argparse
 import json
@@ -57,7 +57,8 @@ def analyze() -> Dict:
         nodetree_class_name = "{}Tree".format(node_baseclass_name)
 
         try:
-            node_group = bpy.data.node_groups.new(nodetree_class_name, nodetree_class_name)
+            node_group = bpy.data.node_groups.new(
+                nodetree_class_name, nodetree_class_name)
         except TypeError:
             continue
 
@@ -106,15 +107,19 @@ def analyze() -> Dict:
                     new_func_def["parameter_details"].append({
                         "name": "key",
                         "description": None,
-                        "data_type": 'typing.Literal[{}] | typing.Literal["{}"]'
-                            .format(i, node_input.name),
+                        "data_type": (
+                            'typing.Literal[{}] | typing.Literal["{}"]'
+                            .format(i, node_input.name)
+                        ),
                         "mod_option": "skip-refine",
                     })
                     new_func_def["return"] = {
                         "type": "return",
                         "description": None,
-                        "data_type": ":class:`bpy.types.{}`"
-                            .format(node_input.__class__.__name__),
+                        "data_type": (
+                            ":class:`bpy.types.{}`"
+                            .format(node_input.__class__.__name__)
+                        ),
                     }
                     new_class_def["methods"].append(new_func_def)
 
@@ -127,8 +132,10 @@ def analyze() -> Dict:
                     "description": None,
                     "class": node_class_name,
                     "module": "bpy.types",
-                    "data_type": ":class:`bpy.types.{}`"
-                        .format(fake_class_name),
+                    "data_type": (
+                        ":class:`bpy.types.{}`"
+                        .format(fake_class_name)
+                    ),
                     "mod_option": "skip-refine",
                 }
                 append_class_def["attributes"].append(new_attribute)
@@ -160,15 +167,19 @@ def analyze() -> Dict:
                     new_func_def["parameter_details"].append({
                         "name": "key",
                         "description": None,
-                        "data_type": 'typing.Literal[{}] | typing.Literal["{}"]'
-                            .format(i, node_output.name),
+                        "data_type": (
+                            'typing.Literal[{}] | typing.Literal["{}"]'
+                            .format(i, node_output.name)
+                        ),
                         "mod_option": "skip-refine",
                     })
                     new_func_def["return"] = {
                         "type": "return",
                         "description": None,
-                        "data_type": ":class:`bpy.types.{}`"
+                        "data_type": (
+                            ":class:`bpy.types.{}`"
                             .format(node_output.__class__.__name__)
+                        ),
                     }
                     new_class_def["methods"].append(new_func_def)
 
@@ -181,8 +192,10 @@ def analyze() -> Dict:
                     "description": None,
                     "class": node_class_name,
                     "module": "bpy.types",
-                    "data_type": ":class:`bpy.types.{}`"
-                        .format(fake_class_name),
+                    "data_type": (
+                        ":class:`bpy.types.{}`"
+                        .format(fake_class_name)
+                    ),
                 }
                 append_class_def["attributes"].append(new_attribute_def)
                 results["append"].append(append_class_def)
@@ -207,7 +220,7 @@ def write_class_info(f: TextIO, mod_kind: str, class_info: Dict) -> None:
             ", ".join(func_info["parameters"])))
         if "overload" in func_info["options"]:
             f.write("      :option function: overload\n")
-        for param_info in func_info["parameter_details"]:  # noqa: FURB122
+        for param_info in func_info["parameter_details"]:
             f.write("      :type {}: {}\n"
                     .format(param_info["name"], param_info["data_type"]))
             f.write("      :mod-option arg {}: {}\n"