diff options
Diffstat (limited to 'astroid/tree/scoped_nodes.py')
| -rw-r--r-- | astroid/tree/scoped_nodes.py | 40 |
1 files changed, 30 insertions, 10 deletions
diff --git a/astroid/tree/scoped_nodes.py b/astroid/tree/scoped_nodes.py index ae8da65f..c11eb4ce 100644 --- a/astroid/tree/scoped_nodes.py +++ b/astroid/tree/scoped_nodes.py @@ -2030,16 +2030,7 @@ class ClassDef(mixins.FilterStmtsMixin, LocalsDictNodeNG, except exceptions.InferenceError: continue - # Cached, because inferring them all the time is expensive - @decorators_mod.cached - def slots(self): - """Get all the slots for this node. - - If the class doesn't define any slot, through `__slots__` - variable, then this function will return a None. - Also, it will return None in the case the slots weren't inferred. - Otherwise, it will return a list of slot names. - """ + def _slots(self): if not self.newstyle: raise TypeError( "The concept of slots is undefined for old-style classes.") @@ -2055,6 +2046,35 @@ class ClassDef(mixins.FilterStmtsMixin, LocalsDictNodeNG, # pylint: disable=unsupported-binary-operation; false positive return [first] + list(slots) + # Cached, because inferring them all the time is expensive + @decorators_mod.cached + def slots(self): + """Get all the slots for this node. + + If the class doesn't define any slot, through `__slots__` + variable, then this function will return a None. + Also, it will return None in the case the slots weren't inferred. + Otherwise, it will return a list of slot names. + """ + def grouped_slots(): + # Not interested in object, since it can't have slots. + for cls in self.mro()[:-1]: + try: + cls_slots = cls._slots() + except NotImplementedError: + continue + if cls_slots is not None: + for slot in cls_slots: + yield slot + else: + yield None + + slots = list(grouped_slots()) + if not all(slot is not None for slot in slots): + return None + + return sorted(slots, key=lambda item: item.value) + def _inferred_bases(self, context=None): # TODO(cpopa): really similar with .ancestors, # but the difference is when one base is inferred, |