Merge pull request #919 from cdce8p/fix-classdef-keywords

Iterate over Keywords when using ClassDef.get_children

Author: hippo91

ChangeLog

@@ -15,6 +15,10 @@ Release Date: TBA
 
   Closes PyCQA/pylint#3974
 
+* Iterate over ``Keywords`` when using ``ClassDef.get_children``
+
+  Closes PyCQA/pylint#3202
+
 
 What's New in astroid 2.5.1?
 ============================

astroid/scoped_nodes.py

@@ -1895,7 +1895,7 @@ def my_meth(self, arg):
     # by a raw factories
 
     # a dictionary of class instances attributes
-    _astroid_fields = ("decorators", "bases", "body")  # name
+    _astroid_fields = ("decorators", "bases", "keywords", "body")  # name
 
     decorators = None
     """The decorators that are applied to this class.
@@ -2920,6 +2920,8 @@ def get_children(self):
             yield self.decorators
 
         yield from self.bases
+        if self.keywords is not None:
+            yield from self.keywords
         yield from self.body
 
     @decorators_mod.cached

tests/unittest_scoped_nodes.py

@@ -1758,6 +1758,12 @@ class TestKlass(object, metaclass=TestMetaKlass,
         cls = astroid["TestKlass"]
         self.assertEqual(len(cls.keywords), 2)
         self.assertEqual([x.arg for x in cls.keywords], ["foo", "bar"])
+        children = list(cls.get_children())
+        assert len(children) == 4
+        assert isinstance(children[1], nodes.Keyword)
+        assert isinstance(children[2], nodes.Keyword)
+        assert children[1].arg == "foo"
+        assert children[2].arg == "bar"
 
     def test_kite_graph(self):
         data = """