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

Arun Singh
Arun Singh
1,530 Points

Code error output

Didn't get the right output for Ingredient.get(); Didn't get the right output for IngredientList.get(); Didn't get the right output forIngredientList.post()`

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,
    '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__()


    @marshal_with(ingredient_fields)
    def get(self):
        ingredients = models.Ingredient.select()
        return ingredients()


    @marshal_with(ingredient_fields)
    def post(self):
        args = self.reqparse.parse_args()
        ingredient = models.Ingredient.create(**args)
        return ingredients(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__()

    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>')

1 Answer

For IngredientList.get() you have to marshal each item in the list. ingredients = [marshal(m, ingredient_fields) for m in models.Ingredient.select()]

For IngredientList.post() since you used the decorator, you can just return ingredient.

For Ingredient.get(), you forgot to marshal the ingredient. You can use the decorator or return marshal(ingredient, ingredient_fields).