From 5434443b02107c636a5e1769a9fb2cbf71b74e90 Mon Sep 17 00:00:00 2001 From: dandds Date: Mon, 3 Jun 2019 09:39:14 -0400 Subject: [PATCH] Adjust `script/seed_sample.py` for new portfolio invitation flow. --- script/seed_sample.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/script/seed_sample.py b/script/seed_sample.py index a4284ead..a0becb96 100644 --- a/script/seed_sample.py +++ b/script/seed_sample.py @@ -152,9 +152,9 @@ def get_users(): def add_members_to_portfolio(portfolio): for user_data in PORTFOLIO_USERS: - ws_role = Portfolios.create_member(portfolio, user_data) - db.session.refresh(ws_role) - PortfolioRoles.enable(ws_role) + invite = Portfolios.invite(portfolio, portfolio.owner, user_data) + user = Users.get_or_create_by_dod_id(user_data["dod_id"]) + PortfolioRoles.enable(invite.role, user) db.session.commit()