11
11
from collections .abc import Iterator
12
12
from typing import TYPE_CHECKING
13
13
14
- import astroid
15
14
from astroid import nodes
16
15
17
16
from pylint .checkers import BaseChecker
182
181
)
183
182
184
183
185
- def _is_exempt_from_public_methods (node : astroid .ClassDef ) -> bool :
184
+ def _is_exempt_from_public_methods (node : nodes .ClassDef ) -> bool :
186
185
"""Check if a class is exempt from too-few-public-methods."""
187
186
# If it's a typing.Namedtuple, typing.TypedDict or an Enum
188
187
for ancestor in node .ancestors ():
@@ -201,10 +200,10 @@ def _is_exempt_from_public_methods(node: astroid.ClassDef) -> bool:
201
200
202
201
root_locals = set (node .root ().locals )
203
202
for decorator in node .decorators .nodes :
204
- if isinstance (decorator , astroid .Call ):
203
+ if isinstance (decorator , nodes .Call ):
205
204
decorator = decorator .func
206
205
match decorator :
207
- case astroid .Name (name = name ) | astroid .Attribute (attrname = name ):
206
+ case nodes .Name (name = name ) | nodes .Attribute (attrname = name ):
208
207
pass
209
208
case _:
210
209
continue
@@ -227,7 +226,7 @@ def _count_boolean_expressions(bool_op: nodes.BoolOp) -> int:
227
226
"""
228
227
nb_bool_expr = 0
229
228
for bool_expr in bool_op .get_children ():
230
- if isinstance (bool_expr , astroid .BoolOp ):
229
+ if isinstance (bool_expr , nodes .BoolOp ):
231
230
nb_bool_expr += _count_boolean_expressions (bool_expr )
232
231
else :
233
232
nb_bool_expr += 1
@@ -662,7 +661,7 @@ def visit_if(self, node: nodes.If) -> None:
662
661
branches = 1
663
662
# don't double count If nodes coming from some 'elif'
664
663
if node .orelse and (
665
- len (node .orelse ) > 1 or not isinstance (node .orelse [0 ], astroid .If )
664
+ len (node .orelse ) > 1 or not isinstance (node .orelse [0 ], nodes .If )
666
665
):
667
666
branches += 1
668
667
self ._inc_branch (node , branches )
@@ -673,7 +672,7 @@ def _check_boolean_expressions(self, node: nodes.If) -> None:
673
672
if the 'if' node test is a BoolOp node.
674
673
"""
675
674
condition = node .test
676
- if not isinstance (condition , astroid .BoolOp ):
675
+ if not isinstance (condition , nodes .BoolOp ):
677
676
return
678
677
nb_bool_expr = _count_boolean_expressions (condition )
679
678
if nb_bool_expr > self .linter .config .max_bool_expr :
0 commit comments