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

from users.models import SchoolClass, User


class SchoolClassNode(DjangoObjectType):
    pk = graphene.Int()

    class Meta:
        model = SchoolClass
        filter_fields = ['name']
        interfaces = (relay.Node,)

    def resolve_pk(self, *args, **kwargs):
        return self.id


class UserNode(DjangoObjectType):
    pk = graphene.Int()
    permissions = graphene.List(graphene.String)
    selected_class = graphene.Field(SchoolClassNode)

    class Meta:
        model = User
        filter_fields = ['username', 'email']
        only_fields = ['username', 'email', 'first_name', 'last_name', 'school_classes', 'last_module', 'avatar_url',
                       'selected_class']
        interfaces = (relay.Node,)

    def resolve_pk(self, info, **kwargs):
        return self.id

    def resolve_permissions(self, info):
        return self.get_all_permissions()

    def resolve_selected_class(self, info):
        return self.selected_class()


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

    def resolve_me(self, info, **kwargs):
        return info.context.user

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