Update TO form based on new model

This commit is contained in:
leigh-mil
2019-06-01 17:02:18 -04:00
committed by Montana
parent c90e574e83
commit 91e41199b7
11 changed files with 92 additions and 870 deletions

View File

@@ -18,8 +18,10 @@ class TaskOrders(BaseDomainClass):
UNCLASSIFIED_FUNDING = []
@classmethod
def create(cls, creator, portfolio):
task_order = TaskOrder(portfolio=portfolio, creator=creator)
def create(cls, creator, portfolio_id, **kwargs):
task_order = TaskOrder(portfolio_id=portfolio_id, creator=creator)
for key, value in kwargs.items():
setattr(task_order, key, value)
db.session.add(task_order)
db.session.commit()
@@ -27,7 +29,8 @@ class TaskOrders(BaseDomainClass):
return task_order
@classmethod
def update(cls, task_order, **kwargs):
def update(cls, task_order_id, **kwargs):
task_order = TaskOrders.get(task_order_id)
for key, value in kwargs.items():
setattr(task_order, key, value)