Django - Remplacer la méthode Model.create ()?

La documentation Django ne répertorie que des exemples de remplacement save()et delete(). Cependant, j'aimerais définir un traitement supplémentaire pour mes modèles uniquement lorsqu'ils sont créés . Pour quiconque connaît Rails, cela équivaudrait à créer un :before_createfiltre. Est-ce possible?

Le remplacement __init__()entraînerait l'exécution du code chaque fois que la représentation python de l'objet est instanciée. Je ne connais pas les rails, mais un :before_createdfiltre me semble être du code à exécuter lorsque l'objet est créé dans la base de données. Si vous souhaitez exécuter du code lorsqu'un nouvel objet est créé dans la base de données, vous devez le remplacer save(), en vérifiant si l'objet a un pkattribut ou non. Le code ressemblerait à ceci:

def save(self, *args, **kwargs):
    if not self.pk:
        # This code only happens if the objects is
        # not in the database yet. Otherwise it would
        # have pk
    super(MyModel, self).save(*args, **kwargs)

un exemple de création d'un signal post_save (à partir de http://djangosnippets.org/snippets/500/ )

from django.db.models.signals import post_save
from django.dispatch import receiver

@receiver(post_save, sender=User)
def create_profile(sender, instance, created, **kwargs):
    """Create a matching profile whenever a user object is created."""
    if created: 
        profile, new = UserProfile.objects.get_or_create(user=instance)

voici une discussion approfondie pour savoir s'il est préférable d'utiliser des signaux ou des méthodes de sauvegarde personnalisées https://web.archive.org/web/20120815022107/http://www.martin-geber.com/thought/2007/10/29/ django-signaux-vs-méthode-de-sauvegarde-personnalisée /

À mon avis, l'utilisation de signaux pour cette tâche est plus robuste, plus facile à lire mais plus longue.

C'est vieux, a une réponse acceptée qui fonctionne (celle de Zach), et une plus idiomatique aussi (celle de Michael Bylstra), mais comme c'est toujours le premier résultat sur Google que la plupart des gens voient, je pense que nous avons besoin d'un plus de bonnes pratiques modernes-django réponse de style ici :

from django.db.models.signals import post_save

class MyModel(models.Model):
    # ...
    @classmethod
    def post_create(cls, sender, instance, created, *args, **kwargs):
        if not created:
            return
        # ...what needs to happen on create

post_save.connect(MyModel.post_create, sender=MyModel)

Le point est le suivant:

1. utiliser des signaux (en savoir plus ici dans la documentation officielle )
2. utilisez une méthode pour un bel espace de noms (si cela a du sens) ... et je l'ai marqué comme @classmethodau lieu de @staticmethodparce que vous devrez probablement faire référence à des membres de classe statiques dans le code

Encore plus propre serait si le noyau Django avait un post_createsignal réel . (Imho si vous devez passer un argument booléen pour changer le comportement d'une méthode, cela devrait être 2 méthodes.)

Pour répondre littéralement à la question, la createméthode dans le gestionnaire d'un modèle est un moyen standard de créer de nouveaux objets dans Django. Pour remplacer, faites quelque chose comme

from django.db import models

class MyModelManager(models.Manager):
    def create(self, **obj_data):
        # Do some extra stuff here on the submitted data before saving...
        # For example...
        obj_data['my_field'] = my_computed_value(obj_data['my_other_field'])

        # Now call the super method which does the actual creation
        return super().create(**obj_data) # Python 3 syntax!!

class MyModel(models.model):
    # An example model
    my_field = models.CharField(max_length=250)
    my_other_field = models.CharField(max_length=250)

    objects = MyModelManager()

Dans cet exemple, je remplace la méthode de la méthode du gestionnaire createpour effectuer un traitement supplémentaire avant que l'instance ne soit réellement créée.

REMARQUE: Code comme

my_new_instance = MyModel.objects.create(my_field='my_field value')

exécutera cette createméthode modifiée , mais un code comme

my_new_unsaved_instance = MyModel(my_field='my_field value')

Ne fera pas.

Le remplacement __init__()vous permettra d'exécuter du code lorsque le modèle est instancié. N'oubliez pas d'appeler les parents __init__().

Vous pouvez remplacer la méthode create avec un gestionnaire personnalisé ou ajouter une méthode de classe sur la classe de modèle. https://docs.djangoproject.com/en/1.11/ref/models/instances/#creating-objects

La réponse préférée est correcte, mais le test pour savoir si l'objet est en cours de création ne fonctionne pas si votre modèle dérive de UUIDModel. Le champ pk aura déjà une valeur.

Dans ce cas, vous pouvez faire ceci:

already_created = MyModel.objects.filter(pk=self.pk).exists()