From 474e9154021bb4febb8f5a2ee3bc7557bd5f6944 Mon Sep 17 00:00:00 2001 From: richard-dds Date: Mon, 17 Sep 2018 15:16:57 -0400 Subject: [PATCH] Ensure user has owner role on workspace --- atst/routes/__init__.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/atst/routes/__init__.py b/atst/routes/__init__.py index bc15e0cb..b3538354 100644 --- a/atst/routes/__init__.py +++ b/atst/routes/__init__.py @@ -23,7 +23,10 @@ def home(): if num_workspaces == 0: return redirect(url_for("requests.requests_index")) elif num_workspaces == 1: - workspace_id = g.current_user.workspace_roles[0].workspace_id + workspace_role = g.current_user.workspace_roles[0] + workspace_id = workspace_role.workspace.id + is_request_owner = workspace_role.role.name == "owner" + if is_request_owner: return redirect( url_for("workspaces.workspace_reports", workspace_id=workspace_id)