Welcome to the Treehouse Community

Want to collaborate on code errors? Have bugs you need feedback on? Looking for an extra set of eyes on your latest project? Get support with fellow developers, designers, and programmers of all backgrounds and skill levels here with the Treehouse Community! While you're at it, check out some resources Treehouse students have shared here.

Looking to learn something new?

Treehouse offers a seven day free trial for new students. Get access to thousands of hours of content and join thousands of Treehouse students and alumni in the community today.

Start your free trial

Python Flask REST API Resourceful Blueprints Ingredient Fields

David Axelrod
David Axelrod
36,073 Points
Posted by David Axelrod
David Axelrod
36,073 Points

return a dict with key ingredients and return the all the records

Confused as to why this isnt working

def get(self):
    ingredients = models.Ingredient.select()
    return { "ingredients" : marshal(ingredients, ingredient_fields) }
resources/ingredients.py 
from flask import Blueprint

from flask.ext.restful import (
    Resource, Api, reqparse, inputs,
    marshal, marshal_with, fields
)

import models

ingredient_fields = {
  "name": fields.String,
  "description": fields.String,
  "recipe" : fields.String,
  "measurement_type": fields.String,
  "quantity" : fields.Float
}


class IngredientList(Resource):
    def __init__(self):
        self.reqparse = reqparse.RequestParser()
        self.reqparse.add_argument(
            'name',
            required=True,
            location=['form', 'json']
        )
        self.reqparse.add_argument(
            'description',
            required=True,
            location=['form', 'json']
        )
        self.reqparse.add_argument(
            'measurement_type',
            required=True,
            location=['form', 'json']
        )
        self.reqparse.add_argument(
            'quantity',
            type=float,
            required=True,
            location=['form', 'json']
        )
        self.reqparse.add_argument(
            'recipe',
            type=inputs.positive,
            required=True,
            location=['form', 'json']
        )
        super().__init__()

    def get(self):
        ingredients = models.Ingredient.select()
        return { "ingredients" : marshal(ingredients, ingredient_fields) }


    @marshal_with(ingredient_fields)
    def post(self):
        args = self.reqparse.parse_args()
        ingredient = models.Ingredient.create(**args)
        return ingredient


class Ingredient(Resource):
    def __init__(self):
          self.reqparse = reqparse.RequestParser()
          self.reqparse.add_argument(
              'name',
              required=True,
              location=['form', 'json']
          )
          self.reqparse.add_argument(
              'description',
              required=True,
              location=['form', 'json']
          )
          self.reqparse.add_argument(
              'measurement_type',
              required=True,
              location=['form', 'json']
          )
          self.reqparse.add_argument(
              'quantity',
              type=float,
              required=True,
              location=['form', 'json']
          )
          self.reqparse.add_argument(
              'recipe',
              type=inputs.positive,
              required=True,
              location=['form', 'json']
          )
          super().__init__()
    @marshal_with(ingredient_fields)
    def get(self, id):
        ingredient = models.Ingredient.get(models.Ingredient.id==id)
        return ingredient

ingredients_api = Blueprint('resources.ingredients', __name__)
api = Api(ingredients_api)
api.add_resource(IngredientList, '/api/v1/ingredients')
api.add_resource(Ingredient, '/api/v1/ingredients/<int:id>')

2 Answers

David Axelrod
David Axelrod
36,073 Points
David Axelrod
David Axelrod
36,073 Points

ahhhHHHhHhHHHHh for the dictionary, needed to run through each record and marshal each individual record

Looks a bit like this

def get(self):
     ingredients = [marshal(model, ingredients_fields)
                                for model in models.Ingredients.select()]
     return {"ingredients": ingredients}
Allison Schaaf
Allison Schaaf
33,322 Points
Allison Schaaf
Allison Schaaf
33,322 Points

This code passed:

from flask import Blueprint

from flask.ext.restful import (
    Resource, Api, reqparse, inputs,
    marshal, marshal_with, fields
)

import models

ingredient_fields = {
    'name': fields.String,
    'description': fields.String,
    'measurement_type': fields.String,
    'quantity': fields.Float,
    'recipe': fields.String
}


class IngredientList(Resource):
    def __init__(self):
        self.reqparse = reqparse.RequestParser()
        self.reqparse.add_argument(
            'name',
            required=True,
            location=['form', 'json']
        )
        self.reqparse.add_argument(
            'description',
            required=True,
            location=['form', 'json']
        )
        self.reqparse.add_argument(
            'measurement_type',
            required=True,
            location=['form', 'json']
        )
        self.reqparse.add_argument(
            'quantity',
            type=float,
            required=True,
            location=['form', 'json']
        )
        self.reqparse.add_argument(
            'recipe',
            type=inputs.positive,
            required=True,
            location=['form', 'json']
        )
        super().__init__()

    def get(self):
        ingredients = [marshal(ingredient, ingredient_fields)
                   for ingredient in models.Ingredient.select()]
        return {'ingredients': ingredients}

    @marshal_with(ingredient_fields)
    def post(self):
        args = self.reqparse.parse_args()
        ingredient = models.Ingredient.create(**args)
        return ingredient


class Ingredient(Resource):
    def __init__(self):
          self.reqparse = reqparse.RequestParser()
          self.reqparse.add_argument(
              'name',
              required=True,
              location=['form', 'json']
          )
          self.reqparse.add_argument(
              'description',
              required=True,
              location=['form', 'json']
          )
          self.reqparse.add_argument(
              'measurement_type',
              required=True,
              location=['form', 'json']
          )
          self.reqparse.add_argument(
              'quantity',
              type=float,
              required=True,
              location=['form', 'json']
          )
          self.reqparse.add_argument(
              'recipe',
              type=inputs.positive,
              required=True,
              location=['form', 'json']
          )
          super().__init__()

    @marshal_with(ingredient_fields)
    def get(self, id):
        ingredient = models.Ingredient.get(models.Ingredient.id==id)
        return ingredient

ingredients_api = Blueprint('resources.ingredients', __name__)
api = Api(ingredients_api)
api.add_resource(IngredientList, '/api/v1/ingredients')
api.add_resource(Ingredient, '/api/v1/ingredients/<int:id>')