diff --git a/script/seed_sample.py b/script/seed_sample.py index 1b17d28b..f29fb5f6 100644 --- a/script/seed_sample.py +++ b/script/seed_sample.py @@ -53,41 +53,6 @@ PORTFOLIO_USERS = [ }, ] -PORTFOLIO_INVITED_USERS = [ - { - "first_name": "Frederick", - "last_name": "Fitzgerald", - "email": "frederick@mil.gov", - "portfolio_role": "developer", - "dod_id": "0000000004", - "status": InvitationStatus.REJECTED_WRONG_USER, - }, - { - "first_name": "Gina", - "last_name": "Guzman", - "email": "gina@mil.gov", - "portfolio_role": "developer", - "dod_id": "0000000005", - "status": InvitationStatus.REJECTED_EXPIRED, - }, - { - "first_name": "Hector", - "last_name": "Harper", - "email": "hector@mil.gov", - "portfolio_role": "developer", - "dod_id": "0000000006", - "status": InvitationStatus.REVOKED, - }, - { - "first_name": "Isabella", - "last_name": "Ingram", - "email": "isabella@mil.gov", - "portfolio_role": "developer", - "dod_id": "0000000007", - "status": InvitationStatus.PENDING, - }, -] - def get_users(): users = [] @@ -102,7 +67,6 @@ def get_users(): def add_members_to_portfolio(portfolio): - get_users() for portfolio_role in PORTFOLIO_USERS: ws_role = Portfolios.create_member(portfolio.owner, portfolio, portfolio_role) db.session.refresh(ws_role) @@ -111,18 +75,6 @@ def add_members_to_portfolio(portfolio): db.session.commit() -def invite_members_to_portfolio(portfolio): - get_users() - for portfolio_role in PORTFOLIO_INVITED_USERS: - ws_role = Portfolios.create_member(portfolio.owner, portfolio, portfolio_role) - invitation = InvitationFactory.build( - portfolio_role=ws_role, status=portfolio_role["status"] - ) - db.session.add(invitation) - - db.session.commit() - - def add_task_orders_to_portfolio(portfolio, to_length=90, clin_01=None, clin_03=None): active_to_offset = random.randint(10, 31) # exp TO ends same day as active TO starts @@ -200,6 +152,7 @@ def create_demo_portfolio(name, data): def seed_db(): + get_users() amanda = Users.get_by_dod_id("2345678901") application_info = [ {