import graphene
from graphene import relay
from wagtail.images.models import Image
from api.graphene_wagtail import generate_image_url

class ModuleInterface(relay.Node):
    pk = graphene.Int()
    hero_image = graphene.Field('books.schema.nodes.WagtailImageNode')
    topic = graphene.Field('books.schema.nodes.TopicNode')

    @staticmethod
    def resolve_pk(parent, info, **kwargs):
        return parent.id