from streamlit.source_util import get_pages


SEASON_MODE = "season"
OFFSEASON_MODE = "offseason"

CURRENT_MODE = SEASON_MODE

MAIN_PAGE_FILE = "Home.py"

MODE_PAGE_EXCLUSION_MAP = {
    SEASON_MODE: [
        "Keepers",
        "ECR",
    ],
    OFFSEASON_MODE: ["Practice_Reports"],
}


def remove_seasonal_pages():
    all_pages = get_pages(MAIN_PAGE_FILE)
    pages_to_remove = MODE_PAGE_EXCLUSION_MAP[CURRENT_MODE]

    page_keys_to_remove = []
    for k, v in all_pages.items():
        if v["page_name"] in pages_to_remove:
            page_keys_to_remove.append(k)
    for k_remove in page_keys_to_remove:
        del all_pages[k_remove]