如何从 django 中的单个表检索多态模型 - 或者如何在 django 中实现多态行为

Question

我可以从单个数据库表中读取多态模型，其行为取决于模型的（布尔）字段吗？

在我的一个模型中，如果实例是“向前”与“向后”或“向左”与“向右”，则行为会略有不同。这会导致大量的 if 子句和代码重复。所以我想要有一个模型的前向和后向变体来封装不同的行为。

但是我怎样才能让模型管理器返回正确类的实例呢？我是否必须覆盖模型的 __init__ ？

也许用一个例子更容易解释。我正在做什么：

class Foo(models.Model):
    forward = models.BooleanField()
    other_fields = ...

    def do_foobar(bar):
        if self.forward:
            gap = bar.end_pos - bar.current_pos
            self.do_forward_move(max = gap)
            if self.pos==bar.end_pos:
                and so on ...
        else:
            gap = bar.current_pos - bar.start_pos
            self.do_backward_move(max = gap)
            if self.pos==bar.start_pos:
                and so on ...

我想要做什么：

class Foo(models.Model):
    forward = models.BooleanField()
    other_fields = ...

    def __init__(*args, **kwargs):
        """ return ForwardFoo or BackwardFoo 
        depending on the value of 'forward'"""
        How?

    def do_foobar(bar):
        gap = self.calculate_gap(bar)
        self.do_move(max = gap)
        if self.end_point_reached():
            and so on ...

class ForwardFoo(Foo):
    def calculate_gap(bar):
        return bar.end_pos - bar.current_pos
and so on ...

for f in Foo.objects.all():
    f.do_foobar(bar)

或者是否有一种完全不同的方法来避免这种代码重复？

Answer 1

代理模型：

class Foo(models.Model):
    # all model attributes here

class ForwardFooManager(models.Manager):
    def get_query_set(self, *args, **kwargs):
        qs = super(ForwardFooManager, self).get_query_set(*args, **kwargs)
        return qs.filter(forward=True)

class ForwardFoo(Foo):
    class Meta:
        proxy = True

    objects = ForwardsFooManager()

    # methods for forward model

class BackwardFooManager(models.Manager):
    def get_query_set(self, *args, **kwargs):
        qs = super(BackwardFooManager, self).get_query_set(*args, **kwargs)
        return qs.filter(forward=False)

class BackwardFoo(Foo):
    class Meta:
        proxy = True

    objects = BackwardFooManager()

    # methods for backward model

上面创建了两种代理模型：一种用于前向，一种用于后向。代理模型没有自己的数据库表；它们使用与其继承的模型相同的数据库表。 （这也意味着您不能向代理模型添加任何其他字段，只能添加方法。）

还有一个自定义管理器，用于强制每个字段仅返回属于每个字段的项目子集。只需添加您需要的任何特定方法即可完成。