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

Move inviting members into separate function so portfolios can be created where all members have accepted invites

Browse Source
This commit is contained in:
leigh-mil 2019-03-04 13:32:53 -05:00
parent b82b5bb9c0
commit 04dc6703b2
1 changed files with 26 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
script/seed_sample.py
View File

@ -101,23 +101,28 @@ def get_users():
return users return users
def add_members_to_portfolio(portfolio, users): def add_members_to_portfolio(portfolio):
for user in users: get_users()
for portfolio_role in PORTFOLIO_USERS: for portfolio_role in PORTFOLIO_USERS:
ws_role = Portfolios.create_member( ws_role = Portfolios.create_member(
portfolio.owner, portfolio, portfolio_role portfolio.owner, portfolio, portfolio_role
) )
db.session.refresh(ws_role) db.session.refresh(ws_role)
PortfolioRoles.enable(ws_role) PortfolioRoles.enable(ws_role)
for portfolio_role in PORTFOLIO_INVITED_USERS: db.session.commit()
ws_role = Portfolios.create_member(
portfolio.owner, portfolio, portfolio_role
) def invite_members_to_portfolio(portfolio):
invitation = InvitationFactory.build( get_users()
portfolio_role=ws_role, status=portfolio_role["status"] for portfolio_role in PORTFOLIO_INVITED_USERS:
) ws_role = Portfolios.create_member(
db.session.add(invitation) portfolio.owner, portfolio, portfolio_role
)
invitation = InvitationFactory.build(
portfolio_role=ws_role, status=portfolio_role["status"]
)
db.session.add(invitation)
db.session.commit() db.session.commit()
@ -182,7 +187,7 @@ def create_demo_portfolio(name, data):
clin_03 = data["budget"] * 0.2 clin_03 = data["budget"] * 0.2
add_task_orders_to_portfolio(portfolio, clin_01=clin_01, clin_03=clin_03) add_task_orders_to_portfolio(portfolio, clin_01=clin_01, clin_03=clin_03)
add_members_to_portfolio(portfolio, users=get_users()) add_members_to_portfolio(portfolio)
for mock_application in data["applications"]: for mock_application in data["applications"]:
application = Application( application = Application(
@ -195,7 +200,6 @@ def create_demo_portfolio(name, data):
def seed_db(): def seed_db():
users = get_users()
amanda = Users.get_by_dod_id("2345678901") amanda = Users.get_by_dod_id("2345678901")
application_info = [ application_info = [
{ {
@ -213,23 +217,23 @@ def seed_db():
amanda, name="TIE Interceptor", defense_component=random_service_branch() amanda, name="TIE Interceptor", defense_component=random_service_branch()
) )
add_task_orders_to_portfolio(tie_interceptor) add_task_orders_to_portfolio(tie_interceptor)
add_members_to_portfolio(tie_interceptor, users=users) add_members_to_portfolio(tie_interceptor)
add_applications_to_portfolio(tie_interceptor, application_info) add_applications_to_portfolio(tie_interceptor, application_info)
tie_fighter = Portfolios.create( tie_fighter = Portfolios.create(
amanda, name="TIE Fighter", defense_component=random_service_branch() amanda, name="TIE Fighter", defense_component=random_service_branch()
) )
add_task_orders_to_portfolio(tie_fighter) add_task_orders_to_portfolio(tie_fighter)
add_members_to_portfolio(tie_fighter, users=users) add_members_to_portfolio(tie_fighter)
add_applications_to_portfolio(tie_fighter, application_info) add_applications_to_portfolio(tie_fighter, application_info)
# create a portfolio 'Y-Wing' for each user # create a portfolio 'Y-Wing' for each user
for user in users: for user in get_users():
portfolio = Portfolios.create( portfolio = Portfolios.create(
user, name="Y-Wing", defense_component=random_service_branch() user, name="Y-Wing", defense_component=random_service_branch()
) )
add_task_orders_to_portfolio(portfolio) add_task_orders_to_portfolio(portfolio)
add_members_to_portfolio(portfolio, users=users) add_members_to_portfolio(portfolio)
add_applications_to_portfolio(portfolio, application_info) add_applications_to_portfolio(portfolio, application_info)