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

fix sample data scripts

Browse Source
This commit is contained in:
dandds 2018-12-10 10:21:32 -05:00
parent d01910cc6f
commit e6d267a568
2 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
script/remove_sample_data.py
View File

@ -101,11 +101,6 @@ def remove_sample_data(all_users=False):
.filter(AuditEvent.request_id.in_([r.id for r in requests])) .filter(AuditEvent.request_id.in_([r.id for r in requests]))
.all() .all()
) )
request_audit = (
db.session.query(AuditEvent)
.filter(AuditEvent.request_id.in_([r.id for r in requests]))
.all()
)
events = [ev for r in requests for ev in r.status_events] events = [ev for r in requests for ev in r.status_events]
revisions = [rev for r in requests for rev in r.revisions] revisions = [rev for r in requests for rev in r.revisions]
workspaces = [r.workspace for r in requests if r.workspace] workspaces = [r.workspace for r in requests if r.workspace]
@ -115,6 +110,7 @@ def remove_sample_data(all_users=False):
.all() .all()
) )
workspace_roles = [role for workspace in workspaces for role in workspace.roles] workspace_roles = [role for workspace in workspaces for role in workspace.roles]
invites = [invite for role in workspace_roles for invite in role.invitations]
projects = [p for workspace in workspaces for p in workspace.projects] projects = [p for workspace in workspaces for p in workspace.projects]
environments = ( environments = (
db.session.query(Environment) db.session.query(Environment)
@ -127,6 +123,7 @@ def remove_sample_data(all_users=False):
roles, roles,
environments, environments,
projects, projects,
invites,
workspace_roles, workspace_roles,
ws_audit, ws_audit,
events, events,

1
script/seed_sample.py
View File

@ -112,6 +112,7 @@ def seed_db():
) )
for workspace_role in WORKSPACE_USERS: for workspace_role in WORKSPACE_USERS:
ws_role = Workspaces.create_member(user, workspace, workspace_role) ws_role = Workspaces.create_member(user, workspace, workspace_role)
db.session.refresh(ws_role)
WorkspaceRoles.enable(ws_role) WorkspaceRoles.enable(ws_role)
for workspace_role in WORKSPACE_INVITED_USERS: for workspace_role in WORKSPACE_INVITED_USERS: