1
1
import graphene
2
2
from graphene_django .types import DjangoObjectType
3
3
4
- from cookbook . ingredients .models import Category , Ingredient
4
+ from .models import Category , Ingredient
5
5
6
6
7
7
class CategoryType (DjangoObjectType ):
@@ -25,16 +25,16 @@ class Query(object):
25
25
name = graphene .String ())
26
26
all_ingredients = graphene .List (IngredientType )
27
27
28
- def resolve_all_categories (self , args , context , info ):
28
+ def resolve_all_categories (self , context , ** kwargs ):
29
29
return Category .objects .all ()
30
30
31
- def resolve_all_ingredients (self , args , context , info ):
31
+ def resolve_all_ingredients (self , context , ** kwargs ):
32
32
# We can easily optimize query count in the resolve method
33
33
return Ingredient .objects .select_related ('category' ).all ()
34
34
35
- def resolve_category (self , args , context , info ):
36
- id = args .get ('id' )
37
- name = args .get ('name' )
35
+ def resolve_category (self , context , ** kwargs ):
36
+ id = kwargs .get ('id' )
37
+ name = kwargs .get ('name' )
38
38
39
39
if id is not None :
40
40
return Category .objects .get (pk = id )
@@ -44,9 +44,9 @@ def resolve_category(self, args, context, info):
44
44
45
45
return None
46
46
47
- def resolve_ingredient (self , args , context , info ):
48
- id = args .get ('id' )
49
- name = args .get ('name' )
47
+ def resolve_ingredient (self , context , ** kwargs ):
48
+ id = kwargs .get ('id' )
49
+ name = kwargs .get ('name' )
50
50
51
51
if id is not None :
52
52
return Ingredient .objects .get (pk = id )
0 commit comments