21

I'm building my first django app. I have a user, and the user has a list of favourites. A user has exactly one list of favourites, and that list belongs exclusively to that user.

class User(models.Model):
    name = models.CharField(max_length=200)

class FavouriteList(models.Model):
    user = models.OneToOneField(User)
    favourites = models.ManyToManyField(Favourite, blank=True)

When a new user is created, I want to ensure that the user has a FavouriteList. I've looked around in the Django documentation and haven't had much luck.

Does anyone know how I can ensure that a model has a child object (e.g. FavouriteList) when it is created?

NT3RP
  • 14,824
  • 9
  • 58
  • 97

3 Answers3

42

The most common way to accomplish this is to use the Django signals system. You can attach a signal handler (just a function somewhere) to the post_save signal for the User model, and create your favorites list inside of that callback.

from django.db.models.signals import post_save
from django.dispatch import receiver
from django.contrib.auth.models import User

@receiver(post_save, sender=User)
def create_favorites(sender, instance, created, **kwargs):
    if created:
        Favorites.objects.create(user=instance)

The above was adapted from the Django signals docs. Be sure to read the signals docs entirely because there are a few issues that can snag you such as where your signal handler code should live and how to avoid duplicate handlers.

TankorSmash
  • 11,649
  • 6
  • 62
  • 103
shadfc
  • 5,456
  • 3
  • 24
  • 19
  • 1
    updated the docs from 1.3 to 1.11. the `dev` docs would never be out of date, but maybe that would have been the better edit – TankorSmash Apr 11 '17 at 20:18
7

An excellent django-annoying plugin solves this for you with AutoOneToOneField. I'm using it for user profiles in all django projects I work on.

kkonrad
  • 1,232
  • 12
  • 30
Alexander Lebedev
  • 5,852
  • 1
  • 19
  • 29
5

One way would be to register a post_save signal with next handler:

def user_save(sender, instance, created, **kwargs):
  if created:
     FavouriteList.objects.get_or_create(user=instance)
Dmitry Shevchenko
  • 30,416
  • 10
  • 55
  • 62