From 8463599d7baca907e284a942eb0064d7028d0c63 Mon Sep 17 00:00:00 2001 From: Montana Date: Thu, 10 Jan 2019 13:47:01 -0500 Subject: [PATCH] Allow users to update TO information after it has been saved in the DB --- atst/routes/task_orders/new.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/atst/routes/task_orders/new.py b/atst/routes/task_orders/new.py index fba36ac7..5417ecc3 100644 --- a/atst/routes/task_orders/new.py +++ b/atst/routes/task_orders/new.py @@ -132,6 +132,18 @@ class UpdateTaskOrderWorkflow(ShowTaskOrderWorkflow): return to_data + @property + def task_order_form_data(self): + to_data = self.form.data.copy() + if "portfolio_name" in to_data: + new_name = self.form.data["portfolio_name"] + old_name = self.task_order.to_dictionary()["portfolio_name"] + if not new_name is old_name: + Portfolios.update(self.task_order.portfolio, {"name": new_name}) + to_data.pop("portfolio_name") + + return to_data + def validate(self): return self.form.validate()