Start TO form template

atst/routes/task_orders/new.py

@@ -1,12 +1,4 @@
-from copy import deepcopy
-
-from flask import (
-    g,
-    redirect,
-    render_template,
-    request as http_request,
-    url_for,
-)
+from flask import g, render_template, request as http_request
 
 from . import task_orders_bp
 from atst.domain.authz.decorator import user_can_access_decorator as user_can
@@ -14,19 +6,12 @@ from atst.domain.task_orders import TaskOrders
 from atst.forms.task_order import TaskOrderForm
 from atst.models.permissions import Permissions
 from atst.utils.flash import formatted_flash as flash
-from atst.utils.localization import translate
-
-
-@task_orders_bp.route("/task_orders/new/get_started")
-# TODO: see if this route still exists in new design
-def get_started():
-    return render_template("task_orders/new/get_started.html")  # pragma: no cover
 
 
 @task_orders_bp.route("/portfolios/<portfolio_id>/task_orders/new")
 @user_can(Permissions.CREATE_TASK_ORDER, message="view new task order form")
 def new(portfolio_id):
-    return render_template("task_orders/new", form=TaskOrderForm())
+    return render_template("task_orders/new.html", form=TaskOrderForm())
 
 
 @task_orders_bp.route("/portfolios/<portfolio_id>/task_orders/new", methods=["POST"])
@@ -37,7 +22,11 @@ def create(portfolio_id):
 
     if form.validate():
         TaskOrders.create(g.current_user, portfolio_id, **form.data)
-    # TODO: ask UX where do you go after save
+        flash("task_order_draft")
+        return render_template("task_orders/new.html", form=form)
+    else:
+        flash("form_errors")
+        return render_template("task_orders/new.html", form=form)
 
 
 @task_orders_bp.route("/portfolios/<portfolio_id>/task_orders/<task_order_id>/edit")
@@ -56,4 +45,8 @@ def update(portfolio_id, task_order_id=None):
 
     if form.validate():
         TaskOrders.update(task_order_id, **form.data)
-    # TODO: ask UX where do you go after save
+        flash("task_order_draft")
+        return render_template("task_orders/new.html", form=form)
+    else:
+        flash("form_errors")
+        return render_template("task_orders/new.html", form=form)