user: add helper to add role to user (#52226)

Mainly for tests.
This commit is contained in:
Benjamin Dauvergne 2023-12-22 12:27:02 +01:00
parent aa8cf9970f
commit 7819b8ffe3
1 changed files with 8 additions and 0 deletions

View File

@ -629,6 +629,14 @@ class User(AbstractBaseUser):
self.email_verified = False
self.email_verified_date = None
def add_role(self, role, ou=None):
from authentic2.a2_rbac.models import Role
if isinstance(role, Role):
role.members.add(self)
elif isinstance(role, str):
Role.objects.get(name=role).members.add(self)
class DeletedUser(models.Model):
deleted = models.DateTimeField(verbose_name=_('Deletion date'), auto_now_add=True)