Django之contenttypes的应用

人生不如戏

发布于 2019-02-13 16:25:34

7560

发布于 2019-02-13 16:25:34

文章被收录于专栏：抠抠空间抠抠空间

Django contenttypes 应用

简介

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

每当我们创建了新的model并执行数据库迁移后，ContentType表中就会自动新增一条记录。比如我在应用app01的models.py中创建表class Electrics(models.Model): pass。从数据库查看ContentType表，显示如下：

那么这个表有什么作用呢？这里提供一个场景，网上商城购物时，会有各种各样的优惠券，比如通用优惠券，满减券，或者是仅限特定品类的优惠券。在数据库中，可以通过外键将优惠券和不同品类的商品表关联起来：

from django.db import models


class Electrics(models.Model):
    """
    id  name
    1   日立冰箱
    2   三星电视
    3   小天鹅洗衣机
    """
    name = models.CharField(max_length=32)


class Foods(models.Model):
    """
    id   name
    1    面包
    2    烤鸭
    """
    name = models.CharField(max_length=32)


class Clothes(models.Model):
    name = models.CharField(max_length=32)


class Coupon(models.Model):
    """
    id     name            Electrics        Foods           Clothes        more...
    1     通用优惠券       null              null            null           
    2     冰箱满减券         2               null            null
    3     面包狂欢节        null              1              null

    """
    name = models.CharField(max_length=32)
    electric_obj = models.ForeignKey(to='Electrics', null=True)
    food_obj = models.ForeignKey(to='Foods', null=True)
    cloth_obj = models.ForeignKey(to='Clothes', null=True)

如果是通用优惠券，那么所有的ForeignKey为null，如果仅限某些商品，那么对应商品ForeignKey记录该商品的id，不相关的记录为null。但是这样做是有问题的：实际中商品品类繁多，而且很可能还会持续增加，那么优惠券表中的外键将越来越多，但是每条记录仅使用其中的一个或某几个外键字段。

通过使用contenttypes 应用中提供的特殊字段GenericForeignKey，我们可以很好的解决这个问题。只需要以下三步：

在model中定义ForeignKey字段，并关联到ContentType表。通常这个字段命名为“content_type”
在model中定义PositiveIntegerField字段，用来存储关联表中的主键。通常这个字段命名为“object_id”
在model中定义GenericForeignKey字段，传入上述两个字段的名字。

为了更方便查询商品的优惠券，我们还可以在商品类中通过GenericRelation字段定义反向关系。

示例代码：

from django.db import models
from django.contrib.contenttypes.models import ContentType
from django.contrib.contenttypes.fields import GenericForeignKey


class Electrics(models.Model):
    name = models.CharField(max_length=32)
    coupons = GenericRelation(to='Coupon')  # 用于反向查询，不会生成表字段

    def __str__(self):
        return self.name


class Foods(models.Model):
    name = models.CharField(max_length=32)
    coupons = GenericRelation(to='Coupon')

    def __str__(self):
        return self.name


class Clothes(models.Model):
    name = models.CharField(max_length=32)
    coupons = GenericRelation(to='Coupon')

    def __str__(self):
        return self.name

class Coupon(models.Model):
    name = models.CharField(max_length=32)

    content_type = models.ForeignKey(to=ContentType) # step 1 添加一个字段，关联ContentType表中的某一个表
    object_id = models.PositiveIntegerField() # step 2 添加一个id字段，就是‘某一张’表在ContentType里的对应的id
    content_object = GenericForeignKey('content_type', 'object_id') # step 3 不会添加字段，只是为了插入或者查询时使用

    def __str__(self):
        return self.name

创建记录和查询

from django.shortcuts import render, HttpResponse
from app01 import models
from django.contrib.contenttypes.models import ContentType


def test(request):
    if request.method == 'GET':
        # ContentType表对象有model_class() 方法，取到对应model
        content = ContentType.objects.filter(app_label='app01', model='electrics').first()  # 表名小写
        cloth_class = content.model_class() # cloth_class 就相当于models.Electrics
        res = cloth_class.objects.all()
        print(res)

        # 为三星电视(id=2)创建一条优惠记录
        s_tv = models.Electrics.objects.filter(id=2).first()
        models.Coupon.objects.create(name='电视优惠券', content_object=s_tv)  #直接用content_object属性添加，无需给contenttype和object_id字段分别赋值

　　　　# 查询优惠券（id=1）绑定了哪些商品 
　　　　coupon_obj = models.Coupon.objects.filter(id=1).first() 
　　　　prod = coupon_obj.content_object 
　　　　print(prod) 

　　　　# 查询三星电视(id=2)的所有优惠券 
　　　　res = s_tv.coupons.all() print(res) 

　　　　# 查询obj的所有优惠券：如果没有定义反向查询字段，通过如下方式： 
　　　　content = ContentType.objects.filter(app_label='app01', model='model_name').first() 
　　　　res = models.OftenAskedQuestion.objects.filter(content_type=content, 
　　　　object_id=obj.pk).all() 
　　　　

　　　　return HttpResponse('....')

总结：

当一张表和多个表FK关联，并且多个FK中只能选择其中一个或其中n个时，可以利用contenttypes app，只需定义三个字段就搞定！

本文参与腾讯云自媒体分享计划，分享自作者个人站点/博客。

原始发表：2018-05-30 ，如有侵权请联系 cloudcommunity@tencent.com 删除

django