| 12345678910111213141516171819202122232425262728293031323334353637383940 |
- """Utils."""
- import sys
- import astroid
- 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) # TODO: pylint_django doesn't support Py2 any more
- 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
- def is_migrations_module(node):
- if not isinstance(node, astroid.Module):
- return False
- return "migrations" in node.path[0] and not node.path[0].endswith("__init__.py")
|
