Fix form so TO saves properly

This commit is contained in:
leigh-mil
2019-06-07 15:29:25 -04:00
parent 454643c3c9
commit 732830a500
2 changed files with 71 additions and 18 deletions

View File

@@ -19,9 +19,9 @@ class TaskOrders(BaseDomainClass):
UNCLASSIFIED_FUNDING = []
@classmethod
def create(cls, creator, portfolio_id, number, clins):
def create(cls, creator, portfolio_id, number, clins, pdf):
task_order = TaskOrder(
portfolio_id=portfolio_id, creator=creator, number=number
portfolio_id=portfolio_id, creator=creator, number=number, pdf=pdf
)
db.session.add(task_order)
db.session.commit()
@@ -31,8 +31,9 @@ class TaskOrders(BaseDomainClass):
return task_order
@classmethod
def update(cls, task_order_id, number, clins):
def update(cls, task_order_id, number, clins, pdf):
task_order = TaskOrders.get(task_order_id)
task_order.pdf = pdf
for clin in task_order.clins:
db.session.delete(clin)