41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
|
from django.contrib.auth.models import AbstractBaseUser, BaseUserManager
|
||
|
|
from django.db import models
|
||
|
|
|
||
|
|
|
||
|
|
class Organization(models.Model):
|
||
|
|
name = models.CharField(max_length=255)
|
||
|
|
|
||
|
|
|
||
|
|
class CustomUserWithM2MManager(BaseUserManager):
|
||
|
|
def create_superuser(self, username, orgs, password):
|
||
|
|
user = self.model(username=username)
|
||
|
|
user.set_password(password)
|
||
|
|
user.save(using=self._db)
|
||
|
|
user.orgs.add(*orgs)
|
||
|
|
return user
|
||
|
|
|
||
|
|
|
||
|
|
class CustomUserWithM2M(AbstractBaseUser):
|
||
|
|
username = models.CharField(max_length=30, unique=True)
|
||
|
|
orgs = models.ManyToManyField(Organization)
|
||
|
|
|
||
|
|
custom_objects = CustomUserWithM2MManager()
|
||
|
|
|
||
|
|
USERNAME_FIELD = 'username'
|
||
|
|
REQUIRED_FIELDS = ['orgs']
|
||
|
|
|
||
|
|
|
||
|
|
class CustomUserWithM2MThrough(AbstractBaseUser):
|
||
|
|
username = models.CharField(max_length=30, unique=True)
|
||
|
|
orgs = models.ManyToManyField(Organization, through='Membership')
|
||
|
|
|
||
|
|
custom_objects = CustomUserWithM2MManager()
|
||
|
|
|
||
|
|
USERNAME_FIELD = 'username'
|
||
|
|
REQUIRED_FIELDS = ['orgs']
|
||
|
|
|
||
|
|
|
||
|
|
class Membership(models.Model):
|
||
|
|
user = models.ForeignKey(CustomUserWithM2MThrough, on_delete=models.CASCADE)
|
||
|
|
organization = models.ForeignKey(Organization, on_delete=models.CASCADE)
|