monolith.py

from graphene import Schema, ObjectType, Int, String, List, NonNull

from utils import file_loader  # custom data loader


class FileNode(ObjectType):
    id = Int(required=True)
    url = String(required=True)


class MessageNode(ObjectType):
    id = Int(required=True)
    thread_id = Int(required=True)
    author_id = Int()
    files = List(NonNull(FileNode))
    body = String(required=True)

    def resolve_files(self, info):
        return file_loader.load(message_id=self.id)


class Query(ObjectType):
    messages = List(NonNull(MessageNode))


schema = Schema(query=Query)