from graphene import relay
from graphene_django import DjangoObjectType
from graphene_django.filter import DjangoFilterConnectionField

from newsteaser.models import NewsTeaser


class NewsTeaserNode(DjangoObjectType):

    class Meta:
        model = NewsTeaser
        filter_fields = []
        interfaces = (relay.Node,)


class AllNewsTeasersQuery(object):
    articles = DjangoFilterConnectionField(NewsTeaserNode)

    def resolve_articles(self, info, **kwargs):
        return NewsTeaser.objects.all()