|
@@ -4,7 +4,9 @@ from django.contrib.auth.models import (
|
|
|
AbstractBaseUser,
|
|
|
AbstractUser,
|
|
|
UserManager,
|
|
|
- PermissionsMixin
|
|
|
+ PermissionsMixin,
|
|
|
+ Group,
|
|
|
+ Permission,
|
|
|
)
|
|
|
|
|
|
|
|
@@ -81,6 +83,20 @@ class CustomUser(AbstractBaseUser):
|
|
|
return self.is_admin
|
|
|
|
|
|
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+old_au_local_m2m = AbstractUser._meta.local_many_to_many
|
|
|
+old_pm_local_m2m = PermissionsMixin._meta.local_many_to_many
|
|
|
+groups = models.ManyToManyField(Group, blank=True)
|
|
|
+groups.contribute_to_class(PermissionsMixin, "groups")
|
|
|
+user_permissions = models.ManyToManyField(Permission, blank=True)
|
|
|
+user_permissions.contribute_to_class(PermissionsMixin, "user_permissions")
|
|
|
+PermissionsMixin._meta.local_many_to_many = [groups, user_permissions]
|
|
|
+AbstractUser._meta.local_many_to_many = [groups, user_permissions]
|
|
|
+
|
|
|
+
|
|
|
|
|
|
|
|
|
|
|
@@ -178,3 +194,7 @@ class CustomUserBadRequiredFields(AbstractBaseUser):
|
|
|
|
|
|
class Meta:
|
|
|
app_label = 'auth'
|
|
|
+
|
|
|
+
|
|
|
+AbstractUser._meta.local_many_to_many = old_au_local_m2m
|
|
|
+PermissionsMixin._meta.local_many_to_many = old_pm_local_m2m
|