Delete TO draft when click "delete it" on cancel modal

atst/routes/task_orders/new.py

@@ -39,6 +39,7 @@ def render_task_orders_edit(template, portfolio_id=None, task_order_id=None, for
         "task_orders.cancel_edit",
         task_order_id=task_order_id,
         portfolio_id=portfolio_id,
+        save=False,
     )
 
     return render_template(template, **render_args)
@@ -137,7 +138,9 @@ def submit_form_step_one_add_pdf(portfolio_id=None, task_order_id=None):
 @task_orders_bp.route("/task_orders/<task_order_id>/form/cancel", methods=["POST"])
 @user_can(Permissions.CREATE_TASK_ORDER, message="cancel task order form")
 def cancel_edit(task_order_id=None, portfolio_id=None):
-    save = http_request.args.get("save", False)
+    # Either save the currently entered data, or delete the TO
+    save = http_request.args.get("save", "True").lower() == "true"
+
     if save:
         form_data = {**http_request.form}
         form = None
@@ -155,6 +158,9 @@ def cancel_edit(task_order_id=None, portfolio_id=None):
                 task_order = TaskOrders.create(
                     g.current_user, portfolio_id, **form.data
                 )
+    elif not save and task_order_id:
+        if task_order_id:
+            TaskOrders.delete(task_order_id)
 
     return redirect(
         url_for("task_orders.portfolio_funding", portfolio_id=g.portfolio.id)