SerializerMethodField 来优化没必要要的查询

在 Django REST framework 善用 SerializerMethodField 来优化没必要要的查询首先来看一个例子，在通常状况下，对于有父子关系的对象，咱们使用下面的方法来建立类。一个Article类，一个Article对象能够有多个Comment实例，那么Django中类的定义以下：python

from django.db import models
from django.contrib.auth.models import User

class Article(models.Model):
    title = models.CharField('文章标题', max_length=1024, blank=False)
    summary = models.TextField('文章简介', blank=False)
    content = models.TextField('文章内容', blank=True, null=True)
    create_user = models.ForeignKey(User, related_name='article_create_user', verbose_name='建立用户')
    create_time = models.DateTimeField('建立时间', auto_now_add=True)
    def __str__(self):
        return self.title

    def __unicode__(self):
        return self.title

    class Meta:
        ordering = ('-create_time',)


class Comment(models.Model):
    article = models.ForeignKey(Article, related_name='article_comments', verbose_name='文章')
    comment = models.CharField('评论', max_length=1024, blank=False)
    create_user = models.ForeignKey(User, related_name='article_comments_create_user', verbose_name='建立用户')
    create_time =models.DateTimeField('建立时间', auto_now_add=True)
    
    def __str__(self):
    return self.comment

    def __unicode__(self):
        return self.comment

    class Meta:
        ordering = ('-create_time',)

根据上面类的定义，那么在 Django REST framework 咱们一般按下面的方式定义 Serializer 类。在 ArticleSerializer 类中增长一个属性 article_comments 来保存当前 Article 对象全部的评论集合。可是此时有个问题若是此 Article 对象所拥有的 Comment 比较多，此时就会影响性能。好比：django

#coding:utf-8
from rest_framework import serializers
from .models import Article, Comment

class ArticleSerializer(serializers.ModelSerializer):
    article_comments = serializers.PrimaryKeyRelatedField(many=True, required=False, read_only=True)
    class Meta:
        model = Article
        fields = ('id', 'title', 'summary', 'content', 'create_user', 'create_time', 'article_comments')

        
class CommentSerializer(serializers.ModelSerializer):
    class Meta:
        model = Comment
        fields = ('id', 'article', 'comment', 'create_user', 'create_time')

其实在不少时候咱们并不须要在查询 Article 对象的时候查询所拥有的 Comment 对象，不少时候咱们只是须要一个 Article 所拥有 Comment 对象的总数就能够了，若是有须要再去查询 Comment 列表详细。此时咱们就能够使用 Django REST framework 提供的 SerializerMethodField 来实现这个目的。以下：api

from rest_framework import serializers
from .models import Article, Comment

class ArticleSerializer(serializers.ModelSerializer):
    article_comments_count = serializers.SerializerMethodField()
    class Meta:
        model = Article
        fields = ('id', 'title', 'summary', 'content', 'create_user', 'create_time', 'article_comments_count')

    def get_article_comments_count(self, obj):
        return obj.article_comments.all().count()

class CommentSerializer(serializers.ModelSerializer):
    class Meta:
        model = Comment
        fields = ('id', 'article', 'comment', 'create_user_id', 'create_user_name', 'create_time')

首先在 ArticleSerializer 中去除 article_comments 属性；而后在 ArticleSerializer 中增长一个属性 article_comments_count，并把这个属性添加到 Meta 的 fields 列表里；添加一个 get_article_comments_count 方法，这个方法的命名规则就是在上面声明的属性前面加上个 “get_” 前缀，并接受一个 obj 参数，这个 obj 参数就是当前的 Article 对象实例。性能

此时在查看 Article 的api中就只显示 Comment 的总数而不显示具体列表了。优化