Invite Officer From Manage Invitations Page

atst/forms/officers.py

@@ -35,6 +35,7 @@ class EditTaskOrderOfficersForm(CacheableForm):
         "email",
         "phone_number",
         "dod_id",
+        "invite",
     ]
 
     def process(self, formdata=None, obj=None, data=None, **kwargs):

atst/models/task_order.py

@@ -74,19 +74,19 @@ class TaskOrder(Base, mixins.TimestampsMixin):
     ko_email = Column(String)  # Email
     ko_phone_number = Column(String)  # Phone Number
     ko_dod_id = Column(String)  # DOD ID
-    ko_invite = Column(Boolean)
+    ko_invite = Column(Boolean, default=False)
     cor_first_name = Column(String)  # First Name
     cor_last_name = Column(String)  # Last Name
     cor_email = Column(String)  # Email
     cor_phone_number = Column(String)  # Phone Number
     cor_dod_id = Column(String)  # DOD ID
-    cor_invite = Column(Boolean)
+    cor_invite = Column(Boolean, default=False)
     so_first_name = Column(String)  # First Name
     so_last_name = Column(String)  # Last Name
     so_email = Column(String)  # Email
     so_phone_number = Column(String)  # Phone Number
     so_dod_id = Column(String)  # DOD ID
-    so_invite = Column(Boolean)
+    so_invite = Column(Boolean, default=False)
     pdf_attachment_id = Column(ForeignKey("attachments.id"))
     _pdf = relationship("Attachment", foreign_keys=[pdf_attachment_id])
     number = Column(String, unique=True)  # Task Order Number
@@ -95,7 +95,7 @@ class TaskOrder(Base, mixins.TimestampsMixin):
     signer_dod_id = Column(String)
     signed_at = Column(DateTime)
     level_of_warrant = Column(Numeric(scale=2))
-    unlimited_level_of_warrant = Column(Boolean)
+    unlimited_level_of_warrant = Column(Boolean, default=False)
 
     @hybrid_property
     def csp_estimate(self):

atst/routes/portfolios/task_orders.py

@@ -12,6 +12,7 @@ from atst.forms.officers import EditTaskOrderOfficersForm
 from atst.models.task_order import Status as TaskOrderStatus
 from atst.forms.ko_review import KOReviewForm
 from atst.forms.dd_254 import DD254Form
+from atst.services.invitation import update_officer_invitations
 
 
 @portfolios_bp.route("/portfolios/<portfolio_id>/task_orders")
@@ -157,6 +158,7 @@ def edit_task_order_invitations(portfolio_id, task_order_id):
         form.populate_obj(task_order)
         db.session.add(task_order)
         db.session.commit()
+        update_officer_invitations(g.current_user, task_order)
 
         return redirect(
             url_for(

atst/routes/task_orders/invite.py

@@ -3,49 +3,7 @@ from flask import redirect, url_for, g
 from . import task_orders_bp
 from atst.domain.task_orders import TaskOrders
 from atst.utils.flash import formatted_flash as flash
-from atst.domain.portfolio_roles import PortfolioRoles
-from atst.services.invitation import Invitation as InvitationService
-
-
-OFFICER_INVITATIONS = [
-    {
-        "field": "ko_invite",
-        "role": "contracting_officer",
-        "subject": "Review a task order",
-        "template": "emails/invitation.txt",
-    },
-    {
-        "field": "cor_invite",
-        "role": "contracting_officer_representative",
-        "subject": "Help with a task order",
-        "template": "emails/invitation.txt",
-    },
-    {
-        "field": "so_invite",
-        "role": "security_officer",
-        "subject": "Review security for a task order",
-        "template": "emails/invitation.txt",
-    },
-]
-
-
-def update_officer_invitations(user, task_order):
-    for officer_type in OFFICER_INVITATIONS:
-        field = officer_type["field"]
-        if getattr(task_order, field) and not getattr(task_order, officer_type["role"]):
-            officer_data = task_order.officer_dictionary(officer_type["role"])
-            officer = TaskOrders.add_officer(
-                user, task_order, officer_type["role"], officer_data
-            )
-            pf_officer_member = PortfolioRoles.get(task_order.portfolio.id, officer.id)
-            invite_service = InvitationService(
-                user,
-                pf_officer_member,
-                officer_data["email"],
-                subject=officer_type["subject"],
-                email_template=officer_type["template"],
-            )
-            invite_service.invite()
+from atst.services.invitation import update_officer_invitations
 
 
 @task_orders_bp.route("/task_orders/invite/<task_order_id>", methods=["POST"])

atst/services/invitation.py

@@ -2,6 +2,48 @@ from flask import render_template
 
 from atst.domain.invitations import Invitations
 from atst.queue import queue
+from atst.domain.task_orders import TaskOrders
+from atst.domain.portfolio_roles import PortfolioRoles
+
+OFFICER_INVITATIONS = [
+    {
+        "field": "ko_invite",
+        "role": "contracting_officer",
+        "subject": "Review a task order",
+        "template": "emails/invitation.txt",
+    },
+    {
+        "field": "cor_invite",
+        "role": "contracting_officer_representative",
+        "subject": "Help with a task order",
+        "template": "emails/invitation.txt",
+    },
+    {
+        "field": "so_invite",
+        "role": "security_officer",
+        "subject": "Review security for a task order",
+        "template": "emails/invitation.txt",
+    },
+]
+
+
+def update_officer_invitations(user, task_order):
+    for officer_type in OFFICER_INVITATIONS:
+        field = officer_type["field"]
+        if getattr(task_order, field) and not getattr(task_order, officer_type["role"]):
+            officer_data = task_order.officer_dictionary(officer_type["role"])
+            officer = TaskOrders.add_officer(
+                user, task_order, officer_type["role"], officer_data
+            )
+            pf_officer_member = PortfolioRoles.get(task_order.portfolio.id, officer.id)
+            invite_service = Invitation(
+                user,
+                pf_officer_member,
+                officer_data["email"],
+                subject=officer_type["subject"],
+                email_template=officer_type["template"],
+            )
+            invite_service.invite()
 
 
 class Invitation: