mirror of
https://fuchsia.googlesource.com/third_party/github.com/pylint-dev/pylint
synced 2024-09-21 16:19:21 +00:00
e01fa86c00
Co-authored-by: Pierre Sassoulas <pierre.sassoulas@gmail.com>
53 lines
2.0 KiB
Python
53 lines
2.0 KiB
Python
# Licensed under the GPL: https://www.gnu.org/licenses/old-licenses/gpl-2.0.html
|
|
# For details: https://github.com/PyCQA/pylint/blob/main/LICENSE
|
|
# Copyright (c) https://github.com/PyCQA/pylint/blob/main/CONTRIBUTORS.txt
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from astroid import nodes
|
|
|
|
from pylint.checkers import BaseChecker
|
|
from pylint.checkers.utils import check_messages
|
|
from pylint.interfaces import IAstroidChecker
|
|
|
|
if TYPE_CHECKING:
|
|
from pylint.lint import PyLinter
|
|
|
|
|
|
class ConfusingConsecutiveElifChecker(BaseChecker):
|
|
"""Checks if "elif" is used right after an indented block that finishes with "if" or "elif" itself."""
|
|
|
|
__implements__ = IAstroidChecker
|
|
|
|
name = "confusing_elif"
|
|
msgs = {
|
|
"R5601": (
|
|
"Consecutive elif with differing indentation level, consider creating a function to separate the inner elif",
|
|
"confusing-consecutive-elif",
|
|
"Used when an elif statement follows right after an indented block which itself ends with if or elif. "
|
|
"It may not be ovious if the elif statement was willingly or mistakenly unindented. "
|
|
"Extracting the indented if statement into a separate function might avoid confusion and prevent errors.",
|
|
)
|
|
}
|
|
|
|
@check_messages("confusing-consecutive-elif")
|
|
def visit_if(self, node: nodes.If) -> None:
|
|
body_ends_with_if = isinstance(
|
|
node.body[-1], nodes.If
|
|
) and self._has_no_else_clause(node.body[-1])
|
|
if node.has_elif_block() and body_ends_with_if:
|
|
self.add_message("confusing-consecutive-elif", node=node.orelse[0])
|
|
|
|
@staticmethod
|
|
def _has_no_else_clause(node: nodes.If):
|
|
orelse = node.orelse
|
|
while orelse and isinstance(orelse[0], nodes.If):
|
|
orelse = orelse[0].orelse
|
|
if not orelse or isinstance(orelse[0], nodes.If):
|
|
return True
|
|
return False
|
|
|
|
|
|
def register(linter: "PyLinter") -> None:
|
|
linter.register_checker(ConfusingConsecutiveElifChecker(linter))
|