forked from mirror/MrDoc
100 lines
2.6 KiB
Python
100 lines
2.6 KiB
Python
# coding:utf-8
|
||
# @文件: serializers_app.py
|
||
# @创建者：州的先生
|
||
# #日期：2020/5/11
|
||
# 博客地址：zmister.com
|
||
|
||
from rest_framework import serializers
|
||
from rest_framework.serializers import ModelSerializer,SerializerMethodField
|
||
from app_doc.models import *
|
||
|
||
# 用户序列化器
|
||
class UserSerializer(ModelSerializer):
|
||
class Meta:
|
||
model = User
|
||
fields = (
|
||
'id', 'last_login', 'is_superuser', 'username', 'email', 'date_joined', 'is_active', 'first_name'
|
||
)
|
||
|
||
|
||
# 文集序列化器
|
||
class ProjectSerializer(ModelSerializer):
|
||
create_time = serializers.DateTimeField(format='%Y-%m-%d %H:%M')
|
||
|
||
class Meta:
|
||
model = Project
|
||
fields = ('__all__')
|
||
|
||
# 协作文集序列化器
|
||
class ProjectCollaSerializer(ModelSerializer):
|
||
project_id = serializers.SerializerMethodField(label="文集ID")
|
||
project_name = serializers.SerializerMethodField(label="文集名称")
|
||
username = serializers.SerializerMethodField(label='文集创建人')
|
||
top_doc = serializers.SerializerMethodField(label="上级")
|
||
|
||
class Meta:
|
||
model = ProjectCollaborator
|
||
fields = ('__all__')
|
||
|
||
def get_project_name(self,obj):
|
||
return obj.project.name
|
||
|
||
def get_project_id(self,obj):
|
||
return obj.project.id
|
||
|
||
def get_username(self,obj):
|
||
username = obj.project.create_user.username
|
||
return username
|
||
|
||
def get_top_doc(self,obj):
|
||
return 0
|
||
|
||
|
||
# 文档序列化器
|
||
class DocSerializer(ModelSerializer):
|
||
|
||
project_name = SerializerMethodField(label="所属文集")
|
||
modify_time = serializers.DateTimeField(format='%Y-%m-%d %H:%M:%S')
|
||
|
||
class Meta:
|
||
model = Doc
|
||
fields = ('__all__')
|
||
|
||
# 返回文档的所属文集
|
||
def get_project_name(self,obj):
|
||
pro_name = Project.objects.get(id=obj.top_doc).name
|
||
return pro_name
|
||
|
||
# 文档模板序列化器
|
||
class DocTempSerializer(ModelSerializer):
|
||
class Meta:
|
||
model = DocTemp
|
||
fields = ('__all__')
|
||
|
||
# 图片序列化器
|
||
class ImageSerializer(ModelSerializer):
|
||
username = serializers.SerializerMethodField(label="用户名")
|
||
class Meta:
|
||
model = Image
|
||
fields = ('__all__')
|
||
|
||
def get_username(self,obj):
|
||
return obj.user.username
|
||
|
||
# 图片分组序列化器
|
||
class ImageGroupSerializer(ModelSerializer):
|
||
class Meta:
|
||
model = ImageGroup
|
||
fields = ('__all__')
|
||
|
||
# 附件序列化器
|
||
class AttachmentSerializer(ModelSerializer):
|
||
username = serializers.SerializerMethodField(label="用户名")
|
||
|
||
class Meta:
|
||
model = Attachment
|
||
fields = ('__all__')
|
||
|
||
def get_username(self,obj):
|
||
return obj.user.username
|