1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
"""Utils."""
import sys
from astroid.bases import Instance
from astroid.exceptions import InferenceError
from astroid.nodes import ClassDef
from pylint_django.compat import Uninferable
PY3 = sys.version_info >= (3, 0)
def node_is_subclass(cls, *subclass_names):
"""Checks if cls node has parent with subclass_name."""
if not isinstance(cls, (ClassDef, Instance)):
return False
if cls.bases == Uninferable:
return False
for base_cls in cls.bases:
try:
for inf in base_cls.inferred():
if inf.qname() in subclass_names:
return True
if inf != cls and node_is_subclass(inf, *subclass_names):
# check up the hierarchy in case we are a subclass of
# a subclass of a subclass ...
return True
except InferenceError:
continue
return False
|
