Add route to update TO officer information

atst/routes/portfolios/task_orders.py

@@ -1,11 +1,13 @@
 from collections import defaultdict
 from operator import itemgetter
 
-from flask import g, render_template, url_for
+from flask import g, redirect, render_template, url_for, request as http_request
 
 from . import portfolios_bp
+from atst.database import db
 from atst.domain.task_orders import TaskOrders
 from atst.domain.portfolios import Portfolios
+from atst.forms.officers import EditTaskOrderOfficersForm
 from atst.models.task_order import Status as TaskOrderStatus
 
 
@@ -69,8 +71,40 @@ def view_task_order(portfolio_id, task_order_id):
 def task_order_invitations(portfolio_id, task_order_id):
     portfolio = Portfolios.get(g.current_user, portfolio_id)
     task_order = TaskOrders.get(g.current_user, task_order_id)
+    form = EditTaskOrderOfficersForm(obj=task_order)
     return render_template(
         "portfolios/task_orders/invitations.html",
         portfolio=portfolio,
         task_order=task_order,
+        form=form,
     )
+
+
+@portfolios_bp.route(
+    "/portfolios/<portfolio_id>/task_order/<task_order_id>/invitations",
+    methods=["POST"],
+)
+def edit_task_order_invitations(portfolio_id, task_order_id):
+    portfolio = Portfolios.get(g.current_user, portfolio_id)
+    task_order = TaskOrders.get(g.current_user, task_order_id)
+    form = EditTaskOrderOfficersForm(formdata=http_request.form, obj=task_order)
+
+    if form.validate():
+        form.populate_obj(task_order)
+        db.session.add(task_order)
+        db.session.commit()
+
+        return redirect(
+            url_for(
+                "portfolios.task_order_invitations",
+                portfolio_id=portfolio.id,
+                task_order_id=task_order.id,
+            )
+        )
+    else:
+        return render_template(
+            "portfolios/task_orders/invitations.html",
+            portfolio=portfolio,
+            task_order=task_order,
+            form=form,
+        )