import graphene
from graphene_django.filter import DjangoFilterConnectionField

from users.models import User
from users.schema_public import UserNode


class AllUsersQuery(object):
    me = graphene.Field(UserNode)
    all_users = DjangoFilterConnectionField(UserNode)

    def resolve_all_users(self, info, **kwargs):
        if not info.context.user.is_superuser:
            return User.objects.none()
        else:
            return User.objects.all()