CODEWITHSUNDEEP
djangodjango-modelsinterfacedry
ohnoes
ohnoes

Reputation: 5622

How do I implement a common interface for Django related object sets?

Here's the deal:

I got two db models, let's say ShoppingCart and Order. Following the DRY principle I'd like to extract some common props/methods into a shared interface ItemContainer.

Everything went fine till I came across the _flush() method which mainly performs a delete on a related object set.

class Order(models.Model, interface.ItemContainer):

# ...

def _flush(self):
    # ...
    self.orderitem_set.all().delete()

So the question is: how do I dynamically know wheter it is orderitem_set or shoppingcartitem_set?

Upvotes: 4

Views: 941

Answers (2)

Carl Meyer
Carl Meyer

Reputation: 126631

You can set the related_name argument of a ForeignKey, so if you want to make minimal changes to your design, you could just have ShoppingCartItem and OrderItem set the same related_name on their ForeignKeys to ShoppingCart and Order, respectively (something like "item_set"):

order = models.ForeignKey(Order, related_name='item_set')

and

cart = models.ForeignKey(ShoppingCart, related_name='item_set')

Upvotes: 2

Van Gale
Van Gale

Reputation: 43922

First, here are two Django snippets that should be exactly what you're looking for:

Second, you might want to re-think your design and switch to the django.contrib content types framework which has a simple .model_class() method. (The first snippet posted above also uses the content type framework).

Third, you probably don't want to use multiple inheritance in your model class. It shouldn't be needed and I wouldn't be surprised if there were some obscure side affects. Just have interface.ItemContainer inherit from models.Model and then Order inherit from only interface.ItemContainer.

Upvotes: 3

Related Questions