如何为Django自定义User模型的两类用户配置信号？

Got it, let's walk through setting up Django signals for your custom User model to handle the automatic creation of those two one-to-one related profiles (I'll assume they're TrainerProfile and ClientProfile based on your is_trainer/is_client flags). Here's a step-by-step breakdown:

If you haven't already, here's an example of what those associated models might look like (tweak the fields to match your actual requirements):

from django.db import models
from django.contrib.auth import get_user_model

User = get_user_model()

class TrainerProfile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE, related_name='trainer_profile')
    # Add trainer-specific fields here, e.g.:
    certification = models.CharField(max_length=100, blank=True)
    bio = models.TextField(blank=True)

class ClientProfile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE, related_name='client_profile')
    # Add client-specific fields here, e.g.:
    fitness_goal = models.CharField(max_length=200, blank=True)
    emergency_contact = models.CharField(max_length=100, blank=True)

We'll use Django's post_save signal to trigger profile creation whenever a User is saved, and handle updates if the user's type (trainer/client) changes.

First, create a signals.py file in your app directory with this code:

from django.db.models.signals import post_save
from django.dispatch import receiver
from django.contrib.auth import get_user_model
from .models import TrainerProfile, ClientProfile

User = get_user_model()

@receiver(post_save, sender=User)
def create_or_update_user_profile(sender, instance, created, **kwargs):
    # Create profiles when a new user is created
    if created:
        if instance.is_trainer:
            TrainerProfile.objects.create(user=instance)
        if instance.is_client:
            ClientProfile.objects.create(user=instance)
    # Update profiles if the user's type changes
    else:
        # Handle trainer profile
        if instance.is_trainer:
            TrainerProfile.objects.get_or_create(user=instance)
        else:
            # Optional: Delete the trainer profile if user is no longer a trainer
            TrainerProfile.objects.filter(user=instance).delete()
        
        # Handle client profile
        if instance.is_client:
            ClientProfile.objects.get_or_create(user=instance)
        else:
            # Optional: Delete the client profile if user is no longer a client
            ClientProfile.objects.filter(user=instance).delete()

Then, you need to register these signals so Django picks them up. Open your app's apps.py file and update the ready() method:

from django.apps import AppConfig

class YourAppConfig(AppConfig):
    default_auto_field = 'django.db.models.BigAutoField'
    name = 'your_app_name'

    def ready(self):
        import your_app_name.signals  # Replace with your actual app name

Don't forget to update your settings.py to use this app config if you haven't already:

INSTALLED_APPS = [
    # ... other apps
    'your_app_name.apps.YourAppConfig',  # Instead of just 'your_app_name'
]

• Create a new user via the admin or a view, setting either is_trainer or is_client (or both, if that's allowed in your logic)
• Check your database: the corresponding TrainerProfile or ClientProfile should be automatically created
• Try updating an existing user's is_trainer/is_client flags and save— the profiles should be created or deleted accordingly (depending on whether you kept the optional delete logic)

A quick note: If you want to prevent users from being both trainer and client, you might want to add validation in your User model's clean() method to enforce that only one flag is set at a time.

内容的提问来源于stack exchange，提问作者Jekson