filemanager.py

import graphene
from graphene import ObjectType, Int, String
from graphene_federation import build_schema, key

from utils import file_loader  # custom data loader


@key(fields='id')
class FileNode(ObjectType):
    id = Int(required=True)
    url = String(required=True)

    def __resolve_reference(self, info, **kwargs):
        if not info.context.user.is_authenticated:
            return None
        return file_loader.load(self.id)


class Query(ObjectType):
    pass


schema = build_schema(Query, types=[FileNode])