refact: add __iter__ protocol

Author: boriel

src/symbols/arglist.py

@@ -1,19 +1,21 @@
 #!/usr/bin/python
 # -*- coding: utf-8 -*-
-# vim: ts=4:et:sw=4:
 
 # ----------------------------------------------------------------------
 # Copyleft (K), Jose M. Rodriguez-Rosa (a.k.a. Boriel)
 #
 # This program is Free Software and is released under the terms of
 #                    the GNU General License
 # ----------------------------------------------------------------------
+from __future__ import annotations
+
+from typing import Iterable
 
 from src.symbols.argument import SymbolARGUMENT
 from src.symbols.symbol_ import Symbol
 
 
-class SymbolARGLIST(Symbol):
+class SymbolARGLIST(Symbol, Iterable[SymbolARGUMENT]):
     """Defines a list of arguments in a function call or array access"""
 
     @property
@@ -42,15 +44,17 @@ def __repr__(self):
     def __len__(self):
         return len(self.args)
 
+    def __iter__(self):
+        return iter(self.args)
+
     @classmethod
-    def make_node(cls, node, *args):
+    def make_node(cls, node: SymbolARGLIST | None, *args: SymbolARGUMENT):
         """This will return a node with an argument_list."""
         if node is None:
             node = cls()
 
         assert isinstance(node, SymbolARGUMENT) or isinstance(node, cls)
-
-        if not isinstance(node, cls):
+        if isinstance(node, SymbolARGUMENT):
             return cls.make_node(None, node, *args)
 
         for arg in args:

src/symbols/paramlist.py

@@ -1,19 +1,19 @@
 #!/usr/bin/python
 # -*- coding: utf-8 -*-
-# vim: ts=4:et:sw=4:
 
 # ----------------------------------------------------------------------
 # Copyleft (K), Jose M. Rodriguez-Rosa (a.k.a. Boriel)
 #
 # This program is Free Software and is released under the terms of
 #                    the GNU General License
 # ----------------------------------------------------------------------
+from typing import Iterable
 
 from src.symbols.id_.interface import SymbolIdABC as SymbolID
 from src.symbols.symbol_ import Symbol
 
 
-class SymbolPARAMLIST(Symbol):
+class SymbolPARAMLIST(Symbol, Iterable[SymbolID]):
     """Defines a list of parameters definitions in a function header"""
 
     def __init__(self, *params):