diff options
author | Marc Mueller <30130371+cdce8p@users.noreply.github.com> | 2021-08-19 12:11:32 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-08-19 12:11:32 +0200 |
commit | 82b991ec9fb683f19bb1a50683e98e0017019573 (patch) | |
tree | 1dfba1ae9fdda79ef0d6286d4088237c6cfc2fa8 | |
parent | d8069ddc9f1959d286bc8613a2cd017e540eefd1 (diff) | |
download | pylint-git-82b991ec9fb683f19bb1a50683e98e0017019573.tar.gz |
Refactor existing code - code_style extension (#4872)
-rw-r--r-- | pylint/extensions/code_style.py | 13 |
1 files changed, 4 insertions, 9 deletions
diff --git a/pylint/extensions/code_style.py b/pylint/extensions/code_style.py index 9fff904b9..3f400fd18 100644 --- a/pylint/extensions/code_style.py +++ b/pylint/extensions/code_style.py @@ -53,11 +53,13 @@ class CodeStyleChecker(BaseChecker): @check_messages("consider-using-tuple") def visit_for(self, node: nodes.For) -> None: - self._check_inplace_defined_list(node) + if isinstance(node.iter, nodes.List): + self.add_message("consider-using-tuple", node=node.iter) @check_messages("consider-using-tuple") def visit_comprehension(self, node: nodes.Comprehension) -> None: - self._check_inplace_defined_list(node) + if isinstance(node.iter, nodes.List): + self.add_message("consider-using-tuple", node=node.iter) def _check_dict_consider_namedtuple_dataclass(self, node: nodes.Dict) -> None: """Check if dictionary values can be replaced by Namedtuple or Dataclass.""" @@ -133,13 +135,6 @@ class CodeStyleChecker(BaseChecker): self.add_message("consider-using-namedtuple-or-dataclass", node=node) return - def _check_inplace_defined_list( - self, node: Union[nodes.For, nodes.Comprehension] - ) -> None: - """Check if in-place defined list can be replaced by a tuple.""" - if isinstance(node.iter, nodes.List): - self.add_message("consider-using-tuple", node=node.iter) - def register(linter: PyLinter) -> None: linter.register_checker(CodeStyleChecker(linter)) |