mirror of
https://github.com/graphql-python/graphene-django.git
synced 2024-11-15 06:08:59 +03:00
56 lines
1.4 KiB
Python
56 lines
1.4 KiB
Python
import graphene
|
|
|
|
from graphene_django.types import DjangoObjectType
|
|
|
|
from cookbook.ingredients.models import Category, Ingredient
|
|
|
|
|
|
class CategoryType(DjangoObjectType):
|
|
class Meta:
|
|
model = Category
|
|
|
|
|
|
class IngredientType(DjangoObjectType):
|
|
class Meta:
|
|
model = Ingredient
|
|
|
|
|
|
class Query(object):
|
|
category = graphene.Field(CategoryType, id=graphene.Int(), name=graphene.String())
|
|
all_categories = graphene.List(CategoryType)
|
|
|
|
ingredient = graphene.Field(
|
|
IngredientType, id=graphene.Int(), name=graphene.String()
|
|
)
|
|
all_ingredients = graphene.List(IngredientType)
|
|
|
|
def resolve_all_categories(self, info, **kwargs):
|
|
return Category.objects.all()
|
|
|
|
def resolve_all_ingredients(self, info, **kwargs):
|
|
return Ingredient.objects.all()
|
|
|
|
def resolve_category(self, info, **kwargs):
|
|
id = kwargs.get("id")
|
|
name = kwargs.get("name")
|
|
|
|
if id is not None:
|
|
return Category.objects.get(pk=id)
|
|
|
|
if name is not None:
|
|
return Category.objects.get(name=name)
|
|
|
|
return None
|
|
|
|
def resolve_ingredient(self, info, **kwargs):
|
|
id = kwargs.get("id")
|
|
name = kwargs.get("name")
|
|
|
|
if id is not None:
|
|
return Ingredient.objects.get(pk=id)
|
|
|
|
if name is not None:
|
|
return Ingredient.objects.get(name=name)
|
|
|
|
return None
|