diff options
author | Claudiu Popa <pcmanticore@gmail.com> | 2014-10-25 23:55:16 +0300 |
---|---|---|
committer | Claudiu Popa <pcmanticore@gmail.com> | 2014-10-25 23:55:16 +0300 |
commit | 1486f9dad2c9cd2527ecc113b5cb593547d37fdc (patch) | |
tree | 87e0b8480fbee35bd96acb488b5cef68d4899db7 /checkers/classes.py | |
parent | 56f4b2abc332627ef8602bcc7b47e092f9bd12ae (diff) | |
download | pylint-git-1486f9dad2c9cd2527ecc113b5cb593547d37fdc.tar.gz |
Use a defaultdict where it makes sense.
Diffstat (limited to 'checkers/classes.py')
-rw-r--r-- | checkers/classes.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/checkers/classes.py b/checkers/classes.py index 497e7da28..4895f18f3 100644 --- a/checkers/classes.py +++ b/checkers/classes.py @@ -18,6 +18,7 @@ from __future__ import generators import sys +from collections import defaultdict import astroid from astroid import YES, Instance, are_exclusive, AssAttr, Class @@ -305,7 +306,7 @@ a metaclass class method.'} def visit_class(self, node): """init visit variable _accessed and check interfaces """ - self._accessed.append({}) + self._accessed.append(defaultdict(list)) self._check_bases_classes(node) self._check_interfaces(node) # if not an interface, exception, metaclass @@ -551,7 +552,7 @@ a metaclass class method.'} attrname = node.attrname # Check self if self.is_first_attr(node): - self._accessed[-1].setdefault(attrname, []).append(node) + self._accessed[-1][attrname].append(node) return if not self.linter.is_message_enabled('protected-access'): return @@ -560,7 +561,7 @@ a metaclass class method.'} def visit_assattr(self, node): if isinstance(node.ass_type(), astroid.AugAssign) and self.is_first_attr(node): - self._accessed[-1].setdefault(node.attrname, []).append(node) + self._accessed[-1][node.attrname].append(node) self._check_in_slots(node) def _check_in_slots(self, node): |