如何在django创建用户时触发一个函数?

Question

我是django的新手，我想知道是否有可能在创建用户后触发函数?对于创建用户，我使用django管理界面。

我想给一个用户分配一个默认的主题，要做到这一点，我有以下类在我的models.py文件

class Profile(models.Model):
user = models.OneToOneField(User, null=False, on_delete=models.CASCADE)
selected_theme = models.TextField(max_length=50, default="Clair")

该类允许我存储用户选择的主题我想使用一个函数，在创建主题时将主题分配给用户。

Answer 1

为此，您需要将以下几行添加到概要文件类中的model.py文件中:

from django.db.models.signals import post_save
from django.dispatch import receiver
class Profile(models.Model):
user = models.OneToOneField(User, null=False, on_delete=models.CASCADE)
selected_theme = models.TextField(max_length=50, default="Clair")
@receiver(post_save, sender=User)
def create_user_profile(sender, instance, created, **kwargs):
if created:
Profile.objects.create(user=instance)
@receiver(post_save, sender=User)
def save_user_profile(sender, instance, **kwargs):
instance.profile.save()