Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Flattening one-to-many relationship in Django

I have a few model classes with basic one-to-many relationships. For example, a book has many recipes, and each recipe has many ingredients:

class Book(models.Model):
    name = models.CharField(max_length=64)

class Recipe(models.Model):
    book = models.ForeignKey(Book)
    name = models.CharField(max_length=64)

class Ingredient(models.Model):
    text = models.CharField(max_length=128)
    recipe = models.ForeignKey(Recipe)

I'd like a flat list of all ingredients in all recipes from a particular book. What's the best way to express this in Python?

If I was using LINQ, I might write something like this:

var allIngredients = from recipe in book.Recipes
                     from ingredient in recipe.Ingredients
                     select ingredient;
like image 466
Jason Anderson Avatar asked Nov 17 '08 01:11

Jason Anderson


2 Answers

Actually, it looks like there's a better approach using filter:

my_book = Book.objects.get(pk=1)
all_ingredients = Ingredient.objects.filter(recipe__book=my_book)
like image 185
Jason Anderson Avatar answered Sep 21 '22 14:09

Jason Anderson


To print each recipe and its ingredients:

mybook = Book.objects.get(name="Jason's Cookbook")
for recipe in mybook.recipe_set.all():
    print recipe.name
    for ingredient in recipe.ingredients:
        print ingredient.text

And if you just want to get a list of all ingredient objects:

mybook = Book.objects.get(name="Jason's Cookbook")
ingredient_list = []
for recipe in mybook.recipe_set.all():
    for ingredient in recipe.ingredients:
        ingredient_list.append(ingredient)

Documentation.

like image 22
Harley Holcombe Avatar answered Sep 22 '22 14:09

Harley Holcombe