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