pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add Authorization.is_in_workspace

Browse Source
This commit is contained in:
richard-dds 2018-08-22 11:05:51 -04:00
parent e2e6e6da4d
commit 3eb9076b03
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
atst/domain/authz.py
View File

@ -6,3 +6,7 @@ class Authorization(object):
def has_workspace_permission(cls, user, workspace, permission): def has_workspace_permission(cls, user, workspace, permission):
workspace_user = WorkspaceUsers.get(workspace.id, user.id) workspace_user = WorkspaceUsers.get(workspace.id, user.id)
return permission in workspace_user.permissions() return permission in workspace_user.permissions()
@classmethod
def is_in_workspace(cls, user, workspace):
return user in workspace.users

2
atst/domain/workspaces.py
View File

@ -39,7 +39,7 @@ class Workspaces(object):
except NoResultFound: except NoResultFound:
raise NotFoundError("workspace") raise NotFoundError("workspace")
if user not in workspace.users: if not Authorization.is_in_workspace(user, workspace):
raise UnauthorizedError(user, "get workspace") raise UnauthorizedError(user, "get workspace")
return workspace return workspace