mirror of
https://github.com/graphql-python/graphene-django.git
synced 2024-11-22 01:27:01 +03:00
4517e32224
* 🔧 Add pre-commit config Similar to graphene and graphene-sqlalchemy * ⬆ Bump black * 👷 Lint on CI * ⬆ Bump flake8-black * 🔧 Keep excluding migrations * ⬆ Bump flake8 * 🔧 Remove black and flake8 from tox config * ⬆ Update pre-commit versions * Upgrade syntax to python 3.7+ * Format with pre-commit dedent docs/schema.py to allow formatting * Fix tests on python 3.7
58 lines
1.4 KiB
Python
58 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
|
|
fields = "__all__"
|
|
|
|
|
|
class IngredientType(DjangoObjectType):
|
|
class Meta:
|
|
model = Ingredient
|
|
fields = "__all__"
|
|
|
|
|
|
class Query:
|
|
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
|