From 99ab0c22bcb9c3a47255a25d05753fe6c50e9728 Mon Sep 17 00:00:00 2001
From: Montana <montana-ctr@friends.dds.mil>
Date: Mon, 10 Jun 2019 14:03:09 -0400
Subject: [PATCH] Add Success banner on submit

---
 atst/routes/task_orders/index.py            | 7 +++++--
 atst/utils/flash.py                         | 2 +-
 templates/portfolios/task_orders/index.html | 2 ++
 3 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/atst/routes/task_orders/index.py b/atst/routes/task_orders/index.py
index 2a230719..63e8a5cf 100644
--- a/atst/routes/task_orders/index.py
+++ b/atst/routes/task_orders/index.py
@@ -6,9 +6,10 @@ from . import task_orders_bp
 from atst.domain.authz.decorator import user_can_access_decorator as user_can
 from atst.domain.portfolios import Portfolios
 from atst.domain.task_orders import TaskOrders
-from atst.models import Permissions
-from atst.models.task_order import Status
 from atst.forms.task_order import TaskOrderForm, SignatureForm
+from atst.models import Permissions
+from atst.models.task_order import Status as TaskOrderStatus
+from atst.utils.flash import formatted_flash as flash
 
 
 @task_orders_bp.route("/task_orders/<task_order_id>")
@@ -43,6 +44,8 @@ def review_task_order(task_order_id):
 @user_can(Permissions.CREATE_TASK_ORDER, "submit task order")
 def submit_task_order(task_order_id):
     task_order = TaskOrders.get(task_order_id)
+    flash("task_order_submitted", task_order=task_order)
+
     return redirect(
         url_for("task_orders.portfolio_funding", portfolio_id=task_order.portfolio.id)
     )
diff --git a/atst/utils/flash.py b/atst/utils/flash.py
index a7d5b52d..e70ba417 100644
--- a/atst/utils/flash.py
+++ b/atst/utils/flash.py
@@ -170,7 +170,7 @@ MESSAGES = {
         "category": "success",
     },
     "task_order_submitted": {
-        "title_template": "Task Order Form Submitted",
+        "title_template": "Your Task Order has been uploaded successfully.",
         "message_template": """
         Your task order form for {{ task_order.portfolio_name }} has been submitted.
         """,
diff --git a/templates/portfolios/task_orders/index.html b/templates/portfolios/task_orders/index.html
index 0e4c52b8..9b561678 100644
--- a/templates/portfolios/task_orders/index.html
+++ b/templates/portfolios/task_orders/index.html
@@ -68,6 +68,8 @@
     <a href="{{ url_for("task_orders.edit", portfolio_id=portfolio.id) }}" class="usa-button usa-button-primary" type="submit">Start a new task order</a>
 {% endcall %}
 
+{% include "fragments/flash.html" %}
+
 <div class="portfolio-funding">
 
   {% if task_orders  %}