Django代理模型和ForeignKey

Question

如何使entry.category成为CategoryProxy的实例？请参阅代码了解详情:

class Category(models.Model): pass

class Entry(models.Model):
    category = models.ForeignKey(Category)

class EntryProxy(Entry):
    class Meta:
        proxy = True

class CategoryProxy(Category):
    class Meta:
        proxy = True

entry = EntryProxy.objects.get(pk=1)
entry.category # !!! I want CategoryProxy instance here

从Category转换为CategoryProxy也是可以的,但是我不熟悉ORM内部以正确复制内部状态...

编辑. 原因:我向CategoryProxy添加了方法并想要使用他:

EntryProxy.objects.get(pk=1).category.method_at_category_proxy()

编辑2. 目前我实现了这样:

EntryProxy._meta.get_field_by_name('category')[0].rel.to = CategoryProxy

但它看起来很糟糕......

Answer 1

要在不命中数据库的情况下从模型类切换到代理类,请执行以下操作:

class EntryProxy(Entry):
    @property
    def category(self):
        new_inst = EntryProxy()
        new_inst.__dict__ = super(EntryProxy, self).category.__dict__
        return new_inst

编辑:上面的代码片段似乎无法在django 1.4上运行.

从django 1.4开始,我手动获取所有值字段,如下所示:

class EntryProxy(Entry):
    @property
    def category(self):
        category = super(EntryProxy, self).category
        new_inst = EntryProxy()
        for attr in [f.attname for f in category.__class__._meta.fields] + ['_state']:
            setattr(new_inst, attr, getattr(category, attr))
        return new_inst

要在不命中数据库的情况下从查询集切换到子代理类,请执行以下操作:

class CategoryProxy(Category):
    @property
    def entry_set(self):
        qs = super(CategoryProxy, self).entry_set
        qs.model = EntryProxy
        return qs

Answer 2

这是一个开放的Django问题:#10961(允许用户使用ForeignKey字段覆盖代理模型上的正向和反向关系)

您可以在定义代理模型后重置相关字段来解决此问题:

EntryProxy.add_to_class('category', CategoryProxy)