import graphene
from graphene_django import DjangoObjectType
from graphene import relay

from wagtail.images.models import Image

from wagtail.images.views.serve import generate_image_url


class WagtailImageNode(DjangoObjectType):

    class Meta:
        model = Image
        fields = [
            "title",
            "width",
            "height",
        ]
        interfaces = (relay.Node,)

    src = graphene.String()
    alt = graphene.String()


    def resolve_src(self, info):
        return generate_image_url(self, 'original')

    def resolve_alt(self, info):
        return ""