Django proxy model and ForeignKey

user_268774 0
Viewed 9923

How to make entry.category to be instance of CategoryProxy? See code for details:

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

Cast from Category to CategoryProxy is ok too, but I am not very familiar with ORM internals to properly copy internal state...

EDIT. Reason: I added method to CategoryProxy and want to use him:

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

EDIT 2. Currently I implemented it like this:

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

but it looks terrible...

django-models django python
7 Answers
user_980800

You can add proxy model to ForeignKey:

class Entry(models.Model):
    category = models.ForeignKey(CategoryProxy, on_delete=models.CASCADE)

and now:

entry = EntryProxy.objects.get(pk=1)
entry.category # CategoryProxy instance
user_53367770
Related