创建时自动更新模型字段

Question

我想在创建模型字段时自动更新它。所以这就是我的情况。我有一个自定义用户模型，我的客户可以使用它来登录。登录后，他们将转到包含表单链接的帐户/个人资料页面。目前，当用户提交表单时，它会创建一个 LevelTest 模型的实例（这是我网站运行所需要的）。这是表单的视图类：

class LevelTestView(generic.CreateView):
    template_name = "leads/leveltest.html"
    form_class = LevelTestModelForm

    def get_success_url(self):
        return reverse("profile-page")

这里是 LevelTestModelForm：

class LevelTestModelForm(forms.ModelForm):
    class Meta:
        model = LevelTest
        fields = (
            'first_name',
            'last_name',
            'age',
            'username',
        )

我要自动填写的是用户名字段。事实上，我希望当用户输入时它甚至不会出现在表单本身。username 是用户模型中的一个字段，所以我只希望新的 LevelTest 的username 字段填充当前用户的用户名。因此，我使用了如下所示的 post_save 信号（不起作用）：

def post_leveltest_created_signal(sender, instance, created, **kwargs):
    if created:
        instance.objects.update(
            username=instance.user.username,
            description='Add Description',
            phone_number=instance.user.cellphone,
            email=instance.user.username,
        )

    
post_save.connect(post_leveltest_created_signal, sender=LevelTest)

我希望你们能帮我调整一下 post_save 信号，这样当用户创建 LevelTest 实例时，LevelTest 的用户名字段（以及 phone_number 和电子邮件）都会填写用户模型的信息。非常感谢！

Answer 1

如果我理解你的正确，你不需要使用信号，你可以更容易地保存用户名：

在您的 CreateView 中扩展 get_form_kwargs 方法，如下所示：

 class LevelTestView:(generic.CreateView)
     ...
     def get_form_kwargs(self):
         kwargs = super().get_form_kwargs()
         kwargs['user'] = self.request.user
         return kwargs

在您的Form 中扩展__init__ 和save 方法，如下所示：

class LevelTestModelForm(forms.ModelForm):
    ...
    def __init__(self, user, *args, **kwargs):
        self.user = user
        super().__init__(*args, **kwargs)
    ...
    def save(self, commit=True):
        leveltest = super().save(commit=False)
        # I think it would be better if you saved only 'user' instance
        # like this - leveltest.user = self.user (of course if you have fk to user model)
        leveltest.username = self.user.username
        leveltest.phone_number=self.user.cellphone
        leveltest.email=self.user.username
        leveltest.save()
        return leveltest

Answer 2

我认为@KIN1991 的答案非常棒，但是您可以通过重写form_valid 方法来进一步最小化/优化代码。像这样：

class LevelTestView:(generic.CreateView)
    ...
    def form_valid(self, form, *args, **kwargs):
       user = self.request.user
       form.instance.username = user.username
       form.instance.phone_number=user.cellphone,
       form.instance.email=user.username
       return super().form_valid(form, *args, **kwargs)