import graphene

from graphene_django.types import DjangoObjectType

from resources.models import (
    QuickLink,
    QuickLinkDescription,
    QuickLinkName,
    QuickLinkUrl,
    Category
)


class QuickLinkUrlType(DjangoObjectType):
    class Meta:
        model = QuickLinkUrl


class QuickLinkDescriptionType(DjangoObjectType):
    class Meta:
        model = QuickLinkDescription


class QuickLinkType(DjangoObjectType):
    class Meta:
        model = QuickLink


class QuickLinkNameType(DjangoObjectType):
    class Meta:
        model = QuickLinkName


class CategoryType(DjangoObjectType):
    class Meta:
        model = Category


class Query(object):
    all_quicklinks = graphene.List(QuickLinkType)

    def resolve_all_quicklinks(self, info, **kwargs):
        return QuickLink.objects.prefetch_related(
            'category',
            'name',
            'link',
            'description'
        ).all()