[PYTHON] django-allauth gère plusieurs types d'utilisateurs à l'aide d'un utilisateur personnalisé (utilisateur multi-type)

Aperçu

--Je veux un système capable de créer plusieurs types d'utilisateurs avec Django

Restrictions Python + Django

--Une seule classe de modèle peut être utilisée pour l'authentification des utilisateurs (connexion et inscription) --Spécifié par ʻAUTH_USER_MODEL` dans settings.py

Aperçu de la mise en œuvre

--Créer un modèle CustomUser et avoir userType --Avoir des informations pour chaque type d'utilisateur dans une table séparée et les lier à une classe d'utilisateurs personnalisée avec ʻOneToOneField. - UserDetailSupplier - UserDetailBuyer --Utilisez un adaptateur pour enregistrer l'utilisateur. --Créez un AccountAdapter en héritant de ʻallauth.account.adapter.DefaultAccountAdapter

Relation entre django-allauth et classe self-made

--django-allauth utilise le nom de l'application ʻaccount, mais crée séparément une application appelée membre` et la gère là.

problème

Exemple spécifique de mise en œuvre

Installation de django-allauth

--Installez avec pip et effectuez divers réglages

Configuration

--Ajoutez ce qui suit au fichier de configuration


# settings.py

#Spécifiez le modèle à utiliser pour l'authentification
AUTH_USER_MODEL = 'member.CustomUser'
#Spécifiez un adaptateur pour enregistrer les informations du formulaire d'inscription dans customusermodel
ACCOUNT_ADAPTER = 'member.adapter.AccountAdapter'

Créer une classe de modèle


# member/models.py

from django.contrib.auth.models import AbstractUser
from django.db import models
from django.contrib.auth.models import PermissionsMixin, UserManager

class UserType(models.Model):
    """Type d'utilisateur"""
    typename = models.CharField(verbose_name='Type d'utilisateur',
                                max_length=150)

    def __str__(self):
        return f'{self.id} - {self.typename}'

USERTYPE_SUPPLIER = 100
USERTYPE_BUYER = 200
USERTYPE_DEFAULT = USERTYPE_BUYER

class CustomUserManager(UserManager):
    """Gestionnaire pour modèle utilisateur étendu"""

    def _create_user(self, email, password, **extra_fields):
        if not email:
            raise ValueError('The given email must be set')
        email = self.normalize_email(email)
        user = self.model(email=email, **extra_fields)
        user.set_password(password)
        user.save(using=self._db)
        return user

    def create_user(self, email, password=None, **extra_fields):
        extra_fields.setdefault('is_staff', False)
        extra_fields.setdefault('is_superuser', False)
        return self._create_user(email, password, **extra_fields)

    def create_superuser(self, email, password, **extra_fields):
        extra_fields.setdefault('is_staff', True)
        extra_fields.setdefault('is_superuser', True)
        if extra_fields.get('is_staff') is not True:
            raise ValueError('Superuser must have is_staff=True.')
        if extra_fields.get('is_superuser') is not True:
            raise ValueError('Superuser must have is_superuser=True.')
        return self._create_user(email, password, **extra_fields)


class CustomUser(AbstractUser):
    """Modèle utilisateur étendu"""

    class Meta(object):
        db_table = 'custom_user'

    #Utilisez la classe de gestionnaire créée
    objects = CustomUserManager()

    #Demandez à un utilisateur de saisir le modèle
    userType = models.ForeignKey(UserType,
                                verbose_name='Type d'utilisateur',
                                null=True,
                                blank=True,
                                on_delete=models.PROTECT)
    def __str__(self):
        return self.username

class UserDetailSupplier(models.Model):
    user = models.OneToOneField(CustomUser,
                                unique=True,
                                db_index=True,
                                related_name='detail_supplier',
                                on_delete=models.CASCADE)
    #Articles pour les utilisateurs fournisseurs
    companyName = models.CharField(
                                   max_length=100,
                                   null=True,
                                   blank=True,
                                )
    def __str__(self):
        user = CustomUser.objects.get(pk=self.user_id)
        return f'{user.id} - {user.username} - {user.email} - {self.id} - {self.companyName}'

class UserDetailBuyer(models.Model):
    user = models.OneToOneField(CustomUser,
                                unique=True,
                                db_index=True,
                                related_name='detail_buyer',
                                on_delete=models.CASCADE)
    #Articles pour les utilisateurs acheteurs
    nearestStation = models.CharField(
                                   max_length=100,
                                   null=True,
                                   blank=True,
                                )
    def __str__(self):
        user = CustomUser.objects.get(pk=self.user_id)
        return f'{user.id} - {user.username} - {user.email} - {self.id} - {self.nearestStation}'

Créer un adaptateur


# member/adapter.py

from allauth.account.adapter import DefaultAccountAdapter
from .models import *

class AccountAdapter(DefaultAccountAdapter):

    def save_user(self, request, user, form, commit=True):
        """
        This is called when saving user via allauth registration.
        We override this to set additional data on user object.
        """
        # Do not persist the user yet so we pass commit=False
        # (last argument)
        user = super(AccountAdapter, self).save_user(request, user, form, commit=False)
        #user.userType = form.cleaned_data.get('userType')
        user.userType = UserType(request.POST['userType'])

        if not user.userType:
            user.userType = UserType(USERTYPE_DEFAULT) #Définir le type d'utilisateur par défaut

        #Enregistrez une fois pour obtenir l'ID utilisateur
        user.save()

        if int(user.userType.id) == USERTYPE_SUPPLIER:
            #Utilisateur fournisseur
            supplier = UserDetailSupplier()
            supplier.user_id = user.id
            supplier.companyName = request.POST['companyName']
            supplier.save()
        else:
            #Autre que cela, les utilisateurs généraux
            user.userType = UserType(USERTYPE_BUYER)
            buyer = UserDetailBuyer()
            buyer.user_id = user.id
            buyer.nearestStation = request.POST.get('nearestStation', False)
            buyer.save()

Créer un modèle

paramètres urls.py

--Ajout de ce qui suit


    path('member/signup_supplier/', TemplateView.as_view(template_name = 'account/signup_supplier.html'), name='signup_supplier'),

Autre

--Ajoutez chaque classe de modèle à admin.py

référence

--Rechercher "django-allauth userType" sur Google https://www.google.com/search?q=django-allauth+userType&oq=django-allauth+userType&aqs=chrome..69i57.9683j0j1&sourceid=chrome&ie=UTF-8

Recommended Posts

django-allauth gère plusieurs types d'utilisateurs à l'aide d'un utilisateur personnalisé (utilisateur multi-type)
Créer une fonction d'authentification à l'aide de django-allauth et CustomUser dans Django
Attrapez plusieurs types d'exceptions