如何在Django RF的ModelSerializer中处理增改逻辑并获取当前用户userid
问题解答
一、将recorded_by字段赋值逻辑迁移至ModelSerializer
要把视图中perform_create里的用户ID赋值逻辑移到序列化器,核心是利用序列化器的context属性获取当前请求对象,进而拿到登录用户信息。具体实现步骤如下:
移除视图中的
perform_create方法
既然逻辑要迁移到序列化器,视图里的perform_create可以直接删掉,让ListCreateAPIView使用默认实现即可。在ModelSerializer中重写
create方法
通过self.context['request']获取请求对象,从中提取用户的userid,添加到验证后的数据中,再调用父类的create方法完成创建:from rest_framework import serializers from .models import YourModel # 替换为你的实际模型类 class YourModelSerializer(serializers.ModelSerializer): class Meta: model = YourModel fields = '__all__' # 或指定具体字段 def create(self, validated_data): # 从context中获取当前登录用户的userid user_id = self.context['request'].user.userid # 将userid添加到验证后的数据中 validated_data['recorded_by'] = user_id # 调用父类方法完成实例创建 return super().create(validated_data)
二、确认update方法实现的正确性
假设你的模型包含resolve(布尔型)、resolve_date(日期时间型)、resolve_by(用户ID字段),且需要创建TransactionLog记录操作,以下是标准且正确的实现方式,你可以对照自己的代码检查:
from datetime import datetime from .models import TransactionLog # 替换为你的TransactionLog模型 class YourModelSerializer(serializers.ModelSerializer): # ... 其他代码 ... def update(self, instance, validated_data): # 设置resolve为True instance.resolve = True # 记录当前时间为resolve_date instance.resolve_date = datetime.now() # 记录操作用户的userid instance.resolve_by = self.context['request'].user.userid # 保存修改后的实例 instance.save() # 创建TransactionLog记录 TransactionLog.objects.create( # 根据你的TransactionLog模型字段填写,比如关联的模型实例、操作类型、操作人等 target_instance=instance, operation='resolve', operated_by=self.context['request'].user.userid, operated_at=datetime.now() ) # 返回更新后的实例(注意:这里不需要再调用super().update(),因为我们已经手动修改并保存了instance) return instance
正确性验证要点:
- 确保
self.context['request'].user已通过认证(视图需配置认证类,比如TokenAuthentication或SessionAuthentication),避免匿名用户导致的AttributeError。 - 如果
resolve_date字段设置为auto_now_add或auto_now,可以不用手动赋值,由数据库自动处理,但如果需要明确记录操作时间,手动赋值更可控。 - 创建
TransactionLog时,要确保关联字段和必填字段都正确赋值,避免数据库报错。 - 手动修改
instance字段后调用instance.save(),而不是依赖super().update(),因为我们的修改逻辑不是基于validated_data的常规字段更新,而是固定的状态变更操作。
内容的提问来源于stack exchange,提问作者Aaron Michael
相关产品推荐
相关产品推荐

