domain methods for checking task order sections

atst/domain/task_orders.py

@@ -6,6 +6,42 @@ from .exceptions import NotFoundError
 
 
 class TaskOrders(object):
+    SECTIONS = {
+        "app_info": [
+            "scope",
+            "defense_component",
+            "app_migration",
+            "native_apps",
+            "complexity",
+            "complexity_other",
+            "dev_team",
+            "dev_team_other",
+            "team_experience",
+        ],
+        "funding": [
+            "start_date",
+            "end_date",
+            "clin_01",
+            "clin_02",
+            "clin_03",
+            "clin_04",
+        ],
+        "oversight": [
+            "ko_first_name",
+            "ko_last_name",
+            "ko_email",
+            "ko_dod_id",
+            "cor_first_name",
+            "cor_last_name",
+            "cor_email",
+            "cor_dod_id",
+            "so_first_name",
+            "so_last_name",
+            "so_email",
+            "so_dod_id",
+        ],
+    }
+
     @classmethod
     def get(cls, task_order_id):
         try:
@@ -35,3 +71,23 @@ class TaskOrders(object):
         db.session.commit()
 
         return task_order
+
+    @classmethod
+    def is_section_complete(cls, task_order, section):
+        if section in TaskOrders.SECTIONS:
+            for attr in TaskOrders.SECTIONS[section]:
+                if not getattr(task_order, attr):
+                    return False
+
+            return True
+
+        else:
+            return False
+
+    @classmethod
+    def all_sections_complete(cls, task_order):
+        for section in TaskOrders.SECTIONS.keys():
+            if not TaskOrders.is_section_complete(task_order, section):
+                return False
+
+        return True

atst/routes/task_orders/__init__.py

@@ -65,6 +65,18 @@ class ShowTaskOrderWorkflow:
     def template(self):
         return self._section["template"]
 
+    @property
+    def display_screens(self):
+        screen_info = TASK_ORDER_SECTIONS.copy()
+
+        if self.task_order:
+            for section in screen_info:
+                if TaskOrders.is_section_complete(self.task_order, section["section"]):
+                    section["complete"] = True
+
+        return screen_info
+
+
 
 class UpdateTaskOrderWorkflow(ShowTaskOrderWorkflow):
     def __init__(self, form_data, user, screen=1, task_order_id=None):
@@ -95,21 +107,21 @@ class UpdateTaskOrderWorkflow(ShowTaskOrderWorkflow):
         return self.task_order
 
 
-@task_orders_bp.route("/task_order/new/<int:screen>")
-@task_orders_bp.route("/task_order/new/<int:screen>/<task_order_id>")
+@task_orders_bp.route("/task_orders/new/<int:screen>")
+@task_orders_bp.route("/task_orders/new/<int:screen>/<task_order_id>")
 def new(screen, task_order_id=None):
     workflow = ShowTaskOrderWorkflow(screen, task_order_id)
     return render_template(
         workflow.template,
         current=screen,
         task_order_id=task_order_id,
-        screens=TASK_ORDER_SECTIONS,
+        screens=workflow.display_screens,
         form=workflow.form,
     )
 
 
-@task_orders_bp.route("/task_order/new/<int:screen>", methods=["POST"])
-@task_orders_bp.route("/task_order/new/<int:screen>/<task_order_id>", methods=["POST"])
+@task_orders_bp.route("/task_orders/new/<int:screen>", methods=["POST"])
+@task_orders_bp.route("/task_orders/new/<int:screen>/<task_order_id>", methods=["POST"])
 def update(screen, task_order_id=None):
     workflow = UpdateTaskOrderWorkflow(
         http_request.form, g.current_user, screen, task_order_id