|
| 1 | +# coding=utf-8 |
| 2 | + |
| 3 | +from ..base import BitbucketCloudBase |
| 4 | + |
| 5 | + |
| 6 | +class WorkspaceMembers(BitbucketCloudBase): |
| 7 | + def __init__(self, url, *args, **kwargs): |
| 8 | + super(WorkspaceMembers, self).__init__(url, *args, **kwargs) |
| 9 | + |
| 10 | + def __get_object(self, data): |
| 11 | + return WorkspaceMember(data, **self._new_session_args) |
| 12 | + |
| 13 | + def each(self): |
| 14 | + """ |
| 15 | + Get all members in the workspace |
| 16 | +
|
| 17 | + :return: A generator for the member objects |
| 18 | +
|
| 19 | + API docs: https://developer.atlassian.com/cloud/bitbucket/rest/api-group-workspaces/#api-workspaces-workspace-members-get |
| 20 | + """ |
| 21 | + for member in self._get_paged(None): |
| 22 | + yield self.__get_object(member) |
| 23 | + |
| 24 | + return |
| 25 | + |
| 26 | + def get(self, member): |
| 27 | + """ |
| 28 | + Returns the requested member |
| 29 | +
|
| 30 | + :param member: string: Member's UUID or Atlassian ID. |
| 31 | +
|
| 32 | + :return: The requested Member object |
| 33 | +
|
| 34 | + API docs: https://developer.atlassian.com/cloud/bitbucket/rest/api-group-workspaces/#api-workspaces-workspace-members-member-get |
| 35 | + """ |
| 36 | + |
| 37 | + return self.__get_object(super(WorkspaceMembers, self).get(member)) |
| 38 | + |
| 39 | + |
| 40 | +class WorkspaceMember(BitbucketCloudBase): |
| 41 | + def __init__(self, data, *args, **kwargs): |
| 42 | + super(WorkspaceMember, self).__init__(None, *args, data=data, expected_type="workspace_membership", **kwargs) |
| 43 | + |
| 44 | + @property |
| 45 | + def links(self): |
| 46 | + """The member links""" |
| 47 | + return self.get_data("links") |
| 48 | + |
| 49 | + @property |
| 50 | + def type(self): |
| 51 | + """The member type""" |
| 52 | + return self.get_data("type") |
| 53 | + |
| 54 | + @property |
| 55 | + def user(self): |
| 56 | + """The member user dictionary""" |
| 57 | + return self.get_data("user") |
| 58 | + |
| 59 | + @property |
| 60 | + def workspace(self): |
| 61 | + """The member workspace dictionary""" |
| 62 | + return self.get_data("workspace") |
0 commit comments