Add portfolio to ShowTaskOrderWorkflow class

This commit is contained in:
Montana
2019-03-08 10:37:59 -05:00
parent aa206192be
commit 24886fc482
2 changed files with 27 additions and 24 deletions

View File

@@ -62,6 +62,12 @@ class ShowTaskOrderWorkflow:
return self._task_order
@property
def portfolio(self):
if self.task_order:
return self.task_order.portfolio
return Portfolios.get(self.user, self.portfolio_id)
@property
def form(self):
form_type = (
@@ -129,11 +135,6 @@ class ShowTaskOrderWorkflow:
return True
return False
def get_portfolio(self):
if self.task_order:
return self.task_order.portfolio
return Portfolios.get(self.user, self.portfolio_id)
class UpdateTaskOrderWorkflow(ShowTaskOrderWorkflow):
def __init__(
@@ -238,7 +239,7 @@ def new(screen, task_order_id=None, portfolio_id=None):
}
if workflow.pf_attributes_read_only():
template_args["portfolio"] = workflow.get_portfolio()
template_args["portfolio"] = workflow.portfolio
if screen == 1:
workflow.form = task_order_form.AppInfoWithExistingPortfolioForm(
obj=workflow.task_order