27 lines
813 B
Python
27 lines
813 B
Python
import graphene
|
|
from graphene import relay
|
|
from graphene_django import DjangoObjectType
|
|
from graphene_django.filter import DjangoFilterConnectionField
|
|
from .chapter import SnapshotChapterNode
|
|
|
|
from books.models.snapshot import Snapshot
|
|
|
|
|
|
class SnapshotNode(DjangoObjectType):
|
|
title = graphene.String()
|
|
|
|
class Meta:
|
|
model = Snapshot
|
|
interfaces = (relay.Node,)
|
|
|
|
# chapters = relay.ConnectionField('books.schema.connections.ChapterSnapshotConnection')
|
|
chapters = DjangoFilterConnectionField(SnapshotChapterNode)
|
|
|
|
def resolve_chapters(self, info, **kwargs):
|
|
# return Chapter.objects.filter(chapter_snapshots__snapshot=self)
|
|
return self.chapters.through.objects.all()
|
|
|
|
@staticmethod
|
|
def resolve_title(parent, info, **kwargs):
|
|
return parent.__str__()
|