From 43757d5d3bca677121845548fc1ce22bdc6b339d Mon Sep 17 00:00:00 2001 From: Patrick Smith Date: Thu, 10 Jan 2019 12:14:07 -0500 Subject: [PATCH] Pass in current user when fetching task order --- atst/routes/task_orders/invite.py | 4 ++-- atst/routes/workspaces/task_orders.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/atst/routes/task_orders/invite.py b/atst/routes/task_orders/invite.py index c43f14b5..ea55ef9b 100644 --- a/atst/routes/task_orders/invite.py +++ b/atst/routes/task_orders/invite.py @@ -1,4 +1,4 @@ -from flask import redirect, url_for +from flask import g, redirect, url_for from . import task_orders_bp from atst.domain.task_orders import TaskOrders @@ -8,7 +8,7 @@ from atst.utils.flash import formatted_flash as flash # TODO: add a real implementation for this @task_orders_bp.route("/task_orders/invite/", methods=["POST"]) def invite(task_order_id): - task_order = TaskOrders.get(task_order_id) + task_order = TaskOrders.get(g.current_user, task_order_id) flash("task_order_submitted", task_order=task_order) return redirect( url_for("workspaces.workspace_members", workspace_id=task_order.workspace.id) diff --git a/atst/routes/workspaces/task_orders.py b/atst/routes/workspaces/task_orders.py index 7fdcbc9a..d99d6c3b 100644 --- a/atst/routes/workspaces/task_orders.py +++ b/atst/routes/workspaces/task_orders.py @@ -14,7 +14,7 @@ def workspace_task_orders(workspace_id): @workspaces_bp.route("/workspaces//task_order/") def view_task_order(workspace_id, task_order_id): workspace = Workspaces.get(g.current_user, workspace_id) - task_order = TaskOrders.get(task_order_id) + task_order = TaskOrders.get(g.current_user, task_order_id) return render_template( "workspaces/task_orders/show.html", workspace=workspace, task_order=task_order )