pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #774 from dod-ccpo/remove-sample-data

Browse Source 
Fix remove_sample_data.py script
This commit is contained in:
George Drummond 2019-04-18 09:41:44 -04:00 committed by GitHub
commit d963130674
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
script/remove_sample_data.py
View File

@ -72,6 +72,9 @@ def remove_sample_data(all_users=False):
task_orders = [to for portfolio in portfolios for to in portfolio.task_orders] 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] invites = [invite for role in portfolio_roles for invite in role.invitations]
applications = [p for portfolio in portfolios for p in portfolio.applications] applications = [p for portfolio in portfolios for p in portfolio.applications]
application_roles = [
a for application in applications for a in application.roles
]
environments = ( environments = (
db.session.query(Environment) db.session.query(Environment)
.filter(Environment.application_id.in_([p.id for p in applications])) .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] roles = [role for env in environments for role in env.roles]
for set_of_things in [ for set_of_things in [
roles, application_roles,
environments,
applications, applications,
environments,
invites, invites,
task_orders,
portfolio_roles, portfolio_roles,
roles,
task_orders,
ws_audit, ws_audit,
]: ]:
for thing in set_of_things: for thing in set_of_things: