diff --git a/script/seed_sample.py b/script/seed_sample.py index f6534f27..d4944d4c 100644 --- a/script/seed_sample.py +++ b/script/seed_sample.py @@ -122,11 +122,22 @@ def add_members_to_portfolio(portfolio, users): db.session.commit() -def add_active_task_order(portfolio, active_exp_days=90, clin_01=None, clin_03=None): - start = random_past_date(year_max=1, year_min=1) +def add_task_orders_to_portfolio(portfolio, to_length=90, clin_01=None, clin_03=None): + active_to_offset = random.randint(10,31) + active_start = (date.today() - timedelta(days=active_to_offset)) # exp TO ends same day as active TO starts + active_end = (active_start + timedelta(to_length)) # pending TO starts the same day active TO ends + pending_end = (active_end + timedelta(to_length)) + exp_start = (active_start - timedelta(to_length)) + + create_task_order(portfolio, start=exp_start, end=active_start) + create_task_order(portfolio, start=active_start, end=active_end, clin_01=clin_01, clin_03=clin_03) + create_task_order(portfolio, start=active_end, end=pending_end) + + +def create_task_order(portfolio, start, end, clin_01=None, clin_03=None): default_kwargs = { "start_date": start, - "end_date": (date.today() + timedelta(days=active_exp_days)), + "end_date": end, "number": random_task_order_number(), "portfolio": portfolio, "clin_02": 0, @@ -143,31 +154,6 @@ def add_active_task_order(portfolio, active_exp_days=90, clin_01=None, clin_03=N db.session.commit() -def add_expired_task_order(portfolio): - start = random_past_date(year_max=3, year_min=2) - task_order = TaskOrderFactory.build( - start_date=start, - end_date=(start + timedelta(days=90)), - number=random_task_order_number(), - portfolio=portfolio, - ) - db.session.add(task_order) - db.session.commit() - - -def add_pending_task_order(portfolio): - start_date = random_future_date(year_min=1, year_max=2) - - task_order = TaskOrderFactory.build( - start_date=start_date, - end_date=(start_date + timedelta(days=90)), - number=random_task_order_number(), - portfolio=portfolio, - ) - db.session.add(task_order) - db.session.commit() - - def add_applications_to_portfolio(portfolio, applications): for application in applications: Applications.create( @@ -195,9 +181,7 @@ def create_demo_portfolio(name, data): clin_01 = data["budget"] * 0.8 clin_03 = data["budget"] * 0.2 - add_active_task_order(portfolio, clin_01=clin_01, clin_03=clin_03) - add_expired_task_order(portfolio) - add_pending_task_order(portfolio) + add_task_orders_to_portfolio(portfolio, clin_01=clin_01, clin_03=clin_03) add_members_to_portfolio(portfolio, users=get_users()) for mock_application in data["applications"]: @@ -225,34 +209,27 @@ def seed_db(): create_demo_portfolio("A-Wing", MockReportingProvider.REPORT_FIXTURE_MAP["A-Wing"]) create_demo_portfolio("B-Wing", MockReportingProvider.REPORT_FIXTURE_MAP["B-Wing"]) - # Create Portfolio for Amanda with TO that is expiring soon and does not have another TO - unfunded_portfolio = Portfolios.create( + tie_interceptor = Portfolios.create( amanda, name="TIE Interceptor", defense_component=random_service_branch() ) - add_active_task_order(unfunded_portfolio, active_exp_days=20) - add_expired_task_order(unfunded_portfolio) - add_members_to_portfolio(unfunded_portfolio, users=users) - add_applications_to_portfolio(unfunded_portfolio, application_info) + add_task_orders_to_portfolio(tie_interceptor) + add_members_to_portfolio(tie_interceptor, users=users) + add_applications_to_portfolio(tie_interceptor, application_info) - # Create Portfolio for Amanda with TO that is expiring soon and has another TO - funded_portfolio = Portfolios.create( + tie_fighter = Portfolios.create( amanda, name="TIE Fighter", defense_component=random_service_branch() ) - add_active_task_order(funded_portfolio, active_exp_days=20) - add_expired_task_order(funded_portfolio) - add_pending_task_order(funded_portfolio) - add_members_to_portfolio(funded_portfolio, users=users) - add_applications_to_portfolio(funded_portfolio, application_info) + add_task_orders_to_portfolio(tie_fighter) + add_members_to_portfolio(tie_fighter, users=users) + add_applications_to_portfolio(tie_fighter, application_info) # create a portfolio 'Y-Wing' for each user for user in users: portfolio = Portfolios.create( user, name="Y-Wing", defense_component=random_service_branch() ) + add_task_orders_to_portfolio(portfolio) add_members_to_portfolio(portfolio, users=users) - add_active_task_order(portfolio) - add_expired_task_order(portfolio) - add_pending_task_order(portfolio) add_applications_to_portfolio(portfolio, application_info)