Add helper to iterate over all annotation nodes of Arguments (#2860)

Author: cdce8p

ChangeLog

@@ -16,6 +16,11 @@ Release date: TBA
   Closes #2741
   Closes pylint-dev/pylint#6094
 
+* Add helper to iterate over all annotations nodes of function arguments,
+  ``Arguments.get_annotations()``.
+
+  Refs #2860
+
 
 What's New in astroid 4.0.2?
 ============================

astroid/nodes/node_classes.py

@@ -1020,6 +1020,23 @@ def get_children(self):
             if elt is not None:
                 yield elt
 
+    def get_annotations(self) -> Iterator[nodes.NodeNG]:
+        """Iterate over all annotations nodes."""
+        for elt in self.posonlyargs_annotations:
+            if elt is not None:
+                yield elt
+        for elt in self.annotations:
+            if elt is not None:
+                yield elt
+        if self.varargannotation is not None:
+            yield self.varargannotation
+
+        for elt in self.kwonlyargs_annotations:
+            if elt is not None:
+                yield elt
+        if self.kwargannotation is not None:
+            yield self.kwargannotation
+
     @decorators.raise_if_nothing_inferred
     def _infer(
         self: nodes.Arguments, context: InferenceContext | None = None, **kwargs: Any

tests/test_nodes.py

@@ -2314,6 +2314,19 @@ def test_arguments_default_value():
     assert node.args.default_value("flavor").value == "good"
 
 
+def test_arguments_annotations():
+    node = extract_node(
+        "def fruit(eat: str, /, peel: bool, *args: int, trim: float, **kwargs: bytes): ..."
+    )
+    assert isinstance(node.args, nodes.Arguments)
+    annotation_names = [
+        ann.name for ann in node.args.get_annotations() if isinstance(ann, nodes.Name)
+    ]
+    assert all(
+        name in annotation_names for name in ("str", "bool", "int", "float", "bytes")
+    )
+
+
 def test_deprecated_nodes_import_from_toplevel():
     # pylint: disable=import-outside-toplevel,no-name-in-module
     with pytest.raises(