mirror of
https://github.com/taskPyroer/DjangoVue3Admin.git
synced 2026-10-02 16:11:42 +00:00
88 lines
3.9 KiB
Python
88 lines
3.9 KiB
Python
"""
|
||
Time: 2023/8/6 16:07
|
||
Author: 公众号【布鲁的Python之旅】，【github】https://github.com/taskPyroer， 【gitee】https://gitee.com/hu_yupeng123/projects
|
||
Version: V 0.1
|
||
File:
|
||
Describe: 自定义序列化器
|
||
"""
|
||
from rest_framework import serializers
|
||
from rest_framework.fields import empty
|
||
from rest_framework.request import Request
|
||
from rest_framework.serializers import ModelSerializer
|
||
|
||
from app_user.models import Users
|
||
|
||
|
||
class CustomModelSerializer(ModelSerializer):
|
||
"""
|
||
增强DRF的ModelSerializer,可自动更新模型的审计字段记录
|
||
(1)self.request能获取到rest_framework.request.Request对象
|
||
"""
|
||
# 修改人的审计字段名称, 默认modifier, 继承使用时可自定义覆盖
|
||
modifier_field_id = 'modifier'
|
||
modifier_name = serializers.SerializerMethodField(read_only=True)
|
||
|
||
def get_modifier_name(self, instance):
|
||
if not hasattr(instance, 'modifier'):
|
||
return None
|
||
queryset = Users.objects.filter(id=instance.modifier).values_list('nickname', flat=True).first()
|
||
if queryset:
|
||
return queryset
|
||
return None
|
||
|
||
# 创建人的审计字段名称, 默认creator, 继承使用时可自定义覆盖
|
||
creator_field_id = 'creator'
|
||
creator_name = serializers.SlugRelatedField(slug_field="nickname", source="creator", read_only=True)
|
||
# # 数据所属部门字段
|
||
# dept_belong_id_field_name = 'dept_belong_id'
|
||
# 添加默认时间返回格式
|
||
create_datetime = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S", required=False, read_only=True)
|
||
update_datetime = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S", required=False)
|
||
|
||
def __init__(self, instance=None, data=empty, request=None, **kwargs):
|
||
super().__init__(instance, data, **kwargs)
|
||
self.request: Request = request or self.context.get('request', None)
|
||
|
||
def save(self, **kwargs):
|
||
return super().save(**kwargs)
|
||
|
||
def create(self, validated_data):
|
||
if self.request:
|
||
if str(self.request.user) != "AnonymousUser":
|
||
if self.modifier_field_id in self.fields.fields:
|
||
validated_data[self.modifier_field_id] = self.get_request_user_id()
|
||
if self.creator_field_id in self.fields.fields:
|
||
validated_data[self.creator_field_id] = self.request.user
|
||
return super().create(validated_data)
|
||
|
||
def update(self, instance, validated_data):
|
||
"""
|
||
self.request：这是一个类的属性，代表HTTP请求对象。它被用于获取当前请求的用户信息。
|
||
self.request.user：表示当前请求的用户。如果用户为匿名用户（未登录），则其值为"AnonymousUser"。
|
||
self.modifier_field_id：这是一个字段标识符，用于指定要修改的字段。
|
||
self.fields.fields：这是一个字段集合，包含了模型序列化器中定义的所有字段。
|
||
validated_data：这是经过验证后的数据字典，其中包含了要更新的字段和对应的值。
|
||
"""
|
||
if self.request:
|
||
if str(self.request.user) != "AnonymousUser":
|
||
if self.modifier_field_id in self.fields.fields:
|
||
validated_data[self.modifier_field_id] = self.get_request_user_id()
|
||
if hasattr(self.instance, self.modifier_field_id):
|
||
setattr(self.instance, self.modifier_field_id, self.get_request_user_id())
|
||
return super().update(instance, validated_data)
|
||
|
||
def get_request_username(self):
|
||
if getattr(self.request, 'user', None):
|
||
return getattr(self.request.user, 'username', None)
|
||
return None
|
||
|
||
def get_request_nickname(self):
|
||
if getattr(self.request, 'user', None):
|
||
return getattr(self.request.user, 'nickname', None)
|
||
return None
|
||
|
||
def get_request_user_id(self):
|
||
if getattr(self.request, 'user', None):
|
||
return getattr(self.request.user, 'id', None)
|
||
return None
|