Django處理枚舉(枚舉模型)及source的使用

更新時間：2024年04月15日 09:12:09 作者：敲代碼敲到頭發(fā)茂密

本文主要介紹了Django處理枚舉(枚舉模型)及source的使用,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧

1、定義模型類、序列化器類

定義模型類models.py；項目模型類、接口模型類、用例模型類

from django.db import models

class Test_Project(models.Model):
    name = models.CharField(help_text='項目名稱', verbose_name='項目名稱',
                            max_length=60,
                            unique=True)
    desc = models.TextField(help_text='項目描述', blank=True, null=True)
    create_time = models.DateTimeField(auto_now_add=True)
    update_time = models.DateTimeField(auto_now=True)

    class Meta:
        db_table = "t_test_project"
        verbose_name = "項目表"
        verbose_name_plural = verbose_name
        ordering = ['id']

    def __str__(self):
        return self.name

CHOICES = [('1', '項目接口'),('2', '外部接口')]

type = models.CharField(verbose_name='接口類型', help_text='接口類型', max_length=40,default='1', choices=CHOICES)

from django.db import models

class Test_Interface(models.Model):
    CHOICES = [
        ('1', '項目接口'),
        ('2', '外部接口')
    ]

    name = models.CharField(help_text='接口名稱', verbose_name='接口名稱',
                            max_length=60,
                            unique=True)
    desc = models.TextField(help_text='接口描述', blank=True, null=True, default='描述')
    type = models.CharField(verbose_name='接口類型', help_text='接口類型',
                            max_length=40,
                            default='1', choices=CHOICES)
    project = models.ForeignKey(Test_Project, on_delete=models.CASCADE)
    playcount=models.IntegerField(help_text='訪問量',verbose_name='訪問量',default=0)
    commentcount = models.IntegerField(default=0, verbose_name='評論量')

    class Meta:
        db_table = 't_test_interface'
        verbose_name = "接口表"
        verbose_name_plural = verbose_name
        ordering = ['id']

    def __str__(self):
        return self.name

from django.db import models

class Test_Testcase(models.Model):
    name = models.CharField(help_text='用例名稱', verbose_name='用例名稱',
                            max_length=60,
                            unique=True)
    desc = models.TextField(help_text='用例名稱', blank=True, null=True)
    create_time = models.DateTimeField(auto_now_add=True)
    update_time = models.DateTimeField(auto_now=True)
    interface = models.ForeignKey(Test_Interface, on_delete=models.CASCADE)

    class Meta:
        db_table = "t_test_testcase"
        verbose_name = "用例表"
        verbose_name_plural = verbose_name
        ordering = ['id']

    def __str__(self):
        return self.name

序列化器類設計

from rest_framework import serializers

from ttest.models import Test_Project,Test_Interface


class TProjectSerializer(serializers.ModelSerializer):

    class Meta:
        model=Test_Project
        fields='__all__'


class TInterfaceSerializer(serializers.ModelSerializer):
    class Meta:
        model=Test_Interface
        fields='__all__'

訪問接口：前端展示的數(shù)據(jù)type：“1”
使用默認的序列化器時，視圖函數(shù)訪問具有choices參數(shù) 的字段或一對一或一對多或多對多字段時，返回的數(shù)據(jù)只有 id 值，就像下面這種方式，性別是0或1

在這里插入圖片描述

2、對上面這些場景使用source參數(shù)

get_xxx_display 用于顯示 choices 參數(shù)對應的文本信息。

序列化器文件：serializers.py

class TProjectSerializer(serializers.ModelSerializer):

    class Meta:
        model=Test_Project
        fields='__all__'


class TInterfaceSerializer(serializers.ModelSerializer):
    type = serializers.ChoiceField(choices=Test_Interface.CHOICES,
                                   source="get_type_display",
                                   read_only=True)  # 設置source="get_屬性_display"即可
    class Meta:
        model=Test_Interface
        fields='__all__'

在這里插入圖片描述

3、支持連表查詢

project_create_time=serializers.CharField(source='project.create_time')

顯示一對一或一對多或多對多字段對應的文本信息。支持連表查詢

from rest_framework import serializers
from ttest.models import Test_Project,Test_Interface


class TProjectSerializer(serializers.ModelSerializer):

    class Meta:
        model=Test_Project
        fields='__all__'


class TInterfaceSerializer(serializers.ModelSerializer):
    type = serializers.ChoiceField(choices=Test_Interface.CHOICES,
                                   source="get_type_display",
                                   read_only=True)  # 設置source="get_屬性_display"即可
    project_create_time=serializers.CharField(source='project.create_time')

    class Meta:
        model=Test_Interface
        fields='__all__'

在這里插入圖片描述

4、自定義序列化輸出方法

自定義序列化輸出方法，會尋找并執(zhí)行’get_xxx’的方法。

from rest_framework import serializers
from ttest.models import Test_Project,Test_Interface


class TProjectSerializer(serializers.ModelSerializer):

    class Meta:
        model=Test_Project
        fields='__all__'


class TInterfaceSerializer(serializers.ModelSerializer):
    type = serializers.ChoiceField(choices=Test_Interface.CHOICES,
                                   source="get_type_display",
                                   read_only=True)  # 設置source="get_屬性_display"即可
    project_create_time=serializers.CharField(source='project.create_time')

    name_zidingyi=serializers.SerializerMethodField()
    def get_name_zidingyi(self,obj):
        return obj.project.name

    class Meta:
        model=Test_Interface
        fields='__all__'

在這里插入圖片描述