基于子类名称的Django过滤器基类

Question

我在Django中有以下模型继承结构：

class Parent(models.Model):
    # stuff

class A(Parent):
    # stuff

class B(Parent):
    # stuff

class C(Parent):
    # stuff

and the list goes on.

我使用django-model-utils的InheritanceManager来过滤对象，如下所示：

Parent.objects.filter(foo=bar).select_subclasses()

当我想要过滤所有子类时，这很有效。我想要做的是过滤A和B对象，而不是C对象。我想用一个查询来实现这一点，比如

Parent.objects.filter(foo=bar, __class__.__name__=A, __class__.__name__=B).select_subclasses()

可以进行这样的过滤操作吗?如果可能的话，如何进行？

Answer 1

当然可以在一个查询中做到这一点！

这一切都来自于Django构建字段来引用父模型和子模型之间的关系的方式。每个孩子都有某种类型的parent-ref，它有一个related_name。您可以对这些进行查询。

您可能会发现最简单的方法是启动./manage.py shell，导入父模型类，然后执行无意义的查询：

Parent.objects.filter(foo='bar')

这将向您显示可用于查询的字段:然后，您可以确定如何构建查询：

Parent.objects.filter(is_active=True).filter(
    models.Q(a__isnull=False) |
    models.Q(b__isnull=False)
).select_subclasses()

这将执行一个查询，该查询将获取A的所有对象、B的所有对象以及is_active设置为True的所有对象，并向下转换它们。

值得指出的是，.select_subclasses()无法检测哪些模型将被包括在内，因此它加入了所有子类。

然而..。您可以将值传递给select_subclasses，以便它只连接(和向下转换)这些子类：

Parent.objects.filter(is_active=True).filter(
    models.Q(a__isnull=False) |
    models.Q(b__isnull=False)
).select_subclasses('a', 'b')

现在，我们不再连接到表"c“到"z"！

Answer 2

这样设置可继承性通常不被认为是最好的，因为这样每个SQL查询都必须进行连接。这可能会使您的性能变得更慢。要提高性能，您可以使用abstract Meta值：

class Parent(models.Model):
    # stuff
    class Meta:
        abstract = True

这样，每个表都是独立的，因此性能会更快。

如果这不适用于您，我认为不可能在单个查询中做这样的事情，因为表/模型中的字段不包含任何信息，它们位于哪个表中。在这种情况下，您很可能不得不以某种方式子类化InheritanceManager，但是我不确定该怎么做。如果你这样做了，那么使用content_types可能会有所帮助。

如果这是太多的工作，那么你总是可以做一个简单的hack (更像是猴子修补...)。我知道这不是很好，但它会起作用的：

class Parent(models.Model):
    # stuff
    table = models.CharField(max_length=8, default='parent')

class A(Parent):
    # stuff
    table = models.CharField(max_length=8, default='a')

class B(Parent):
    # stuff
    table = models.CharField(max_length=8, default='b')


# and then something like
# please note that I've never used django-model-utils
# so don't know the correct syntax
Parent.objects.filter(foo=bar, table__in=['parent', 'a']).select_subclasses()

Answer 3

另一个对我有效的老套解决方案，不需要向数据库添加更多信息：

letters = Parent.objects.filter(foo=bar)
for letter in letters:
    if type(letter) == C:
        letters.exclude(id=c.id)

或者，如果在不添加字段的情况下向模型添加信息：

class Parent(models.Model):
    # stuff

class A(Parent):
    code = 'A'
    # stuff

class B(Parent):
    code = 'B'
    # stuff

class C(Parent):
    code = 'C'
    # stuff

然后..。

letters = Parent.objects.filter(foo=bar)
for letter in letters:
    if letter.child().code == 'C':
        letters.exclude(id=c.id)

为我的目的而工作，尽管仍然是一个黑客...