From 45be1e5100f03fa8e859d7c1c8aa9d0a1613ccba Mon Sep 17 00:00:00 2001 From: George Drummond Date: Wed, 17 Apr 2019 16:05:33 -0400 Subject: [PATCH] Fix remove_sample_data.py script --- script/remove_sample_data.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/script/remove_sample_data.py b/script/remove_sample_data.py index 6528db8d..73da5b11 100644 --- a/script/remove_sample_data.py +++ b/script/remove_sample_data.py @@ -72,6 +72,9 @@ def remove_sample_data(all_users=False): task_orders = [to for portfolio in portfolios for to in portfolio.task_orders] invites = [invite for role in portfolio_roles for invite in role.invitations] applications = [p for portfolio in portfolios for p in portfolio.applications] + application_roles = [ + a for application in applications for a in application.roles + ] environments = ( db.session.query(Environment) .filter(Environment.application_id.in_([p.id for p in applications])) @@ -80,12 +83,13 @@ def remove_sample_data(all_users=False): roles = [role for env in environments for role in env.roles] for set_of_things in [ - roles, - environments, + application_roles, applications, + environments, invites, - task_orders, portfolio_roles, + roles, + task_orders, ws_audit, ]: for thing in set_of_things: