def __init_forecast(state: State):
active_player = state.active_player
top_cards = set(state.infection_deck[:6])
ccp = ChooseCardsPhase(
next_phase=state.previous_phase,
cards_to_choose_from=top_cards,
count=len(top_cards),
player=active_player,
after=__forecast_after,
)
state.start_choose_cards_phase(ccp)
def __init_cure_virus(state: State):
active_player = state.active_player
cure_cards = {
city
for city in state.players[active_player].city_cards
if CITY_DATA[city].color == state.virus_to_cure
}
ccp = ChooseCardsPhase(
next_phase=state.previous_phase,
cards_to_choose_from=cure_cards,
count=num_cards_for_cure(state.active_player),
player=active_player,
after=__cure_virus_after,
)
state.start_choose_cards_phase(ccp)