Django:select_related与多表继承

时间:2018-11-16 15:55:33

标签: python django inheritance

我具有以下模型结构:

class A(models.Model):
    prop_a = models.CharField(max_length=255)

class B(A):
    prop_b = models.CharField(max_length=255)

class C(A):
    prop_c = models.CharField(max_length=255)

class D(models.Model):
    fk = models.ForeignKey('A')

因此,从本质上讲,我有一个模型(D),该模型具有“抽象”模型(A)的外键,该模型由BC子类化。

现在,当我运行D.objects.all().select_related()时,仅查询A的属性。我认为这是因为在查询时,Django不知道fk是哪个子类的实例(在当前结构中我也不知道)。

有什么方法可以在不更改模型结构的情况下查询子类的属性?

我还尝试了prefetch_related并尝试使用django-model-utils中的InheritanceManager,但都无济于事。

编辑:为澄清起见,我正在寻找一种方法来遍历D的对象,访问fk(即{{1} }或B对象),而只命中一次数据库。

运行C会生成查询

D.objects.all().select_related()

我们可以说其中之一是'SELECT "mapping_d"."id", "mapping_d"."fk_id", "mapping_a"."id", "mapping_a"."prop_a" FROM "mapping_d" INNER JOIN "mapping_a" ON ("mapping_d"."fk_id" = "mapping_a"."id")'类型的对象d,其属性D的类型为fk

现在,当我访问值C时,Django将运行一个附加查询以获取C的属性:

d.fk.c

我想避免对D对象的此附加查询。以建议的方式使用InheritanceManager似乎不行。

1 个答案:

答案 0 :(得分:1)

我通常使用Inheritance Manager并按预期运行。

1)记住在A上设置继承管理器:

class A(models.Model):
    # ...
    objects = InheritanceManager()

2)进行正确的查询:

some_d_object = D.objects.get( pk = 1 )
related = some_d_object.fk.all().select_related()

与之相比,查询D.objects.all().select_related()将返回D个对象。没有人继承自D,因此没有子类,只有D对象。

3)享受继承。

已编辑,我要在OP注释中编辑此问题。

  

使用这种模式,当我访问相关对象时,Django将执行附加的数据库查询。由于我有很多D对象,因此这变得非常昂贵。您是否知道是否可以立即获取所有内容? –

好吧,我们尽量不要写答案,因为我真的不太了解OP正在寻找的巫婆对象。问题很好,因为他几乎写了a Minimal, Complete, and Verifiable example,但我不知道期望值,让我们尝试一些示例。

1)如果您想要所有对象,就像:

all_objects = A.objects.all().select_subclasses()

2)如果要过滤,请执行此操作。我写了一个示例:

filtered_objects = A.objects.filter( d__pk = 1 ).select_subclasses()

3)在单个查询中获取所有子类以及D类:

很遗憾,您不能申请prefetch_related(也许可以,但是我不知道怎么做)。然后,您需要使用itertools按摩数据。

all_objects_and_D = list(
                     A
                     .objects
                     .select_related('D')
                     .select_subclasses()
                     .order_by( 'D__pk' )
                     )
#at this point use itertools to massage your data
keyfunc = lambda x: x.D
data = [  { k, list(g) } 
          for k, g in 
          itertools.groupby(all_objects_and_D, key=keyfunc )
       ]