En aquest tema, aprendrem com crear models d'usuari personalitzats en Django. Això és especialment útil quan necessitem afegir camps addicionals o modificar el comportament per defecte del model d'usuari proporcionat per Django.

Objectius

  • Comprendre la necessitat de models d'usuari personalitzats.
  • Aprendre a crear un model d'usuari personalitzat.
  • Configurar Django per utilitzar el model d'usuari personalitzat.
  • Implementar un gestor d'usuaris personalitzat.

Per què utilitzar models d'usuari personalitzats?

El model d'usuari per defecte de Django (User) és suficient per a moltes aplicacions, però pot ser limitant si necessitem:

  • Afegir camps addicionals (per exemple, data de naixement, número de telèfon).
  • Modificar el comportament per defecte (per exemple, canviar el camp de nom d'usuari per defecte a correu electrònic).

Creació d'un Model d'Usuari Personalitzat

  1. Crear una nova aplicació

Primer, creem una nova aplicació per gestionar els usuaris:

python manage.py startapp accounts

  1. Definir el Model d'Usuari Personalitzat

A accounts/models.py, definim el nostre model d'usuari personalitzat:

from django.contrib.auth.models import AbstractBaseUser, BaseUserManager, PermissionsMixin
from django.db import models

class CustomUserManager(BaseUserManager):
    def create_user(self, email, password=None, **extra_fields):
        if not email:
            raise ValueError('The Email field 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_superuser(self, email, password=None, **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(AbstractBaseUser, PermissionsMixin):
    email = models.EmailField(unique=True)
    first_name = models.CharField(max_length=30, blank=True)
    last_name = models.CharField(max_length=30, blank=True)
    is_active = models.BooleanField(default=True)
    is_staff = models.BooleanField(default=False)
    date_joined = models.DateTimeField(auto_now_add=True)

    objects = CustomUserManager()

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = []

    def __str__(self):
        return self.email

  1. Configurar Django per utilitzar el Model d'Usuari Personalitzat

A settings.py, afegim la següent línia per indicar a Django que utilitzi el nostre model d'usuari personalitzat:

AUTH_USER_MODEL = 'accounts.CustomUser'

  1. Crear i Aplicar Migracions

Generem les migracions per al nostre nou model d'usuari i les apliquem:

python manage.py makemigrations
python manage.py migrate

  1. Actualitzar els Formularis d'Autenticació

Si estem utilitzant els formularis d'autenticació per defecte de Django, necessitarem actualitzar-los per utilitzar el nostre model d'usuari personalitzat. A accounts/forms.py:

from django import forms
from django.contrib.auth.forms import UserCreationForm, UserChangeForm
from .models import CustomUser

class CustomUserCreationForm(UserCreationForm):
    class Meta:
        model = CustomUser
        fields = ('email', 'first_name', 'last_name')

class CustomUserChangeForm(UserChangeForm):
    class Meta:
        model = CustomUser
        fields = ('email', 'first_name', 'last_name')

  1. Registrar el Model d'Usuari Personalitzat a l'Admin

Finalment, registrem el nostre model d'usuari personalitzat a l'administració de Django. A accounts/admin.py:

from django.contrib import admin
from django.contrib.auth.admin import UserAdmin
from .forms import CustomUserCreationForm, CustomUserChangeForm
from .models import CustomUser

class CustomUserAdmin(UserAdmin):
    add_form = CustomUserCreationForm
    form = CustomUserChangeForm
    model = CustomUser
    list_display = ['email', 'first_name', 'last_name', 'is_staff']
    fieldsets = (
        (None, {'fields': ('email', 'password')}),
        ('Personal info', {'fields': ('first_name', 'last_name')}),
        ('Permissions', {'fields': ('is_active', 'is_staff', 'is_superuser', 'groups', 'user_permissions')}),
        ('Important dates', {'fields': ('last_login', 'date_joined')}),
    )
    add_fieldsets = (
        (None, {
            'classes': ('wide',),
            'fields': ('email', 'password1', 'password2', 'is_staff', 'is_active')}
        ),
    )
    search_fields = ('email',)
    ordering = ('email',)

admin.site.register(CustomUser, CustomUserAdmin)

Exercici Pràctic

Objectiu

Crear un model d'usuari personalitzat que inclogui un camp addicional per al número de telèfon.

Passos

  1. Modifica el model CustomUser per incloure un camp phone_number.
  2. Actualitza els formularis d'autenticació per incloure el nou camp.
  3. Aplica les migracions necessàries.

Solució

1. Modificar el Model CustomUser

A accounts/models.py:

class CustomUser(AbstractBaseUser, PermissionsMixin):
    email = models.EmailField(unique=True)
    first_name = models.CharField(max_length=30, blank=True)
    last_name = models.CharField(max_length=30, blank=True)
    phone_number = models.CharField(max_length=15, blank=True)  # Nou camp
    is_active = models.BooleanField(default=True)
    is_staff = models.BooleanField(default=False)
    date_joined = models.DateTimeField(auto_now_add=True)

    objects = CustomUserManager()

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = []

    def __str__(self):
        return self.email

2. Actualitzar els Formularis d'Autenticació

A accounts/forms.py:

class CustomUserCreationForm(UserCreationForm):
    class Meta:
        model = CustomUser
        fields = ('email', 'first_name', 'last_name', 'phone_number')  # Afegir phone_number

class CustomUserChangeForm(UserChangeForm):
    class Meta:
        model = CustomUser
        fields = ('email', 'first_name', 'last_name', 'phone_number')  # Afegir phone_number

3. Aplicar les Migracions

python manage.py makemigrations
python manage.py migrate

Conclusió

En aquesta secció, hem après com crear i configurar un model d'usuari personalitzat en Django. Això ens permet afegir camps addicionals i modificar el comportament per defecte del model d'usuari de Django per adaptar-lo a les necessitats específiques de la nostra aplicació.

© Copyright 2024. Tots els drets reservats