add CreateCategory and CreateIngredient Mutations

Author: jonirrings

cookbook/ingredients/schema.py

@@ -21,7 +21,6 @@ class Query(object):
                               name=graphene.String())
     all_categories = graphene.List(CategoryType)
 
-
     ingredient = graphene.Field(IngredientType,
                                 id=graphene.Int(),
                                 name=graphene.String())
@@ -55,4 +54,38 @@ def resolve_ingredient(self, info, **kwargs):
         if name is not None:
             return Ingredient.objects.get(name=name)
 
-        return None
+        return None
+
+
+class CreateCategory(graphene.Mutation):
+    class Arguments:
+        name = graphene.String(required=True)
+
+    category = graphene.Field(lambda: CategoryType)
+    ok = graphene.Boolean()
+
+    def mutate(self, info, name):
+        category = Category.objects.create(name=name)
+        ok=True
+        return CreateCategory(category=category,ok=ok)
+
+
+class CreateIngredient(graphene.Mutation):
+    class Arguments:
+        name = graphene.String(required=True)
+        notes = graphene.String(required=True)
+        category = graphene.Int(required=True)
+
+    ingredient = graphene.Field(lambda: IngredientType)
+    ok = graphene.Boolean()
+
+    def mutate(self, info, **kwargs):
+        kwargs['category'] = Category.objects.get(pk=kwargs['category'])
+        ingredient = Ingredient.objects.create(**kwargs)
+        ok = True
+        return CreateIngredient(ingredient=ingredient,ok=ok)
+
+
+class Mutation(object):
+    create_category = CreateCategory.Field()
+    create_ingredient = CreateIngredient.Field()

cookbook/schema.py

@@ -9,4 +9,8 @@ class Query(cookbook.ingredients.schema.Query, graphene.ObjectType):
     pass
 
 
-schema = graphene.Schema(query=Query)
+class Mutation(cookbook.ingredients.schema.Mutation, graphene.ObjectType):
+    pass
+
+
+schema = graphene.Schema(query=Query, mutation=Mutation)