【Django】ContentType组件

理解
表结构
使用

好，现在我们有这样一个需求，我们的商城里有很多的商品，然而节日要来了，我们要搞活动.

那么，我们就要设计优惠券，优惠券都有什么类型呢？满减的、折扣的、立减的、等等等...

我们对应着活动类型，对某类商品设计优惠卷，比如：

家电是一类商品、食物是一类商品，那么我们就可以设计家电折扣优惠券，以及食物满减优惠券等.

所以，我们一顺手，表结构就出来了：

from django.db import models

is_true = {'null': True, 'blank': True}

# 家用电器表

class Appliance(models.Model):

    """

    id  name

    1   冰箱

    2   电视

    3   洗衣机

    """

    name = models.CharField(max_length=64)

# 食品表

class Food(models.Model):

    """

    id  name

    1   面包

    2   牛掰

    """

    name = models.CharField(max_length=64)

# 水果表

class Fruit(models.Model):

    """

    id  name

    1   苹果

    2   香蕉

    """

    name = models.CharField(max_length=64)

# class ...

# 优惠卷表

class Coupon(models.Model):

    """

    id    title    appliance_id    food_id    fruit_id

    1   通用优惠卷     null          null        null

    2   冰箱折扣券       1           null        null

    3   电视折扣券       2           null        null

    4   苹果满减卷     null          null         1

    """

    title = models.CharField(max_length=64)

    appliance = models.ForeignKey(to='Appliance', **is_true)

    food = models.ForeignKey(to='Food', **is_true)

    fruit = models.ForeignKey(to='Fruit', **is_true)

    # ...

    # 实际上我们的商品种类会特别多，导致我们这张表的外键也越来越多

殊不知，我们的大Django早就为我们提供了更高明的用法——ContentType组件.

理解

ContentType是Django中内置的一个应用，可以追踪项目中所有的APP和model的对应关系，并记录在ContentType表中.

当我们的项目做数据迁移后，会在数据库中生成一些Django自带的表，其中就包含django_content_type表.

我们先来看看这张表：

再来看看这张表内默认的数据：

可见，自动就建立了所有APP与其数据表的对应关系.

表结构

步骤

在model中定义ForeignKey字段，并关联到ContentType表，通常这个字段命名为content-type.
在model中定义PositiveIntergerField字段，用来存储步骤一中被外键的表中的主键，通常这个字段命名为object_id.
在model中定义GenericForeignKey字段，传入上面两个字段的名字.
为了方便反向查询，可在被外键表中定义GenericRelation字段并指向创建外键的表.

好，根据开篇的表结构示例以及如上的步骤，我们的新表结构又出来了：

from django.db import models

from django.contrib.contenttypes.models import ContentType

from django.contrib.contenttypes.fields import GenericForeignKey, GenericRelation

# 家用电器表

class Appliance(models.Model):

    name = models.CharField(max_length=64)

    # 4. 为了方便反向查询，可在被外键表中定义GenericRelation字段并指向创建外键的表

    coupons = GenericRelation(to='Coupon')

# 食物表

class Food(models.Model):

    name = models.CharField(max_length=64)

# 水果表

class Fruit(models.Model):

    name = models.CharField(max_length=64)

# 优惠卷表

class Coupon(models.Model):

    title = models.CharField(max_length=64)

    # 1. 在model中定义ForeignKey字段，并关联到ContentType表

    content_type = models.ForeignKey(to=ContentType)  # 这里就不要加引号了

    # 2. 在model中定义PositiveIntergerField字段，用来存储步骤一中被外键的表中的主键

    object_id = models.PositiveIntegerField()

    # 3. 在model中定义GenericForeignKey字段，传入上面两个字段的名字.

    content_object = GenericForeignKey('content_type', 'object_id')

数据迁移后，django_content_type表中自动就增加了APP与其model的对应字段.

使用

首先我们准备些数据

# 在Python脚本中调用Django环境

import os

if __name__ == '__main__':

    # 注意将下面的'blog097.settings'改为对应的配置文件路径

    os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'blog097.settings')

    import django

    django.setup()

    from blog import models

    [models.Appliance.objects.create(name=i) for i in ("冰箱", "电视", "洗衣机")]

    [models.Food.objects.create(name=i) for i in ("面包", "牛奶")]

    [models.Fruit.objects.create(name=i) for i in ("苹果", "香蕉")]

如果不能理解此操作，可见博文：【在Python脚本中调用Django环境】

然后开始我们的ORM操作

# 在Python脚本中调用Django环境

import os

if __name__ == '__main__':

    # 注意将下面的'blog097.settings'改为对应的配置文件路径

    os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'blog097.settings')

    import django

    django.setup()

    from blog import models

    from django.contrib.contenttypes.models import ContentType

    """通过ContentType获得表名"""

    content_type_obj = ContentType.objects.filter(app_label='blog', model='appliance').first()  # 打印结果：appliance

    """获得表内的所有对象，相当于models.Applicance"""

    model_class = content_type_obj.model_class()  # <class 'blog.models.Appliance'>

    obj_list = model_class.objects.all()

    [print(obj.name) for obj in obj_list]  # 冰箱、电视、洗衣机

    """为冰箱添加一条优惠记录"""

    ice_box = models.Appliance.objects.filter(name="冰箱").first()

    models.Coupon.objects.create(title="冰箱折扣券", content_object=ice_box)

    """查询冰箱的所有优惠卷"""

    # 我们定义了反向查询

    coupon_list01 = ice_box.coupons.all()

    [print(coupon.title) for coupon in coupon_list01]

    # 如果没有定义反向查询

    coupon_list02 = models.Coupon.objects.filter(content_type=content_type_obj, object_id=ice_box.id).all()

    [print(coupon.title) for coupon in coupon_list02]

巴特西

【Django】ContentType组件

理解

表结构

使用

最新文章

热门文章