Skip to content

Commit a5cf49a

Browse files
author
Jared Storts
committed
[EF-369] Add roles to project members qgl query
1 parent b2749c4 commit a5cf49a

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

labelbox/schema/project.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ def members(self) -> PaginatedCollection:
150150
"""
151151
id_param = "projectId"
152152
query_str = """query ProjectMemberOverviewPyApi($%s: ID!) {
153-
project(where: {id : $%s}) { id members(skip: %%d first: %%d){ id user { %s } role { id name } }
153+
project(where: {id : $%s}) { id members(skip: %%d first: %%d){ id user { %s } role { id name } accessFrom }
154154
}
155155
}""" % (id_param, id_param, query.results_query_part(Entity.User))
156156
return PaginatedCollection(self.client, query_str,
@@ -1360,6 +1360,7 @@ def __check_data_rows_have_been_processed(
13601360
class ProjectMember(DbObject):
13611361
user = Relationship.ToOne("User", cache=True)
13621362
role = Relationship.ToOne("Role", cache=True)
1363+
access_from = Field.String("access_from")
13631364

13641365

13651366
class LabelingParameterOverride(DbObject):

tests/integration/test_user_management.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ def test_project_invite(client, organization, project_pack, queries):
6060
for proj_role in [project_role_1, project_role_2]])
6161

6262
project_members = project_1.members()
63+
6364
project_member = [
6465
member for member in project_members
6566
if member.user().uid == client.get_user().uid
@@ -68,6 +69,7 @@ def test_project_invite(client, organization, project_pack, queries):
6869
assert len(project_member) == 1
6970
project_member = project_member[0]
7071

72+
assert project_member.access_from == 'ORGANIZATION'
7173
assert project_member.role().name.upper() == roles['ADMIN'].name.upper()
7274
queries.cancel_invite(client, invite.uid)
7375

0 commit comments

Comments
 (0)