Mini Shell

"""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")