122 lines
3.9 KiB
Python
122 lines
3.9 KiB
Python
import graphene
|
|
from api.utils import get_object
|
|
from core.logger import get_logger
|
|
from graphene import relay
|
|
from graphene_django.filter import DjangoFilterConnectionField
|
|
from wagtail.models import Locale
|
|
|
|
from ..models import Book, Chapter, Module, ModuleCategory, ModuleLevel, Snapshot, Topic
|
|
from .connections import ModuleConnection, TopicConnection
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
class BookQuery(object):
|
|
node = relay.Node.Field()
|
|
topic = graphene.Field(
|
|
"books.schema.nodes.topic.TopicOr404Node", slug=graphene.String()
|
|
)
|
|
module = graphene.Field(
|
|
"books.schema.nodes.module.ModuleNode", slug=graphene.String(), id=graphene.ID()
|
|
)
|
|
chapter = relay.Node.Field("books.schema.nodes.chapter.ChapterNode")
|
|
content_block = relay.Node.Field("books.schema.nodes.content.ContentBlockNode")
|
|
snapshot = relay.Node.Field("books.schema.nodes.snapshot.SnapshotNode")
|
|
|
|
topics = relay.ConnectionField(TopicConnection)
|
|
modules = relay.ConnectionField(ModuleConnection)
|
|
chapters = DjangoFilterConnectionField("books.schema.nodes.chapter.ChapterNode")
|
|
|
|
module_level = graphene.Field(
|
|
"books.schema.nodes.module_level.ModuleLevelNode", id=graphene.ID(required=True)
|
|
)
|
|
module_levels = graphene.List("books.schema.nodes.module_level.ModuleLevelNode")
|
|
|
|
module_category = graphene.Field(
|
|
"books.schema.nodes.module_category.ModuleCategoryNode",
|
|
id=graphene.ID(required=True),
|
|
)
|
|
module_categories = graphene.List(
|
|
"books.schema.nodes.module_category.ModuleCategoryNode"
|
|
)
|
|
|
|
def resolve_books(self, *args, **kwargs):
|
|
return Book.objects.filter(**kwargs).live()
|
|
|
|
def resolve_topics(self, *args, **kwargs):
|
|
default_locale = Locale.get_default()
|
|
return Topic.objects.live().filter(locale=default_locale)
|
|
|
|
def resolve_modules(self, *args, **kwargs):
|
|
return Module.objects.filter(**kwargs).live()
|
|
|
|
def resolve_chapters(self, *args, **kwargs):
|
|
return Chapter.objects.filter(**kwargs).live()
|
|
|
|
def resolve_snapshot(self, info, **kwargs):
|
|
id = kwargs.get("id")
|
|
snapshot = get_object(Snapshot, id)
|
|
return snapshot
|
|
|
|
def resolve_module(self, info, **kwargs):
|
|
slug = kwargs.get("slug")
|
|
id = kwargs.get("id")
|
|
module = None
|
|
try:
|
|
if id is not None:
|
|
module = get_object(Module, id)
|
|
|
|
elif slug is not None:
|
|
module = Module.objects.get(slug=slug)
|
|
|
|
return module
|
|
|
|
except Module.DoesNotExist:
|
|
return None
|
|
|
|
def resolve_topic(self, info, **kwargs):
|
|
slug = kwargs.get("slug")
|
|
id = kwargs.get("id")
|
|
|
|
if id is not None:
|
|
return get_object(Topic, id)
|
|
if slug is not None:
|
|
try:
|
|
default_locale = Locale.get_default()
|
|
return Topic.objects.get(slug=slug, locale=default_locale)
|
|
except Topic.DoesNotExist:
|
|
from books.schema.nodes.topic import NotFoundFailure
|
|
|
|
return NotFoundFailure
|
|
return None
|
|
|
|
def resolve_module_level(self, info, **kwargs):
|
|
module_level_id = kwargs.get("id")
|
|
try:
|
|
if module_level_id is not None:
|
|
return get_object(Module, module_level_id)
|
|
|
|
except Module.DoesNotExist:
|
|
return None
|
|
|
|
def resolve_module_levels(self, *args, **kwargs):
|
|
return ModuleLevel.objects.all()
|
|
|
|
def resolve_module_category(self, info, **kwargs):
|
|
id = kwargs.get("id")
|
|
try:
|
|
if id is not None:
|
|
return get_object(Module, id)
|
|
|
|
except Module.DoesNotExist:
|
|
return None
|
|
|
|
def resolve_module_categories(self, *args, **kwargs):
|
|
return ModuleCategory.objects.all()
|
|
|
|
|
|
class ModuleTypeQuery(graphene.ObjectType):
|
|
node = relay.Node.Field()
|
|
name = graphene.String()
|
|
id = graphene.ID()
|