feat: walk updates

src/ansys/fluent/core/codegen/walk_api.py

@@ -8,37 +8,37 @@
     >>> from ansys.fluent.core.codegen import walk_api
     >>> from ansys.fluent.core.generated.solver import settings_252
     >>> walk_api.walk_api(settings_252.root, lambda p: print(p), current_path=[])
+    >>> walk_api.walk_api(settings_252.root, lambda p, api_item_type: print(p, api_item_type), current_path=[])
 
 """
 
+from inspect import signature
 from typing import List
 
 
-def walk_api(api_root_cls, on_each_path, current_path: str | List[str] = ""):
+def walk_api(
+    api_cls, on_each_path, current_path: str | List[str] = "", api_item_type: str = ""
+):
     """
     Recursively traverse the API hierarchy, calling `on_each_path` for each item.
 
     Parameters:
-    - api_root_cls: The root class of the API hierarchy.
+    - api_cls: The current class of the API hierarchy.
     - on_each_path: A callback function to call for each path.
     - current_path: The current path in the hierarchy (default: empty string).
       Paths can be either dot-separated strings or string lists. The type is
       determined by the client.
     """
     # Skip the root path
     if current_path:
-        on_each_path(current_path)
+        if len(signature(on_each_path).parameters) == 2:
+            on_each_path(current_path, api_item_type)
+        else:
+            on_each_path(current_path)
 
-    # Get child names and their respective classes
-    all_names = [
-        name
-        for attr in ("child_names", "argument_names", "command_names", "query_names")
-        for name in getattr(api_root_cls, attr, [])
-    ]
-    child_classes = getattr(api_root_cls, "_child_classes", {})
-
-    # Traverse each child
-    for child_name in all_names:
+    child_classes = getattr(api_cls, "_child_classes", {})
+
+    def _traverse(child_name, api_item_type):
         if child_name in child_classes:
             child_cls = child_classes[child_name]
             # Construct the new path
@@ -49,9 +49,36 @@ def walk_api(api_root_cls, on_each_path, current_path: str | List[str] = ""):
                     f"{current_path}.{child_name}" if current_path else child_name
                 )
             # Recursively walk the child
-            walk_api(child_cls, on_each_path, new_path)
+            walk_api(child_cls, on_each_path, new_path, api_item_type)
+
+            # Delegate directly to any child_object_type (relevant for named objects)
+            child_object_type = getattr(api_cls, "child_object_type", None)
+            if child_object_type:
+                walk_api(child_cls, on_each_path, current_path, api_item_type)
+
+    # Get child names and their respective classes
+    if api_item_type == "method":
+        arg_names = [
+            name for attr in ("argument_names",) for name in getattr(api_cls, attr, [])
+        ]
+
+        for arg_name in arg_names:
+            _traverse(arg_name, api_item_type="argument")
+
+    else:
+
+        child_names = [
+            name for attr in ("child_names",) for name in getattr(api_cls, attr, [])
+        ]
+
+        for child_name in child_names:
+            _traverse(child_name, api_item_type="")
+
+        method_names = [
+            name
+            for attr in ("command_names", "query_names")
+            for name in getattr(api_cls, attr, [])
+        ]
 
-    # Delegate directly to any child_object_type (relevant for named objects)
-    child_object_type = getattr(api_root_cls, "child_object_type", None)
-    if child_object_type:
-        walk_api(child_cls, on_each_path, current_path)
+        for method_name in method_names:
+            _traverse(method_name, api_item_type="method")