创建项目后设置用户属性

Question

在我的应用中，用户可以通过命名来创建项目。

这是我的模型：

class Project(models.Model):
    name = models.CharField(max_length=250)
    team_id = models.ForeignKey(Team, blank=True, null=True)
    project_hr_admin = models.ForeignKey(MyUser, blank=True, null=True)

    def get_absolute_url(self):
        return reverse('website:ProjectDetails', kwargs = {'pk' : self.pk})

    def __str__(self):
        return self.name

这是我的看法：

class ProjectCreate(CreateView):
    model = Project
    fields = ['name']
    template_name = 'project_form.html'

当用户使用自己的ID创建项目project_hr_admin = models.ForeignKey(MyUser, blank=True, null=True) 并将其保存在数据库中时如何设置？？

谢谢你

Answer 1

希望这会有所帮助。

class ProjectCreate(CreateView):
    model = Project
    fields = ['name']
    template_name = 'project_form.html'


    def form_valid(self, form):
        form.instance.project_hr_admin = self.request.user
        return super(ProjectCreate, self).form_valid(form)

或者你也可以重写form_valid函数为

    def form_valid(self, form):
        obj = form.save(commit=False)
        obj.project_hr_admin = self.request.user
        obj.save()
        return HttpResponseRedirect('YOUR_SUCCESS_URL')

来源：Django Documentation