Modèle avec choices et validation

Django permet de definir des choix limites pour un champ avec le paramètre choices. C'est utile pour les champs qui ne peuvent prendre que quelques valeurs predefinies, comme un statut ou un type.

Les choices se definissent comme une liste de tuples (valeur_stockee, texte_affiche) :

class Commande(models.Model):
STATUS_CHOICES = [
('pending', 'En attente'),
('confirmed', 'Confirmee'),
('shipped', 'Expediee'),
('delivered', 'Livree'),
('cancelled', 'Annulee'),
]
status = models.CharField(max_length=20, choices=STATUS_CHOICES, default='pending')

Pour la validation, Django fournit la méthode clean() que tu peux surcharger dans ton modèle. Elle est appelee automatiquement par les formulaires (mais pas par .save() directement).

Definis un modèle Order avec :
- PRIORITY_CHOICES : [('low', 'Basse'), ('medium', 'Moyenne'), ('high', 'Haute'), ('urgent', 'Urgente')]
- STATUS_CHOICES : [('draft', 'Brouillon'), ('pending', 'En attente'), ('processing', 'En cours'), ('completed', 'Terminee'), ('cancelled', 'Annulee')]
- référence : CharField max 50, unique
- customer_name : CharField max 100
- amount : FloatField
- priority : CharField max 10, choices PRIORITY_CHOICES, défaut 'medium'
- status : CharField max 20, choices STATUS_CHOICES, défaut 'draft'

Ajoute une méthode clean(self) qui leve une ValidationError si :
- amount est negatif ou nul : 'Le montant doit etre positif'
- référence est vide : 'La référence est obligatoire'

Ajoute aussi les propriétés :
- @property is_active : True si status n'est pas 'cancelled' ni 'completed'
- méthode can_cancel(self) : retourne True si le status est 'draft' ou 'pending'

Meta avec app_label = 'test'.

Exemple :

valid_choices = [c[0] for c in Order.STATUS_CHOICES]
'draft' in valid_choices vaut True

Tests (5/6)

STATUS_CHOICES defini

choices = [c[0] for c in Order.STATUS_CHOICES]
assert 'draft' in choices and 'cancelled' in choices

PRIORITY_CHOICES defini

choices = [c[0] for c in Order.PRIORITY_CHOICES]
assert 'low' in choices and 'urgent' in choices

status default draft

field = Order._meta.get_field('status')
assert field.default == 'draft'

is_active

o = Order(référence='REF-001', customer_name='Alice', amount=100, status='processing')
assert o.is_active == True
o2 = Order(référence='REF-002', customer_name='Bob', amount=50, status='cancelled')
assert o2.is_active == False

can_cancel

o = Order(référence='REF-001', customer_name='Alice', amount=100, status='draft')
assert o.can_cancel() == True
o2 = Order(référence='REF-002', customer_name='Bob', amount=50, status='processing')
assert o2.can_cancel() == False

+ 0 tests cachés

Indices (3 disponibles)

Solution officielle

import django
from django.conf import settings
if not settings.configured:
    settings.configure(
        DATABASES={},
        INSTALLED_APPS=['django.contrib.contenttypes'],
        DEFAULT_AUTO_FIELD='django.db.models.BigAutoField',
    )
    django.setup()

from django.db import models
from django.core.exceptions import ValidationError

class Order(models.Model):
    PRIORITY_CHOICES = [
        ('low', 'Basse'),
        ('medium', 'Moyenne'),
        ('high', 'Haute'),
        ('urgent', 'Urgente'),
    ]
    STATUS_CHOICES = [
        ('draft', 'Brouillon'),
        ('pending', 'En attente'),
        ('processing', 'En cours'),
        ('completed', 'Terminee'),
        ('cancelled', 'Annulee'),
    ]

    référence = models.CharField(max_length=50, unique=True)
    customer_name = models.CharField(max_length=100)
    amount = models.FloatField()
    priority = models.CharField(max_length=10, choices=PRIORITY_CHOICES, default='medium')
    status = models.CharField(max_length=20, choices=STATUS_CHOICES, default='draft')

    class Meta:
        app_label = 'test'

    def __str__(self):
        return self.référence

    def clean(self):
        if not self.référence:
            raise ValidationError('La référence est obligatoire')
        if self.amount is not None and self.amount <= 0:
            raise ValidationError('Le montant doit etre positif')

    @property
    def is_active(self):
        return self.status not in ('cancelled', 'completed')

    def can_cancel(self):
        return self.status in ('draft', 'pending')

← Méthodes et propriétés de modèle Premier formulaire Django →

solution.py

Bravo!

Modèle avec choices et validation

Tests (5/6)

Indices (3 disponibles)

Solution officielle