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

Merge pull request #842 from dod-ccpo/fix-remove-sample-script

Browse Source 
Fix remove sample script
This commit is contained in:
leigh-mil 2019-05-23 09:37:49 -04:00 committed by GitHub
commit 422f9b5298
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
script/remove_sample_data.py
View File

@ -63,18 +63,23 @@ def remove_sample_data(all_users=False):
all_portfolios = Portfolios.for_user(user) all_portfolios = Portfolios.for_user(user)
portfolios = [p for p in all_portfolios if p.owner == user] portfolios = [p for p in all_portfolios if p.owner == user]
ws_audit = ( portfolio_audit = (
db.session.query(AuditEvent) db.session.query(AuditEvent)
.filter(AuditEvent.portfolio_id.in_([w.id for w in portfolios])) .filter(AuditEvent.portfolio_id.in_([w.id for w in portfolios]))
.all() .all()
) )
portfolio_roles = [role for portfolio in portfolios for role in portfolio.roles] portfolio_roles = [role for portfolio in portfolios for role in portfolio.roles]
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] portfolio_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 = [ application_roles = [
a for application in applications for a in application.roles a for application in applications for a in application.roles
] ]
application_invites = [
invite for role in application_roles for invite in role.invitations
]
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]))
@ -83,14 +88,15 @@ 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 [
portfolio_invites,
application_invites,
roles,
environments,
application_roles, application_roles,
applications, applications,
environments,
invites,
portfolio_roles, portfolio_roles,
roles,
task_orders, task_orders,
ws_audit, portfolio_audit,
]: ]:
for thing in set_of_things: for thing in set_of_things:
db.session.delete(thing) db.session.delete(thing)