Django REST Framework创建数据时自动关联当前登录用户

要实现创建Advertise实例时自动把owner字段设为当前登录用户，我们可以通过重写CreateAPIView的钩子方法来完成，同时调整序列化器的字段配置避免前端传入不必要的参数。下面是针对你现有代码的具体修改方案：

1. 修改视图类（view.py） #

在你的advertise视图类中重写perform_create方法，这个方法会在实例即将被创建前调用，我们可以在这里把当前登录用户传递给序列化器的save方法：

from rest_framework.generics import CreateAPIView
from rest_framework.permissions import IsAuthenticated  # 可选，用于限制仅登录用户创建
from .models import Advertise
from .serializers import AdvertiseMainSerializer

class advertise(CreateAPIView):
    serializer_class = AdvertiseMainSerializer
    queryset = Advertise.objects.all()
    # 可选：添加权限类，确保只有登录用户才能创建广告
    permission_classes = [IsAuthenticated]

    def perform_create(self, serializer):
        # 自动将当前登录用户赋值给owner字段
        serializer.save(owner=self.request.user)

小提示：按照PEP8规范，类名建议采用大驼峰命名，比如把advertise改成AdvertiseCreateView会更易读。

2. 调整序列化器（serializer.py） #

因为我们会在后端自动设置owner字段，所以需要让序列化器不再要求前端传入这个参数。有两种方式可以实现：

方式一：排除owner字段（仅返回标题和描述） #

如果创建接口的返回结果不需要包含owner信息，可以直接在fields里去掉它：

from rest_framework.serializers import ModelSerializer
from .models import Advertise

class AdvertiseMainSerializer(ModelSerializer):
    class Meta:
        model = Advertise
        fields = ['title', 'description']

方式二：将owner设为只读字段（返回结果包含owner） #

如果需要在创建成功后返回owner的信息，就把owner加入fields并设置为只读：

from rest_framework.serializers import ModelSerializer
from .models import Advertise

class AdvertiseMainSerializer(ModelSerializer):
    class Meta:
        model = Advertise
        fields = ['owner', 'title', 'description']
        read_only_fields = ['owner']

关键原理说明 #

• perform_create是DRF为CreateAPIView提供的钩子方法，专门用于自定义实例创建的逻辑，这里我们通过self.request.user获取当前已认证的登录用户（前提是你已经配置了合适的认证方式，比如Session认证或Token认证）。
• 调整序列化器的字段配置是为了避免前端提交数据时因缺少owner参数而报错，同时也符合前后端分离的设计原则——敏感或后端自动处理的字段不应该由前端传入。

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