Suppress not-an-iterable message for model_utils.managers. Fixes #117

Author: atodorov

pylint_django/augmentations/__init__.py

@@ -11,7 +11,7 @@
 from pylint.checkers.newstyle import NewStyleConflictChecker
 from pylint.checkers.variables import VariablesChecker
 from pylint.__pkginfo__ import numversion as PYLINT_VERSION
-from pylint.checkers.typecheck import TypeChecker
+from pylint.checkers.typecheck import IterableChecker, TypeChecker
 from pylint.checkers.variables import ScopeConsumer
 
 from pylint_plugin_utils import augment_visit, suppress_message
@@ -462,6 +462,29 @@ def is_manager_attribute(node):
     return _attribute_is_magic(node, MANAGER_ATTRS.union(QS_ATTRS), parents)
 
 
+def is_model_manager(node):
+    """Checks that node is an attribute named `objects` which usually represents
+
+       This function fixes issues #117 and #144.
+    """
+    try:
+        # try to get the model manager node, i.e. 'objects'
+        manager = node.func.expr
+    except:  # noqa: E722, pylint: disable=bare-except
+        return False
+
+    parents = ('django.db.models.manager.Manager',
+               '.Manager',
+               'django.db.models.query.QuerySet',
+               '.QuerySet',
+               'model_utils.managers.InheritanceManager',
+               'model_utils.managers.QueryManager',
+               'model_utils.managers.SoftDeletableManager',
+              )
+    return _attribute_is_magic(manager, ['objects'], parents)
+    #return isinstance(manager, Attribute) and manager.attrname == 'objects'
+
+
 def is_admin_attribute(node):
     """Checks that node is attribute of BaseModelAdmin."""
     parents = ('django.contrib.admin.options.BaseModelAdmin',
@@ -795,3 +818,6 @@ def apply_augmentations(linter):
         VariablesChecker.leave_module = wrap(current_leave_module, ignore_import_warnings_for_related_fields)
         # VariablesChecker.leave_module is now wrapped
     # else VariablesChecker.leave_module is already wrapped
+
+    # supress not-an-iterable for model_utils.managers
+    suppress_message(linter, IterableChecker._check_iterable, 'not-an-iterable', is_model_manager)