Clean up code from old version of TO builder

This commit is contained in:
leigh-mil 2019-07-18 11:32:13 -04:00
parent b3515bb14d
commit 74d4974437

View File

@ -1,5 +1,4 @@
import datetime
from flask import current_app as app
from atst.database import db
from atst.models.clin import CLIN
@ -11,10 +10,6 @@ class TaskOrders(BaseDomainClass):
model = TaskOrder
resource_name = "task_order"
SECTIONS = {"app_info": ["portfolio_name"], "funding": [], "oversight": []}
UNCLASSIFIED_FUNDING = []
@classmethod
def create(cls, creator, portfolio_id, number, clins, pdf):
task_order = TaskOrder(
@ -70,43 +65,6 @@ class TaskOrders(BaseDomainClass):
db.session.add(clin)
db.session.commit()
@classmethod
def section_completion_status(cls, task_order, section):
if section in TaskOrders.mission_owner_sections():
passed = []
failed = []
for attr in TaskOrders.SECTIONS[section]:
if getattr(task_order, attr) is not None:
passed.append(attr)
else:
failed.append(attr)
if not failed:
return "complete"
elif passed and failed:
return "draft"
return "incomplete"
@classmethod
def all_sections_complete(cls, task_order):
for section in TaskOrders.SECTIONS.keys():
if (
TaskOrders.section_completion_status(task_order, section)
is not "complete"
):
return False
return True
@classmethod
def mission_owner_sections(cls):
section_list = TaskOrders.SECTIONS
if not app.config.get("CLASSIFIED"):
section_list["funding"] = TaskOrders.UNCLASSIFIED_FUNDING
return section_list
@classmethod
def sort(cls, task_orders: [TaskOrder]) -> [TaskOrder]:
# Sorts a list of task orders on two keys: status (primary) and time_created (secondary)