diff --git a/portal/graphql_schema.py b/portal/graphql_schema.py index 467fe0aa29eac8708fb5dc9889f4218cb15b80c9..137399ba6d7e877960f6801e148e06530ab33790 100644 --- a/portal/graphql_schema.py +++ b/portal/graphql_schema.py @@ -89,11 +89,18 @@ class Query(graphene.ObjectType): in_id_list=graphene.List(graphene.String) ) datasets = graphene.List(DataSetType, in_id_list=graphene.List(graphene.String)) - lab = graphene.Field(LabType, name=graphene.String(), dbid=graphene.ID()) + lab = graphene.Field( + LabType, + name=graphene.String(), + dbid=graphene.ID() + ) labs = graphene.List( LabType, in_name_list=graphene.List(graphene.String) ) + my_profile = graphene.Field( + ProfileType + ) profiles = graphene.List( ProfileType, in_email_list=graphene.List(graphene.String) @@ -190,6 +197,9 @@ class Query(graphene.ObjectType): return qs.distinct() + def resolve_my_profile(self, info, **kwargs): + return Profile.objects.get(user=info.context.user) + def resolve_profiles(self, info, **kwargs): email_list = kwargs.get('in_email_list') qs = Profile.objects.all()