From ed08bdfd52230b5c1aef9b5c2967eb8306e9ef16 Mon Sep 17 00:00:00 2001
From: leigh-mil <leigh-ctr@friends.dds.mil>
Date: Mon, 11 Mar 2019 16:03:15 -0400
Subject: [PATCH] Check if officer has been invited to the TO

---
 atst/routes/task_orders/new.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/atst/routes/task_orders/new.py b/atst/routes/task_orders/new.py
index 799fd024..cd803b80 100644
--- a/atst/routes/task_orders/new.py
+++ b/atst/routes/task_orders/new.py
@@ -92,11 +92,14 @@ class ShowTaskOrderWorkflow:
             elif self._section["section"] == "oversight":
                 if self.user.dod_id == self.task_order.cor_dod_id:
                     self._form.am_cor.data = True
-                if self.task_order.contracting_officer:
+                if self.task_order.contracting_officer or self.task_order.ko_invite:
                     self._form.ko_invite.data = True
-                if self.task_order.contracting_officer_representative:
+                if (
+                    self.task_order.contracting_officer_representative
+                    or self.task_order.cor_invite
+                ):
                     self._form.cor_invite.data = True
-                if self.task_order.security_officer:
+                if self.task_order.security_officer or self.task_order.so_invite:
                     self._form.so_invite.data = True
 
         else: