From 6c0df3b0cb4125053c705485e5ed9be7ffb20b5b Mon Sep 17 00:00:00 2001 From: richard-dds Date: Thu, 6 Sep 2018 11:49:57 -0400 Subject: [PATCH] Use check_workspace_permission --- atst/domain/workspaces.py | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/atst/domain/workspaces.py b/atst/domain/workspaces.py index f3fc8c63..d7cf9f5b 100644 --- a/atst/domain/workspaces.py +++ b/atst/domain/workspaces.py @@ -30,10 +30,9 @@ class Workspaces(object): except NoResultFound: raise NotFoundError("workspace") - if not Authorization.has_workspace_permission( - user, workspace, Permissions.VIEW_WORKSPACE - ): - raise UnauthorizedError(user, "get workspace") + Authorization.check_workspace_permission( + user, workspace, Permissions.VIEW_WORKSPACE, "get workspace" + ) return workspace @@ -58,10 +57,9 @@ class Workspaces(object): @classmethod def get_with_members(cls, user, workspace_id): workspace = Workspaces.get(user, workspace_id) - if not Authorization.has_workspace_permission( - user, workspace, Permissions.VIEW_WORKSPACE_MEMBERS - ): - raise UnauthorizedError(user, "view workspace members") + Authorization.check_workspace_permission( + user, workspace, Permissions.VIEW_WORKSPACE_MEMBERS, "view workspace members" + ) return workspace @classmethod