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
- Crear una nova aplicació
Primer, creem una nova aplicació per gestionar els usuaris:
- 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
- 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:
- Crear i Aplicar Migracions
Generem les migracions per al nostre nou model d'usuari i les apliquem:
- 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')
- 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
- Modifica el model
CustomUserper incloure un campphone_number. - Actualitza els formularis d'autenticació per incloure el nou camp.
- 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.email2. 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_number3. Aplicar les Migracions
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ó.
Curs de Desenvolupament Web amb Django
Mòdul 1: Introducció a Django
- Què és Django?
- Configuració de l'Entorn de Desenvolupament
- Creació del teu Primer Projecte Django
- Comprensió de l'Estructura del Projecte Django
Mòdul 2: Conceptes Bàsics de Django
- Aplicacions Django i Estructura del Projecte
- Enrutament d'URL i Vistes
- Plantilles i Fitxers Estàtics
- Models i Bases de Dades
- Interfície d'Administració de Django
Mòdul 3: Django Intermedi
Mòdul 4: Django Avançat
- Consultes Avançades amb Django ORM
- Models d'Usuari Personalitzats
- Senyalitzacions de Django
- Proves en Django
- Optimització del Rendiment
