graphql_python_template/app/schema/queries/todo.py
2020-10-22 13:07:38 +02:00

28 lines
933 B
Python

from typing import List
import graphene
from app.models.todo import Todo
from app.schema.types.todo import TodoType, TodoResponseField
from app.usecases import read_all_todos, read_todo_by_id
class TodoQuery(graphene.ObjectType):
"""
Defines the query and how to interact with
"""
list_todos = graphene.Field(graphene.List(TodoType))
def resolve_list_todos(self, info) -> List[Todo]:
return read_all_todos()
get_todo = graphene.Field(TodoResponseField,
todo_id=graphene.String(required=True))
def resolve_get_todo(self, info, todo_id: str) -> TodoResponseField:
todo, is_success = read_todo_by_id(todo_id)
error_message = "This element does not exist." if not is_success else None
return TodoResponseField(todo=todo,
is_success=is_success,
error_message=error_message)