1- import graphene
1+ import graphene
22
3- from graphene_django .types import DjangoObjectType
3+ from graphene_django .types import DjangoObjectType
44
5- from cookbook .ingredients .models import Category , Ingredient
5+ from cookbook .ingredients .models import Category , Ingredient
66
77
8- class CategoryType (DjangoObjectType ):
9- class Meta :
10- model = Category
8+ class CategoryType (DjangoObjectType ):
9+ class Meta :
10+ model = Category
1111
1212
13- class IngredientType (DjangoObjectType ):
14- class Meta :
15- model = Ingredient
13+ class IngredientType (DjangoObjectType ):
14+ class Meta :
15+ model = Ingredient
1616
1717
18- class Query (object ):
19- category = graphene .Field (CategoryType ,
20- id = graphene .Int (),
21- name = graphene .String ())
22- all_categories = graphene .List (CategoryType )
18+ class Query (object ):
19+ category = graphene .Field (CategoryType , id = graphene .Int (), name = graphene .String ())
20+ all_categories = graphene .List (CategoryType )
2321
22+ ingredient = graphene .Field (
23+ IngredientType , id = graphene .Int (), name = graphene .String ()
24+ )
25+ all_ingredients = graphene .List (IngredientType )
2426
25- ingredient = graphene .Field (IngredientType ,
26- id = graphene .Int (),
27- name = graphene .String ())
28- all_ingredients = graphene .List (IngredientType )
27+ def resolve_all_categories (self , info , ** kwargs ):
28+ return Category .objects .all ()
2929
30- def resolve_all_categories (self , info , ** kwargs ):
31- return Category .objects .all ()
30+ def resolve_all_ingredients (self , info , ** kwargs ):
31+ return Ingredient .objects .all ()
3232
33- def resolve_all_ingredients (self , info , ** kwargs ):
34- return Ingredient .objects .all ()
33+ def resolve_category (self , info , ** kwargs ):
34+ id = kwargs .get ("id" )
35+ name = kwargs .get ("name" )
3536
36- def resolve_category (self , info , ** kwargs ):
37- id = kwargs .get ('id' )
38- name = kwargs .get ('name' )
37+ if id is not None :
38+ return Category .objects .get (pk = id )
3939
40- if id is not None :
41- return Category .objects .get (pk = id )
40+ if name is not None :
41+ return Category .objects .get (name = name )
4242
43- if name is not None :
44- return Category .objects .get (name = name )
43+ return None
4544
46- return None
45+ def resolve_ingredient (self , info , ** kwargs ):
46+ id = kwargs .get ("id" )
47+ name = kwargs .get ("name" )
4748
48- def resolve_ingredient (self , info , ** kwargs ):
49- id = kwargs .get ('id' )
50- name = kwargs .get ('name' )
49+ if id is not None :
50+ return Ingredient .objects .get (pk = id )
5151
52- if id is not None :
53- return Ingredient .objects .get (pk = id )
52+ if name is not None :
53+ return Ingredient .objects .get (name = name )
5454
55- if name is not None :
56- return Ingredient .objects .get (name = name )
57-
58- return None
55+ return None
0 commit comments