django中的content_type

一。应用场景：

现在有3张表映射（一对多）一张表：表1，2，3一对多表4，每种车都有自己的价格策略

表1：巴士(Bus) →  字段：id，name

表2：出租车(Taxi) → 字段：id，name

表3：地铁(Metro) → 字段：id，name

表4：价格策略(PrecePolicy) → 字段：id，price，period，对应的车的表名称，id（对应表名称的行id）

 二。原始解决方案：

表1，2，3分别创建，表4的创建为：

class PricePolicy(models.Model):
    """
    价格策略
    """
    price = models.IntegerField()
    period = models.IntegerField()

    table_name = models.CharField(verbose_name="关联的表名称")
    object_id = models.CharField(verbose_name="关联的表中数据行的id")

缺点：这样创建和修改表比较麻烦

三。最终解决方案

1. 添加content_type字段，用来通过contentType表关联车型表

2. 添加object_id字段，用来关联车型表中的某一行

3. 添加content_object字段，用来使以上两个表关联起来

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

class PricePolicy(models.Model):
    """
    价格策略
    """
    price = models.IntegerField()
    period = models.IntegerField()

    content_type = models.ForeignKey(ContentType, verbose_name="关联的表名称", on_delete=models.CASCADE)
    object_id = models.IntegerField(verbose_name="关联的表中数据行的id")
    # 以下字段不添加数据表的列数，只帮助快速实现content_type操作
    content_object = GenericForeignKey('content_type', 'object_id')

使用：

1. 为某一类型的车添加价格，也就是往PrecePolicy中添加数据

1.1 先获取bus的对象，可以根据任意途径获取

1.2 再添加PricePlicy，这里只设置content_object参数和PricePlicy的字段就好了

# 先获取bus的对象，可以根据任意途径获取
obj  = models.Bus.objects.filter(name = 'mini').first() 

# 再通过PricePlicy进行添加，这里只设置content_object参数就好了
models.PricePolicy.objects.create(price=29.9, period=90, content_object=obj)

2. 根据车型id获取车型，并获取该车型的所有价格策略

2.1 先在表类中加入GenericRelation关联反向查找的表

2.2 在view视图函数中，先获取要查找的车型对象

2.3 再通过车型获取所有的价格策略

from django.contrib.contenttype.fields import GenericRelation

class Bus(models.Model):
    name = models.CharField(max_length=32)
    
    # 这个字段不用于生成表字段，只用于反向查找
    price_policy_list = GenericRelation("PricePolicy")


# 先获取车型
bus = models.Bus.objects.filter(id=1).first()

# 再通过车型获取所有的价格策略
bus.price_policy_list.all()

3. 删除车型，或者删除某一车型的某一行价格

3.1 删除车型：

只需要找出车型对象，执行delete()操作就性了，不过会把这个车型关联的价格策略全部删除

Bus.objects.filter(id=1).delete()

3.2 删除某车型的某一行价格：

# 先获取某一车型对象
busObj = Bus.objects.filter(id=1)

# 通过车型对象获取然后筛选所有要删除的价格
allBusPrice = busObj.price_policy_list.filter(price=30)

# 删除
allBusPrice.delete()

只需要找出某车型的所有对象，再通过filter过滤找到要删除的价格，执行delete()就可以了

4. 修改某车型的数据，或者修改某车型的某一行数据

4.1 修改车型：

只要找出某车型对象，然后修改，然后执行save()就行了（id还是关联contenttype表的，所以修改name不怕）

# 这里添加first表示获取对象而不是queryset
busObj = Bus.objects.filter(id=1).first() 
# 修改字段数据
busObj.name = 'big'
# 保存修改
busObj.save()

4.2 修改某车型的某一行数据

先找出某车型对象对应的价格的所有对象，再通过filter过滤找到要修改的价格对象，然后修改，保存就行了

# 获取bus对象
busObj = Bus.objects.filter(id=1).first() 

# 通过车型对象获取然后筛选要修改的价格对象（多个也可以，用循环就行）
changeBusPrice = busObj.price_policy_list.filter(price=30).first()

# 修改价格
changeBusPrice.price = 50

# 保存
changeBusPrice.save()