diff --git a/pylint_django/augmentations/__init__.py b/pylint_django/augmentations/__init__.py index a6c2f63d..fe9a6ca8 100644 --- a/pylint_django/augmentations/__init__.py +++ b/pylint_django/augmentations/__init__.py @@ -248,6 +248,8 @@ 'clear', 'related_name', 'related_query_name', + 'remove', + 'set', 'limit_choices_to', 'symmetrical', 'through', diff --git a/pylint_django/tests/input/func_noerror_manytomanyfield.py b/pylint_django/tests/input/func_noerror_manytomanyfield.py index d1c6d967..c6ab9719 100644 --- a/pylint_django/tests/input/func_noerror_manytomanyfield.py +++ b/pylint_django/tests/input/func_noerror_manytomanyfield.py @@ -45,6 +45,15 @@ def grant_permissions(self): self.user_permissions.add(perm) return self.user_permissions + def add_permission(self, permission): + self.user_permissions.add(permission) + + def remove_permission(self, permission): + self.user_permissions.remove(permission) + + def set_permissions(self, permissions): + self.user_permissions.set(permissions) + def save(self, *args, **kwargs): ''' Saving while granting new permissions ''' self.is_staff = True