zoukankan      html  css  js  c++  java
  • Django REST framework 善用 SerializerMethodField 来优化

    首先来看一个例子,在一般情况下,对于有父子关系的对象,我们使用下面的方法来创建类。
    一个Article类,一个Article对象可以有多个Comment实例,那么Django中类的定义如下:

    #coding:utf-8
    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 比较多,此时就会影响性能。比如:

    #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 来实现这个目的。如下:

    #coding:utf-8
    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 的总数而不显示具体列表了。

  • 相关阅读:
    自动化测试成功11333
    自动化测试成功11222
    自动化测试成功1111
    自动化测试成功112
    自动化测试成功18
    自动化测试成功1
    自动化测试成功
    富文本测试
    关于VMware虚拟机磁盘收缩的几种方法
    8个让你更有效率的简单方法
  • 原文地址:https://www.cnblogs.com/fan-1994716/p/14906685.html
Copyright © 2011-2022 走看看