89 lines
3.8 KiB
Python
Executable File
89 lines
3.8 KiB
Python
Executable File
|
|
from rest_framework import serializers
|
|
from django_restql.mixins import DynamicFieldsMixin
|
|
from rest_framework.fields import empty
|
|
from rest_framework.request import Request
|
|
|
|
class BaseSerizlierMixin:
|
|
create_by_name = serializers.CharField(source='create_by.name', read_only=True)
|
|
update_by_name = serializers.CharField(source='update_by.name', read_only=True)
|
|
|
|
class PkSerializer(serializers.Serializer):
|
|
ids = serializers.ListField(
|
|
child=serializers.CharField(max_length=20), label="主键ID列表")
|
|
soft = serializers.BooleanField(
|
|
label="是否软删除", default=True, required=False)
|
|
|
|
|
|
class GenSignatureSerializer(serializers.Serializer):
|
|
path = serializers.CharField(label="图片地址")
|
|
|
|
|
|
class TreeSerializerMixin:
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
request = self.context.get('request', None)
|
|
self.with_children = request.query_params.get('with_children', 'no') if request else 'no'
|
|
if self.with_children in ['yes', 'count']:
|
|
if 'children' not in self.fields:
|
|
self.fields['children'] = serializers.SerializerMethodField()
|
|
if 'children_count' not in self.fields:
|
|
self.fields['children_count'] = serializers.SerializerMethodField()
|
|
|
|
def get_children(self, obj):
|
|
if hasattr(obj, 'parent') and self.with_children == 'yes':
|
|
serializer_class = self.__class__
|
|
return serializer_class(obj.__class__.objects.filter(parent=obj), many=True, context=self.context).data
|
|
return []
|
|
|
|
def get_children_count(self, obj):
|
|
if hasattr(obj, 'parent') and self.with_children in ['yes', 'count']:
|
|
return obj.__class__.objects.filter(parent=obj).count()
|
|
return 0
|
|
|
|
class CustomModelSerializer(DynamicFieldsMixin, TreeSerializerMixin, serializers.ModelSerializer):
|
|
"""
|
|
自定义serializer/包含创建和新增字段处理
|
|
"""
|
|
|
|
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 create(self, validated_data):
|
|
if self.request:
|
|
if getattr(self.request, 'user', None):
|
|
if getattr(self.Meta.model, 'create_by', None):
|
|
validated_data['create_by'] = self.request.user
|
|
validated_data['update_by'] = self.request.user
|
|
if 'belong_dept' in validated_data:
|
|
pass
|
|
elif getattr(self.request.user, 'belong_dept', None):
|
|
if hasattr(self.Meta.model, 'belong_dept'):
|
|
validated_data['belong_dept'] = self.request.user.belong_dept
|
|
return super().create(validated_data)
|
|
|
|
def update(self, instance, validated_data):
|
|
if self.request:
|
|
if hasattr(instance, 'update_by'):
|
|
validated_data['update_by'] = getattr(
|
|
self.request, 'user', None)
|
|
return super().update(instance, validated_data)
|
|
|
|
|
|
class QuerySerializer(serializers.Serializer):
|
|
field = serializers.CharField(label='字段名')
|
|
compare = serializers.ChoiceField(
|
|
label='比较式', choices=["", "!", "gte", "gt", "lte", "lt", "in", "contains", "isnull"])
|
|
value = serializers.JSONField(label='值', allow_null=True)
|
|
|
|
|
|
class ComplexSerializer(serializers.Serializer):
|
|
page = serializers.IntegerField(min_value=0, required=False)
|
|
page_size = serializers.IntegerField(min_value=1, required=False)
|
|
ordering = serializers.CharField(required=False)
|
|
querys = serializers.ListField(child=QuerySerializer(
|
|
many=True), label="查询列表", required=False)
|
|
annotate_field_list = serializers.ListField(child=serializers.CharField(), label="RawSQL字段列表", required=False)
|