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,17 +25,14 @@ 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 ):
29
29
return Category .objects .all ()
30
30
31
- def resolve_all_ingredients (self , args , context , info ):
31
+ def resolve_all_ingredients (self , context ):
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' )
38
-
35
+ def resolve_category (self , context , id = None , name = None ):
39
36
if id is not None :
40
37
return Category .objects .get (pk = id )
41
38
@@ -44,10 +41,7 @@ def resolve_category(self, args, context, info):
44
41
45
42
return None
46
43
47
- def resolve_ingredient (self , args , context , info ):
48
- id = args .get ('id' )
49
- name = args .get ('name' )
50
-
44
+ def resolve_ingredient (self , context , id = None , name = None ):
51
45
if id is not None :
52
46
return Ingredient .objects .get (pk = id )
53
47
0 commit comments