Django提交团队时NOT NULL约束失败，需自动关联当前用户ID

你的问题很典型——因为creator字段是必填的外键，但你的表单和视图都没有自动把当前登录用户的ID赋值给它，导致保存时触发了非空约束错误。不用设置null=True或者默认值，我们可以通过重写CreateView的方法来自动填充当前用户，具体方案如下：

方案1：重写CreateView的form_valid方法 #

这是最直接且推荐的方式，form_valid是表单验证通过后执行保存逻辑的钩子，我们可以在这里手动为creator字段赋值：

修改你的views.py：

from django.contrib.auth.mixins import LoginRequiredMixin
from django.views.generic import CreateView
from .models import Team
from .forms import TeamCreationForm

# 加上LoginRequiredMixin，确保只有登录用户才能创建团队
class TeamsCreateView(LoginRequiredMixin, CreateView):
    model = Team
    form_class = TeamCreationForm

    def form_valid(self, form):
        # 在保存表单前，把当前登录用户赋值给creator字段
        form.instance.creator = self.request.user
        # 调用父类的form_valid完成后续保存逻辑
        return super().form_valid(form)

方案2：在表单的save方法中处理（可选） #

如果你想把逻辑放在表单层面，可以修改表单的save方法，同时在视图中把用户传递给表单：

先修改forms.py： #

from django import forms
from .models import Team

class TeamCreationForm(forms.ModelForm):
    class Meta:
        model = Team
        fields = ('name',)

    # 新增接收user参数的save方法
    def save(self, user, commit=True):
        team = super().save(commit=False)
        team.creator = user
        if commit:
            team.save()
        return team

再修改views.py的form_valid： #

def form_valid(self, form):
    # 调用表单的save方法并传入当前用户
    self.object = form.save(user=self.request.user)
    return super().form_valid(form)

额外提醒：正确导入用户模型 #

确保你在models.py中正确导入用户模型，推荐使用get_user_model()或者settings.AUTH_USER_MODEL来兼容自定义用户模型的情况：

# 方式1：使用settings.AUTH_USER_MODEL
from django.conf import settings
from django.db import models

class Team(models.Model):
    creator = models.ForeignKey(settings.AUTH_USER_MODEL, related_name='teams', on_delete=models.CASCADE)
    name = models.CharField(max_length=20, unique=True)
    rank = models.IntegerField(default=0)

# 方式2：导入get_user_model
from django.contrib.auth import get_user_model
from django.db import models

User = get_user_model()
class Team(models.Model):
    creator = models.ForeignKey(User, related_name='teams', on_delete=models.CASCADE)
    name = models.CharField(max_length=20, unique=True)
    rank = models.IntegerField(default=0)

这样修改后，用户创建团队时，creator字段会自动填充当前登录用户的ID，既满足了非空约束的要求，又不需要设置默认值或允许为空。

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