「Django」contenttypes基本用法

當一張表和多個表ForeignKey關聯，而且多個FK中只能選擇其中一個或其中n個時，能夠利用contenttypes，只需定義三個字段就搞定！

contenttypes 是Django內置的一個應用，能夠追蹤項目中全部app和model的對應關係，並記錄在ContentType表中。

每當咱們建立了新的model並執行數據庫遷移後，ContentType表中就會自動新增一條記錄。好比我在應用app的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」，不然GenericRelation反向查找的時候會報錯
• 在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

contenttype寫法

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字段分別賦值
<br>　　　　# 查詢優惠券（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() <br>　　　　<br><br>　　　　return HttpResponse('....')

記錄和查詢

------------------------------------------------------------------------------------------------------------------------------------------------

PS：在model中定義PositiveIntegerField字段，必須用「object_id」，不然GenericRelation反向查找的時候會報錯！！！