From 5bf8022a51eedb459396fbcee771c5b5ad1f57b3 Mon Sep 17 00:00:00 2001 From: George Drummond Date: Mon, 15 Apr 2019 13:41:09 -0400 Subject: [PATCH] Fix specs --- atst/routes/portfolios/applications.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/atst/routes/portfolios/applications.py b/atst/routes/portfolios/applications.py index 20603b3c..62eb5000 100644 --- a/atst/routes/portfolios/applications.py +++ b/atst/routes/portfolios/applications.py @@ -57,6 +57,15 @@ def create_application(portfolio_id): ) +def environments_obj(application): + environments_obj = {} + + for env in application.environments: + environments_obj[env.name] = [user.full_name for user in env.users] + + return environments_obj + + @portfolios_bp.route("/portfolios//applications//edit") @user_can(Permissions.EDIT_APPLICATION, message="view application edit form") def edit_application(portfolio_id, application_id): @@ -64,16 +73,12 @@ def edit_application(portfolio_id, application_id): application = Applications.get(application_id) form = ApplicationForm(name=application.name, description=application.description) - environments_obj = {} - for env in application.environments: - environments_obj[env.name] = [user.full_name for user in env.users] - return render_template( "portfolios/applications/edit.html", portfolio=portfolio, application=application, form=form, - environments_obj=environments_obj, + environments_obj=environments_obj(application=application), ) @@ -98,6 +103,7 @@ def update_application(portfolio_id, application_id): portfolio=portfolio, application=application, form=form, + environments_obj=environments_obj(application=application), )