50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
import graphene
|
|
from graphene import relay
|
|
from graphene_django import DjangoObjectType
|
|
from graphene_django.filter import DjangoFilterConnectionField
|
|
|
|
from users.models import SchoolClass, User
|
|
|
|
|
|
class UserNode(DjangoObjectType):
|
|
pk = graphene.Int()
|
|
permissions = graphene.List(graphene.String)
|
|
|
|
class Meta:
|
|
model = User
|
|
filter_fields = ['username', 'email']
|
|
only_fields = ['username', 'email', 'first_name', 'last_name', 'school_classes', 'last_module', 'avatar_url']
|
|
interfaces = (relay.Node,)
|
|
|
|
def resolve_pk(self, info, **kwargs):
|
|
return self.id
|
|
|
|
def resolve_permissions(self, info):
|
|
return self.get_all_permissions()
|
|
|
|
|
|
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 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()
|